def test_cannot_add_unamed_catalog(self):
repo = CatalogRepo(TEST_SESSION_FACTORY)
user_repo = UserRepo(TEST_SESSION_FACTORY)
owner = User(username="******", email="bob@bob")
user_repo.add_user(owner)
catalog = Catalog(owner_id=owner.id)
with self.assertRaises(SQLAlchemyError):
repo.add(catalog)
def test_can_get_catalog_by_id_when_catalog_with_owner_and_name_added(self):
repo = CatalogRepo(TEST_SESSION_FACTORY)
user_repo = UserRepo(TEST_SESSION_FACTORY)
owner = User(username="******", email="bob@bob")
user_repo.add_user(owner)
catalog = Catalog(owner_id=owner.id, name="Bob's catalog.")
repo.add(catalog)
actual_catalog = repo.get_by_id(catalog.id)
self.assertEqual(owner.id, actual_catalog.owner_id)
self.assertEqual(catalog.name, actual_catalog.name)
self.assertIsNone(actual_catalog.description)
self.assertIsNotNone(actual_catalog.created_at_utc)
def test_get_all_returns_all_when_catalog_contains_one_entry(self):
repo = CatalogRepo(TEST_SESSION_FACTORY)
user_repo = UserRepo(TEST_SESSION_FACTORY)
owner = User(username="******", email="bob@bob")
user_repo.add_user(owner)
catalog = Catalog(owner_id=owner.id, name="Bob's catalog.")
repo.add(catalog)
catalogs = repo.get_all()
self.assertEqual(1, len(catalogs))
self.assertEqual(owner.id, catalogs[0].owner_id)
self.assertEqual(catalog.name, catalogs[0].name)
self.assertIsNone(catalogs[0].description)
self.assertIsNotNone(catalogs[0].created_at_utc)
def test_get_all_returns_all_when_catalog_contains_more_than_one_entry(self):
repo = CatalogRepo(TEST_SESSION_FACTORY)
user_repo = UserRepo(TEST_SESSION_FACTORY)
owner = User(username="******", email="bob@bob")
user_repo.add_user(owner)
expected_catalogs = []
expected_catalogs.append(Catalog(owner_id=owner.id, name="A"))
expected_catalogs.append(Catalog(owner_id=owner.id, name="B"))
expected_catalogs.append(Catalog(owner_id=owner.id, name="C"))
for catalog in expected_catalogs:
repo.add(catalog)
actual_catalogs = repo.get_all()
self.assertEqual(sorted(expected_catalogs), sorted(actual_catalogs))
def test_can_add_two_catalogs_with_same_name_and_different_owner(self):
repo = CatalogRepo(TEST_SESSION_FACTORY)
user_repo = UserRepo(TEST_SESSION_FACTORY)
bob = User(username="******", email="bob@bob")
user_repo.add_user(bob)
tony = User(username="******", email="tony@tony")
user_repo.add_user(tony)
expected_catalogs = []
catalog = Catalog(owner_id=bob.id, name="A")
expected_catalogs.append(catalog)
repo.add(catalog)
catalog = Catalog(owner_id=tony.id, name="A")
repo.add(catalog)
expected_catalogs.append(catalog)
self.assertEqual(sorted(expected_catalogs), sorted(repo.get_all()))
def setUp(self):
"""Reset the database before each test is run"""
CategoryRepoTest._HELPER.reset_database()
# add a test owners
user_repo = UserRepo(TEST_SESSION_FACTORY)
self._test_owner = User(username="******", email="bob@bob")
self._test_owner2 = User(username="******", email="tony@tony")
user_repo.add_user(self._test_owner)
user_repo.add_user(self._test_owner2)
# add a test catalogs
catalogRepo = CatalogRepo(TEST_SESSION_FACTORY)
self._test_catalog = Catalog(owner_id=self._test_owner.id,
name="catalog")
self._test_catalog2 = Catalog(owner_id=self._test_owner2.id,
name="catalog two")
catalogRepo.add(self._test_catalog)
catalogRepo.add(self._test_catalog2)
def test_cannot_add_unowned_catalog(self):
repo = CatalogRepo(TEST_SESSION_FACTORY)
catalog = Catalog(owner_id=None, name="catalog")
with self.assertRaises(SQLAlchemyError):
repo.add(catalog)