class Test_users_manager_get(unittest.TestCase): def setUp(self): init_DB(status="development", folder=db_path) self.assertTrue(os.path.exists(db_path)) self.db_users_manager = UsersController(db_path) self.assertIsNotNone(self.db_users_manager) load_env_vars() def tearDown(self): self.db_users_manager = None shutil.rmtree(db_path) self.assertFalse(os.path.exists(db_path)) def test_get_user_existing(self): user = self.db_users_manager.get_user("Mathieu") self.assertTrue("Mathieu", user) def test_get_user_non_existing(self): user = self.db_users_manager.get_user("Mathie2u") self.assertIsNone(user) def test_get_all_users(self): existing_users = [(u'David Lawrence'), (u'Mathieu'), (u'Nuno')] all_users = self.db_users_manager.get_users() for usr in all_users: self.assertIn(usr, existing_users)
class Test_users_manager_user_add(unittest.TestCase): def setUp(self): init_DB(status="development", folder=db_path) self.assertTrue(os.path.exists(db_path)) self.db_users_manager = UsersController(db_path) self.assertIsNotNone(self.db_users_manager) load_env_vars() def tearDown(self): self.db_users_manager = None shutil.rmtree(db_path) self.assertFalse(os.path.exists(db_path)) def test_add_unexistant(self): user_to_add = "Jim" has_been_added = self.db_users_manager.add_user_if_unexistant( user_to_add) self.assertTrue(has_been_added) self.assertEqual( self.db_users_manager.get_user(user_to_add), user_to_add) def test_not_add_existant(self): user_to_add = "Mathieu" has_been_added = self.db_users_manager.add_user_if_unexistant( user_to_add) self.assertFalse(has_been_added)