def setUp(self):
""" Set up the environnment, ran before every test. """
super(FlaskUiAdminTest, self).setUp()
mirrormanager2.app.APP.config['TESTING'] = True
mirrormanager2.app.SESSION = self.session
mirrormanager2.app.ADMIN.SESSION = self.session
mirrormanager2.app.APP.SESSION = self.session
mirrormanager2.admin.SESSION = self.session
mirrormanager2.admin.ADMIN.SESSION = self.session
mirrormanager2.admin.APP.SESSION = self.session
for view in mirrormanager2.admin.VIEWS:
view.session = self.session
self.app = mirrormanager2.app.APP.test_client()
# Fill the DB a little bit
tests.create_base_items(self.session)
tests.create_site(self.session)
tests.create_site_admin(self.session)
tests.create_hosts(self.session)
tests.create_location(self.session)
tests.create_netblockcountry(self.session)
tests.create_directory(self.session)
tests.create_category(self.session)
tests.create_hostcategory(self.session)
tests.create_hostcategoryurl(self.session)
tests.create_hostcategorydir(self.session)
tests.create_categorydirectory(self.session)
tests.create_version(self.session)
tests.create_repository(self.session)
tests.create_repositoryredirect(self.session)
def setUp(self):
""" Set up the environnment, ran before every test. """
super(FlaskUiAdminTest, self).setUp()
mirrormanager2.app.APP.config['TESTING'] = True
mirrormanager2.app.SESSION = self.session
mirrormanager2.app.ADMIN.SESSION = self.session
mirrormanager2.app.APP.SESSION = self.session
mirrormanager2.admin.SESSION = self.session
mirrormanager2.admin.ADMIN.SESSION = self.session
mirrormanager2.admin.APP.SESSION = self.session
for view in mirrormanager2.admin.VIEWS:
view.session = self.session
self.app = mirrormanager2.app.APP.test_client()
# Fill the DB a little bit
tests.create_base_items(self.session)
tests.create_site(self.session)
tests.create_site_admin(self.session)
tests.create_hosts(self.session)
tests.create_location(self.session)
tests.create_netblockcountry(self.session)
tests.create_directory(self.session)
tests.create_category(self.session)
tests.create_hostcategory(self.session)
tests.create_hostcategoryurl(self.session)
tests.create_hostcategorydir(self.session)
tests.create_categorydirectory(self.session)
tests.create_version(self.session)
tests.create_repository(self.session)
tests.create_repositoryredirect(self.session)
def test_get_reporedirect(self):
""" Test the get_reporedirect function of mirrormanager2.lib.
"""
results = mirrormanager2.lib.get_reporedirect(self.session)
self.assertEqual(results, [])
tests.create_repositoryredirect(self.session)
results = mirrormanager2.lib.get_reporedirect(self.session)
self.assertEqual(len(results), 3)
self.assertEqual(results[0].from_repo, 'fedora-rawhide')
self.assertEqual(results[0].to_repo, 'rawhide')
self.assertEqual(results[1].from_repo, 'fedora-install-rawhide')
self.assertEqual(results[1].to_repo, 'rawhide')
self.assertEqual(results[2].from_repo, 'epel-6.0')
self.assertEqual(results[2].to_repo, 'epel-6')
def test_get_reporedirect(self):
""" Test the get_reporedirect function of mirrormanager2.lib.
"""
results = mirrormanager2.lib.get_reporedirect(self.session)
self.assertEqual(results, [])
tests.create_repositoryredirect(self.session)
results = mirrormanager2.lib.get_reporedirect(self.session)
self.assertEqual(len(results), 3)
self.assertEqual(results[0].from_repo, 'fedora-rawhide')
self.assertEqual(results[0].to_repo, 'rawhide')
self.assertEqual(results[1].from_repo, 'fedora-install-rawhide')
self.assertEqual(results[1].to_repo, 'rawhide')
self.assertEqual(results[2].from_repo, 'epel-6.0')
self.assertEqual(results[2].to_repo, 'epel-6')
