def test_create_retrieve_company(self):
"""
Should be able to retrieve a company after creating it.
"""
with transaction():
company = Company(
name="name",
type=CompanyType.private,
).create()
retrieved_company = Company.retrieve(company.id)
assert_that(retrieved_company.name, is_(equal_to("name")))
assert_that(retrieved_company.type, is_(equal_to(CompanyType.private)))
def test_create_retrieve_company(self):
"""
Should be able to retrieve a company after creating it.
"""
with transaction():
company = Company(
name="name",
type=CompanyType.private,
).create()
retrieved_company = Company.retrieve(company.id)
assert_that(retrieved_company.name, is_(equal_to("name")))
assert_that(retrieved_company.type, is_(equal_to(CompanyType.private)))
def test_create_update_company(self):
"""
Should be able to update a company after creating it.
"""
with transaction():
company = Company(name="name", ).create()
with transaction():
updated_company = Company(
id=company.id,
name="new_name",
).update()
assert_that(updated_company.name, is_(equal_to("new_name")))
with transaction():
retrieved_company = Company.retrieve(company.id)
assert_that(retrieved_company.name, is_(equal_to("new_name")))
def test_create_update_with_diff_company(self):
"""
Should be able to update a company after creating it and get a diff.
"""
with transaction():
company = Company(name="name").create()
with transaction():
_, diff = Company(
id=company.id,
name="new_name",
).update_with_diff()
assert_that(list(diff.keys()), contains_inanyorder("name", "updated_at"))
assert_that(diff["name"].before, is_(equal_to("name")))
assert_that(diff["name"].after, is_(equal_to("new_name")))
with transaction():
retrieved_company = Company.retrieve(company.id)
assert_that(retrieved_company.name, is_(equal_to("new_name")))
def test_create_update_company(self):
"""
Should be able to update a company after creating it.
"""
with transaction():
company = Company(
name="name",
).create()
with transaction():
updated_company = Company(
id=company.id,
name="new_name",
).update()
assert_that(updated_company.name, is_(equal_to("new_name")))
with transaction():
retrieved_company = Company.retrieve(company.id)
assert_that(retrieved_company.name, is_(equal_to("new_name")))
def test_create_update_with_diff_company(self):
"""
Should be able to update a company after creating it and get a diff.
"""
with transaction():
company = Company(name="name").create()
with transaction():
_, diff = Company(
id=company.id,
name="new_name",
).update_with_diff()
assert_that(list(diff.keys()),
contains_inanyorder("name", "updated_at"))
assert_that(diff["name"].before, is_(equal_to("name")))
assert_that(diff["name"].after, is_(equal_to("new_name")))
with transaction():
retrieved_company = Company.retrieve(company.id)
assert_that(retrieved_company.name, is_(equal_to("new_name")))
def _retrieve_company(self, company_id):
with SessionContext(self.graph):
return Company.retrieve(company_id)
def _retrieve_company(self, company_id):
with SessionContext(self.graph):
return Company.retrieve(company_id)