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_siteadmin(self): """ Test the get_siteadmin function of mirrormanager2.lib. """ results = mirrormanager2.lib.get_siteadmin(self.session, 1) self.assertEqual(results, None) tests.create_site(self.session) results = mirrormanager2.lib.get_siteadmin(self.session, 1) self.assertEqual(results, None) tests.create_site_admin(self.session) results = mirrormanager2.lib.get_siteadmin(self.session, 1) self.assertEqual(results.site.name, 'test-mirror') self.assertEqual(results.username, 'ralph') results = mirrormanager2.lib.get_siteadmin(self.session, 4) self.assertEqual(results.site.name, 'test-mirror2') self.assertEqual(results.username, 'pingou')