def test_cdimage_archiver_policy(self): """ If we use the cdimage policy, then the file path but should delegate to the default policy for builds without a projectbuild. """ project, dependency = self.create_dependencies() projectbuild = build_project(project, queue_build=False) build = BuildFactory.create(job=dependency.job, build_id=projectbuild.build_key) ArtifactFactory.create(build=build, filename="testing.gz") update_projectbuilds(build) archive = ArchiveFactory.create(policy="cdimage") archive.add_build(build) archived = archive.get_archived_artifacts_for_build(build).order_by( "archived_path") policy = CdimageArchivePolicy() paths = [] for item in archived: projectbuild = (item.projectbuild_dependency and item.projectbuild_dependency.projectbuild or None) paths.append( policy.get_path_for_artifact(item.artifact, build=build, dependency=item.dependency, projectbuild=projectbuild)) self.assertEqual( "\n".join(paths), "\n".join(archived.values_list("archived_path", flat=True)))
def test_cdimage_archiver_policy(self): """ If we use the cdimage policy, then the file path but should delegate to the default policy for builds without a projectbuild. """ project, dependency = self.create_dependencies() projectbuild = build_project(project, queue_build=False) build = BuildFactory.create( job=dependency.job, build_id=projectbuild.build_key) ArtifactFactory.create(build=build, filename="testing.gz") update_projectbuilds(build) archive = ArchiveFactory.create(policy="cdimage") archive.add_build(build) archived = archive.get_archived_artifacts_for_build(build).order_by( "archived_path") policy = CdimageArchivePolicy() paths = [] for item in archived: projectbuild = (item.projectbuild_dependency and item.projectbuild_dependency.projectbuild or None) paths.append(policy.get_path_for_artifact( item.artifact, build=build, dependency=item.dependency, projectbuild=projectbuild)) self.assertEqual( "\n".join(paths), "\n".join(archived.values_list("archived_path", flat=True)))
def test_mappings_into_cdimage_structure(self): """ Artifact paths and filenames are translated into a cdimage-like path, using the project name and build id. """ mock_build = mock.Mock() mock_build.project.name = "my-project" mock_build.build_id = "20010101.0" mock_artifact_1 = mock.Mock() mock_artifact_1.url = "/first/path/thing.txt" mock_artifact_1.filename = "thing.txt" mock_artifact_2 = mock.Mock() mock_artifact_2.url = "/second/path/stuff.txt" mock_artifact_2.filename = "stuff.txt" mock_build.get_current_artifacts() mock_build.get_current_artifacts.return_value = [ mock_artifact_1, mock_artifact_2] policy = CdimageArchivePolicy(projectbuild=mock_build) mappings = policy.get_mappings() expected = { "/first/path/thing.txt": "my-project/20010101.0/thing.txt", "/second/path/stuff.txt": "my-project/20010101.0/stuff.txt" } self.assertEqual(expected, mappings)
def test_archive_build_several_projectbuild_dependencies(self): """ If we archive a build that is used in several projectbuilds, then we should get multiple copies of the artifact. """ project1, dependency1, dependency2 = self.create_dependencies( 2) project2 = ProjectFactory.create(name="Project 2") ProjectDependency.objects.create( project=project2, dependency=dependency1) projectbuild = build_project(project1, queue_build=False) build1 = BuildFactory.create( job=dependency1.job, build_id=projectbuild.build_key) build2 = BuildFactory.create( job=dependency2.job, build_id=projectbuild.build_key) artifact1 = ArtifactFactory.create(build=build1, filename="file1.gz") artifact2 = ArtifactFactory.create(build=build2, filename="file2.gz") archive = ArchiveFactory.create(policy="cdimage") update_projectbuilds(build1) create_projectbuilds_for_autotracking(build1) archive.add_build(build1) self.assertEqual(3, archive.items.count()) update_projectbuilds(build2) create_projectbuilds_for_autotracking(build2) archive.add_build(build2) self.assertEqual(5, archive.items.count()) artifacts = ArchiveArtifact.objects.all().order_by("archived_path") policy = CdimageArchivePolicy() self.assertEqual( "{dependency1}\n{dependency2}\n" "project-1/{build}/file1.gz\nproject-1/{build}/file2.gz\n" "project-2/{build}/file1.gz".format( dependency1=policy.get_path_for_artifact( artifact=artifact1, build=build1, dependency=dependency1), dependency2=policy.get_path_for_artifact( artifact=artifact2, build=build2, dependency=dependency2), build=projectbuild.build_id), "\n".join(artifacts.values_list("archived_path", flat=True)))
def test_archive_build_several_projectbuild_dependencies(self): """ If we archive a build that is used in several projectbuilds, then we should get multiple copies of the artifact. """ project1, dependency1, dependency2 = self.create_dependencies(2) project2 = ProjectFactory.create(name="Project 2") ProjectDependency.objects.create(project=project2, dependency=dependency1) projectbuild = build_project(project1, queue_build=False) build1 = BuildFactory.create(job=dependency1.job, build_id=projectbuild.build_key) build2 = BuildFactory.create(job=dependency2.job, build_id=projectbuild.build_key) artifact1 = ArtifactFactory.create(build=build1, filename="file1.gz") artifact2 = ArtifactFactory.create(build=build2, filename="file2.gz") archive = ArchiveFactory.create(policy="cdimage") update_projectbuilds(build1) create_projectbuilds_for_autotracking(build1) archive.add_build(build1) self.assertEqual(3, archive.items.count()) update_projectbuilds(build2) create_projectbuilds_for_autotracking(build2) archive.add_build(build2) self.assertEqual(5, archive.items.count()) artifacts = ArchiveArtifact.objects.all().order_by("archived_path") policy = CdimageArchivePolicy() self.assertEqual( "{dependency1}\n{dependency2}\n" "project-1/{build}/file1.gz\nproject-1/{build}/file2.gz\n" "project-2/{build}/file1.gz".format( dependency1=policy.get_path_for_artifact( artifact=artifact1, build=build1, dependency=dependency1), dependency2=policy.get_path_for_artifact( artifact=artifact2, build=build2, dependency=dependency2), build=projectbuild.build_id), "\n".join(artifacts.values_list("archived_path", flat=True)))
def test_get_path_for_artifact(self): """ The CdimageArchivePolicy should calculate a cdimage-like path using the project name and build id. url. """ project = ProjectFactory.create(name="My Test Project") dependency = DependencyFactory.create() ProjectDependency.objects.create( project=project, dependency=dependency) from projects.helpers import build_project projectbuild = build_project(project, queue_build=False) build = BuildFactory.create( job=dependency.job, build_id=projectbuild.build_id) artifact = ArtifactFactory.create( filename="thing.txt", build=build) policy = CdimageArchivePolicy() self.assertEqual( "%s/%s/thing.txt" % ("my-test-project", build.build_id), policy.get_path_for_artifact(artifact, projectbuild=projectbuild))
def test_get_path_for_artifact_with_dependency(self): """ If the cdimage archiver gets a dependency with no projectbuild, it should delegate to the DefaultPolicy to generate the archive filename. """ project = ProjectFactory.create(name="My Test Project") dependency = DependencyFactory.create() ProjectDependency.objects.create( project=project, dependency=dependency) from projects.helpers import build_project projectbuild = build_project(project, queue_build=False) build = BuildFactory.create( job=dependency.job, build_id=projectbuild.build_id) artifact = ArtifactFactory.create( filename="thing.txt", build=build) policy = CdimageArchivePolicy() self.assertEqual( "%s/%s/thing.txt" % ("my-test-project", build.build_id), policy.get_path_for_artifact( artifact, dependency=dependency, projectbuild=projectbuild))
def test_get_path_for_artifact_with_dependency(self): """ If the cdimage archiver gets a dependency with no projectbuild, it should delegate to the DefaultPolicy to generate the archive filename. """ project = ProjectFactory.create(name="My Test Project") dependency = DependencyFactory.create() ProjectDependency.objects.create(project=project, dependency=dependency) from projects.helpers import build_project projectbuild = build_project(project, queue_build=False) build = BuildFactory.create(job=dependency.job, build_id=projectbuild.build_id) artifact = ArtifactFactory.create(filename="thing.txt", build=build) policy = CdimageArchivePolicy() self.assertEqual( "%s/%s/thing.txt" % ("my-test-project", build.build_id), policy.get_path_for_artifact(artifact, dependency=dependency, projectbuild=projectbuild))
def test_get_path_for_artifact(self): """ The CdimageArchivePolicy should calculate a cdimage-like path using the project name and build id. url. """ project = ProjectFactory.create(name="My Test Project") dependency = DependencyFactory.create() ProjectDependency.objects.create(project=project, dependency=dependency) from projects.helpers import build_project projectbuild = build_project(project, queue_build=False) build = BuildFactory.create(job=dependency.job, build_id=projectbuild.build_id) artifact = ArtifactFactory.create(filename="thing.txt", build=build) policy = CdimageArchivePolicy() self.assertEqual( "%s/%s/thing.txt" % ("my-test-project", build.build_id), policy.get_path_for_artifact(artifact, dependency=dependency, projectbuild=projectbuild))