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)
