class UpdateVersionInformationTest(TestCase): def setUp(self): self.repo1 = Repository.objects.create( name='repo1', shorthand='repo1') self.package = create_source_package({ 'name': 'dummy-package', 'version': '1.0.0', }) self.repo1.add_source_package(self.package) self.update = UpdateVersionInformation() def test_extract_versions_if_no_hidden_flag(self): versions = self.update._extract_versions_for_package( self.package.source_package_name) self.assertTrue(versions['version_list']) def test_extract_versions_if_hidden_flag_false(self): RepositoryFlag.objects.create(repository=self.repo1, name='hidden', value=False) versions = self.update._extract_versions_for_package( self.package.source_package_name) self.assertTrue(versions['version_list']) def test_extract_versions_if_hidden_flag_true(self): RepositoryFlag.objects.create(repository=self.repo1, name='hidden', value=True) versions = self.update._extract_versions_for_package( self.package.source_package_name) self.assertFalse(versions['version_list'])
def setUp(self): self.repo1 = Repository.objects.create( name='repo1', shorthand='repo1') self.package = create_source_package({ 'name': 'dummy-package', 'version': '1.0.0', }) self.repo1.add_source_package(self.package) self.update = UpdateVersionInformation()