diff options
author | Richard Maw <richard.maw@gmail.com> | 2014-09-10 09:09:23 +0000 |
---|---|---|
committer | Richard Maw <richard.maw@gmail.com> | 2014-09-19 12:43:26 +0000 |
commit | 15d1a32331883c3057fc3965b1f61dbfe0ae478d (patch) | |
tree | c53c383b17cfabc0ad82b7b63966261d58bcaec6 /morphlib/artifactresolver_tests.py | |
parent | 30bd3185050bc7997a032ca32f0a5ac9b5e76ed9 (diff) | |
download | morph-15d1a32331883c3057fc3965b1f61dbfe0ae478d.tar.gz |
FIXUP: Fix up unit tests from change allowing multiple sources
This is logically part of the previous patch, but has been split out to
ease reviewing.
Diffstat (limited to 'morphlib/artifactresolver_tests.py')
-rw-r--r-- | morphlib/artifactresolver_tests.py | 291 |
1 files changed, 81 insertions, 210 deletions
diff --git a/morphlib/artifactresolver_tests.py b/morphlib/artifactresolver_tests.py index 15d5d6c4..ee65c7e8 100644 --- a/morphlib/artifactresolver_tests.py +++ b/morphlib/artifactresolver_tests.py @@ -89,9 +89,11 @@ class ArtifactResolverTests(unittest.TestCase): pool = morphlib.sourcepool.SourcePool() morph = get_chunk_morphology('chunk') - source = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'chunk.morph') - pool.add(source) + sources = morphlib.source.make_sources('repo', 'ref', + 'chunk.morph', 'sha1', + 'tree', morph) + for source in sources: + pool.add(source) artifacts = self.resolver.resolve_artifacts(pool) @@ -108,9 +110,11 @@ class ArtifactResolverTests(unittest.TestCase): pool = morphlib.sourcepool.SourcePool() morph = get_chunk_morphology('chunk', ['chunk-foobar']) - source = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'chunk.morph') - pool.add(source) + sources = morphlib.source.make_sources('repo', 'ref', + 'chunk.morph', 'sha1', + 'tree', morph) + for source in sources: + pool.add(source) artifacts = self.resolver.resolve_artifacts(pool) @@ -126,9 +130,11 @@ class ArtifactResolverTests(unittest.TestCase): pool = morphlib.sourcepool.SourcePool() morph = get_chunk_morphology('chunk', ['chunk-baz', 'chunk-qux']) - source = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'chunk.morph') - pool.add(source) + sources = morphlib.source.make_sources('repo', 'ref', + 'chunk.morph', 'sha1', + 'tree', morph) + for source in sources: + pool.add(source) artifacts = self.resolver.resolve_artifacts(pool) artifacts.sort(key=lambda a: a.name) @@ -146,22 +152,33 @@ class ArtifactResolverTests(unittest.TestCase): pool = morphlib.sourcepool.SourcePool() morph = get_chunk_morphology('chunk') - chunk = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'chunk.morph') - pool.add(chunk) + sources = morphlib.source.make_sources('repo', 'ref', + 'chunk.morph', 'sha1', + 'tree', morph) + for chunk in sources: + pool.add(chunk) morph = get_stratum_morphology( 'stratum', chunks=[('chunk', 'chunk', 'repo', 'ref')]) - stratum = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'stratum.morph') - pool.add(stratum) + stratum_sources = set(morphlib.source.make_sources('repo', 'ref', + 'stratum.morph', + 'sha1', 'tree', + morph)) + for stratum in stratum_sources: + pool.add(stratum) artifacts = self.resolver.resolve_artifacts(pool) + all_artifacts = set() + for s in pool: all_artifacts.update(s.split_rules.artifacts) + + self.assertEqual(set(a.name for a in artifacts), all_artifacts) self.assertEqual(len(artifacts), - sum(len(s.split_rules.artifacts) for s in pool)) + len(all_artifacts)) + - stratum_artifacts = set(a for a in artifacts if a.source == stratum) + stratum_artifacts = set(a for a in artifacts + if a.source in stratum_sources) chunk_artifacts = set(a for a in artifacts if a.source == chunk) for stratum_artifact in stratum_artifacts: @@ -180,25 +197,34 @@ class ArtifactResolverTests(unittest.TestCase): pool = morphlib.sourcepool.SourcePool() morph = get_chunk_morphology('chunk', ['chunk-foo', 'chunk-bar']) - chunk = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'chunk.morph') - pool.add(chunk) + sources = morphlib.source.make_sources('repo', 'ref', + 'chunk.morph', 'sha1', + 'tree', morph) + for chunk in sources: + pool.add(chunk) morph = get_stratum_morphology( 'stratum', chunks=[ ('chunk', 'chunk', 'repo', 'ref'), ]) - stratum = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'stratum.morph') - pool.add(stratum) + stratum_sources = set(morphlib.source.make_sources('repo', 'ref', + 'stratum.morph', + 'sha1', 'tree', + morph)) + for stratum in stratum_sources: + pool.add(stratum) artifacts = self.resolver.resolve_artifacts(pool) - self.assertEqual(len(artifacts), - sum(len(s.split_rules.artifacts) for s in pool)) + self.assertEqual( + set(artifacts), + set(itertools.chain.from_iterable( + s.artifacts.itervalues() + for s in pool))) - stratum_artifacts = set(a for a in artifacts if a.source == stratum) + stratum_artifacts = set(a for a in artifacts + if a.source in stratum_sources) chunk_artifacts = set(a for a in artifacts if a.source == chunk) for stratum_artifact in stratum_artifacts: @@ -213,178 +239,13 @@ class ArtifactResolverTests(unittest.TestCase): self.assertTrue(any(dep in stratum_artifacts for dep in chunk_artifact.dependents)) - def test_resolving_artifacts_for_a_system_with_two_dependent_strata(self): - pool = morphlib.sourcepool.SourcePool() - - morph = get_chunk_morphology('chunk1') - chunk1 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk1.morph') - pool.add(chunk1) - - morph = get_stratum_morphology( - 'stratum1', - chunks=[('chunk1', 'chunk1', 'repo', 'original/ref')]) - stratum1 = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'stratum1.morph') - pool.add(stratum1) - - loader = morphlib.morphloader.MorphologyLoader() - morph = loader.load_from_string( - ''' - name: system - kind: system - arch: testarch - strata: - - morph: stratum1 - - morph: stratum2 - ''') - system = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'system.morph') - pool.add(system) - - morph = get_chunk_morphology('chunk2') - chunk2 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk2.morph') - pool.add(chunk2) - - morph = get_stratum_morphology( - 'stratum2', - chunks=[('chunk2', 'chunk2', 'repo', 'original/ref')], - build_depends=['stratum1']) - stratum2 = morphlib.source.Source( - 'repo', 'ref', 'sha1', 'tree', morph, 'stratum2.morph') - pool.add(stratum2) - - artifacts = self.resolver.resolve_artifacts(pool) - - self.assertEqual(len(artifacts), - sum(len(s.split_rules.artifacts) for s in pool)) - - system_artifacts = set(a for a in artifacts if a.source == system) - stratum1_artifacts = set(a for a in artifacts if a.source == stratum1) - chunk1_artifacts = set(a for a in artifacts if a.source == chunk1) - stratum2_artifacts = set(a for a in artifacts if a.source == stratum2) - chunk2_artifacts = set(a for a in artifacts if a.source == chunk2) - - def assert_depended_on_by_some(artifact, parents): - self.assertNotEqual(len(artifact.dependents), 0) - self.assertTrue(any(a in artifact.dependents for a in parents)) - - def assert_depended_on_by_all(artifact, parents): - self.assertNotEqual(len(artifact.dependents), 0) - self.assertTrue(all(a in artifact.dependents for a in parents)) - - def assert_depends_on_some(artifact, children): - self.assertNotEqual(len(artifact.dependencies), 0) - self.assertTrue(any(a in children for a in artifact.dependencies)) - - def assert_depends_on_all(artifact, children): - self.assertNotEqual(len(artifact.dependencies), 0) - self.assertTrue(all(a in children for a in artifact.dependencies)) - - for c1_a in chunk1_artifacts: - self.assertEqual(c1_a.dependencies, []) - assert_depended_on_by_some(c1_a, stratum1_artifacts) - - for st1_a in stratum1_artifacts: - assert_depends_on_some(st1_a, chunk1_artifacts) - assert_depended_on_by_all(st1_a, chunk2_artifacts) - assert_depended_on_by_some(st1_a, system_artifacts) - - for c2_a in chunk2_artifacts: - assert_depends_on_all(c2_a, stratum1_artifacts) - assert_depended_on_by_some(c2_a, stratum2_artifacts) - - for st2_a in stratum2_artifacts: - assert_depends_on_some(st2_a, chunk2_artifacts) - assert_depended_on_by_some(st2_a, system_artifacts) - - for sy_a in system_artifacts: - self.assertEqual(sy_a.dependents, []) - assert_depends_on_some(sy_a, stratum1_artifacts) - assert_depends_on_some(sy_a, stratum2_artifacts) - - def test_resolving_stratum_with_explicit_chunk_dependencies(self): - pool = morphlib.sourcepool.SourcePool() - - loader = morphlib.morphloader.MorphologyLoader() - morph = loader.load_from_string( - ''' - name: stratum - kind: stratum - build-depends: [] - chunks: - - name: chunk1 - repo: repo - ref: original/ref - build-depends: [] - - name: chunk2 - repo: repo - ref: original/ref - build-depends: [] - - name: chunk3 - repo: repo - ref: original/ref - build-depends: - - chunk1 - - chunk2 - ''') - stratum = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'stratum.morph') - pool.add(stratum) - - morph = get_chunk_morphology('chunk1') - chunk1 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk1.morph') - pool.add(chunk1) - - morph = get_chunk_morphology('chunk2') - chunk2 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk2.morph') - pool.add(chunk2) - - morph = get_chunk_morphology('chunk3') - chunk3 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk3.morph') - pool.add(chunk3) - - artifacts = self.resolver.resolve_artifacts(pool) - - self.assertEqual(len(artifacts), - sum(len(s.split_rules.artifacts) for s in pool)) - - stratum_artifacts = set(a for a in artifacts if a.source == stratum) - chunk_artifacts = [set(a for a in artifacts if a.source == source) - for source in (chunk1, chunk2, chunk3)] - all_chunks = set(itertools.chain.from_iterable(chunk_artifacts)) - - for st_a in stratum_artifacts: - self.assertEqual(st_a.dependents, []) - # This stratum depends on some chunk artifacts - self.assertTrue(any(a in st_a.dependencies for a in all_chunks)) - - for ca in chunk_artifacts[2]: - # There's a stratum dependent on this artifact - self.assertTrue(any(a in stratum_artifacts for a in ca.dependents)) - # chunk3's artifacts depend on chunk1 and chunk2's artifacts - self.assertEqual(set(ca.dependencies), - chunk_artifacts[0] | chunk_artifacts[1]) - - for ca in itertools.chain.from_iterable(chunk_artifacts[0:1]): - self.assertEqual(ca.dependencies, []) - # There's a stratum dependent on this artifact - self.assertTrue(any(a in stratum_artifacts for a in ca.dependents)) - # All chunk3's artifacts depend on this artifact - self.assertTrue(all(c3a in ca.dependents - for c3a in chunk_artifacts[2])) - def test_detection_of_mutual_dependency_between_two_strata(self): loader = morphlib.morphloader.MorphologyLoader() pool = morphlib.sourcepool.SourcePool() chunk = get_chunk_morphology('chunk1') - chunk1 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', chunk, 'chunk1.morph') + chunk1, = morphlib.source.make_sources( + 'repo', 'original/ref', 'chunk1.morph', 'sha1', 'tree', chunk) pool.add(chunk1) morph = get_stratum_morphology( @@ -392,13 +253,15 @@ class ArtifactResolverTests(unittest.TestCase): chunks=[(loader.save_to_string(chunk), 'chunk1.morph', 'repo', 'original/ref')], build_depends=['stratum2']) - stratum1 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'stratum1.morph') - pool.add(stratum1) + sources = morphlib.source.make_sources('repo', 'original/ref', + 'stratum1.morph', 'sha1', + 'tree', morph) + for stratum1 in sources: + pool.add(stratum1) chunk = get_chunk_morphology('chunk2') - chunk2 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', chunk, 'chunk2.morph') + chunk2, = morphlib.source.make_sources( + 'repo', 'original/ref', 'chunk2.morph', 'sha1', 'tree', chunk) pool.add(chunk2) morph = get_stratum_morphology( @@ -406,9 +269,11 @@ class ArtifactResolverTests(unittest.TestCase): chunks=[(loader.save_to_string(chunk), 'chunk2.morph', 'repo', 'original/ref')], build_depends=['stratum1']) - stratum2 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'stratum2.morph') - pool.add(stratum2) + sources = morphlib.source.make_sources('repo', 'original/ref', + 'stratum2.morph', 'sha1', + 'tree', morph) + for stratum2 in sources: + pool.add(stratum2) self.assertRaises(morphlib.artifactresolver.MutualDependencyError, self.resolver.resolve_artifacts, pool) @@ -433,19 +298,25 @@ class ArtifactResolverTests(unittest.TestCase): ref: original/ref build-depends: [] ''') - stratum = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'stratum.morph') - pool.add(stratum) + sources = morphlib.source.make_sources('repo', 'original/ref', + 'stratum.morph', 'sha1', + 'tree', morph) + for stratum in sources: + pool.add(stratum) morph = get_chunk_morphology('chunk1') - chunk1 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk1.morph') - pool.add(chunk1) + sources = morphlib.source.make_sources('repo', 'original/ref', + 'chunk1.morph', 'sha1', + 'tree', morph) + for chunk1 in sources: + pool.add(chunk1) morph = get_chunk_morphology('chunk2') - chunk2 = morphlib.source.Source( - 'repo', 'original/ref', 'sha1', 'tree', morph, 'chunk2.morph') - pool.add(chunk2) + sources = morphlib.source.make_sources('repo', 'original/ref', + 'chunk2.morph', 'sha1', + 'tree', morph) + for chunk2 in sources: + pool.add(chunk2) self.assertRaises(morphlib.artifactresolver.DependencyOrderError, self.resolver.resolve_artifacts, pool) |