class TestDbUowCascadeOnRefresh(TestCase):
connection = Connection()
def setUp(self):
self.session = Session(0, self.connection['test_tori_db_uow_cascade_on_refresh'])
data_set = {
'regions': [
{'_id': 1, 'name': 'Asia'},
{'_id': 2, 'name': 'Europe'},
{'_id': 3, 'name': 'North America'}
],
'countries': [
{'_id': 1, 'region': 3, 'name': 'Canada'},
{'_id': 2, 'region': 2, 'name': 'England'},
{'_id': 3, 'region': 1, 'name': 'Japan'},
{'_id': 4, 'region': 1, 'name': 'Thailand'}
]
}
self.session.collection(Region)._api.remove()
for region in data_set['regions']:
self.session.collection(Region)._api.insert(region)
self.session.collection(Country)._api.remove()
for country in data_set['countries']:
self.session.collection(Country)._api.insert(country)
def test_cascade_from_owning_side(self):
japan = self.session.collection(Country).get(3)
self.assertEqual('Japan', japan.name)
self.assertEqual('Asia', japan.region.name)
# At this point, Region 1 is loaded into the memory.
# Bypass the identity map and then update the data manually.
self.session.collection(Region)._api.update({'_id': 1}, {'$set': {'name': 'Asia and Oceanic'}})
# Now, try to persist the data.
japan.name = u'日本'
self.session.persist(japan)
self.session.flush()
# Confirm that only the name of the country is updated
self.assertEqual(u'日本', japan.name)
self.assertEqual('Asia', japan.region.name)
# Refresh the entity
self.session.refresh(japan)
# Confirm that only the name of the region is updated after refreshing
self.assertEqual(u'日本', japan.name)
self.assertEqual('Asia and Oceanic', japan.region.name)
def _test_cascade_from_inverted_side(self):
europe = self.session.collection(Region).get(2)
self.assertEqual('Europe', europe.name)
self.assertEqual('England', europe.countries[0].name)
# At this point, Region 1 is loaded into the memory.
# Bypass the identity map and then update the data manually.
self.session.collection(Region)._api.update({'_id': 2}, {'$set': {'name': 'United Kingdom of Great Britain and Ireland'}})
# Now, try to persist the data.
europe.name = 'Europian Union'
self.session.persist(europe)
self.session.flush()
# Confirm that only the name of the country is updated
self.assertEqual('Europian Union', europe.name)
self.assertEqual('England', europe.countries[0].name)
# Refresh the entity
self.session.refresh(europe)
# Confirm that refreshing doesn't work with reverse-mapping properties.
self.assertEqual('Europian Union', europe.name)
self.assertEqual('England', europe.countries[0].name)
