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()
