def test_subscriber_is_notified_on_update():
data_source = MongoDataSource(User)
user = User(name='Anders', age=12)
user.save()
notified = {
'notified_data_source': None,
'notified_entity_id': None
}
def notify(data_source, entity):
notified['notified_data_source'] = data_source
notified['notified_entity_id'] = entity.id
data_source.subscribe(notify)
user.name = 'Emil'
user.save()
assert notified == {
'notified_data_source': data_source,
'notified_entity_id': user.id
}
data_source.disconnect()
def test_subscriber_is_notified_on_bulk_insert_with_load():
data_source = MongoDataSource(User)
user1 = User(name='Anders', age=12)
user2 = User(name='Henrik', age=50)
notified = {
'notified_data_sources': [],
'notified_entity_ids': []
}
def notify(data_source, entity):
notified['notified_data_sources'].append(data_source)
notified['notified_entity_ids'].append(str(entity.id))
data_source.subscribe(notify)
users = User.objects.insert([user1, user2])
user_ids = [str(user.id) for user in users]
assert notified == {
'notified_data_sources': [data_source, data_source],
'notified_entity_ids': user_ids
}
data_source.disconnect()
def test_finding_record():
data_source = MongoDataSource(User)
user = User(name='Anders', age=12)
user.save()
assert data_source.find(str(user.id)) == user
data_source.disconnect()
def test_serialization():
data_source = MongoDataSource(User)
user = User(name='Anders', age=12)
user.save()
assert data_source.serialize(user) == str(user.id)
data_source.disconnect()
def test_serialization():
data_source = MongoDataSource(User)
user = User(name='Anders', age=12)
user.save()
assert data_source.serialize(user) == str(user.id)
data_source.disconnect()
def test_finding_record():
data_source = MongoDataSource(User)
user = User(name='Anders', age=12)
user.save()
assert data_source.find(str(user.id)) == user
data_source.disconnect()
def test_subscriber_is_not_notified_when_bulk_inserting_without_load():
data_source = MongoDataSource(User)
user1 = User(name='Anders', age=12)
user2 = User(name='Henrik', age=50)
notified = False
def notify(data_source, entity):
notified = True # NOQA
data_source.subscribe(notify)
User.objects.insert([user1, user2], load_bulk=False)
assert notified == False
data_source.disconnect()
def test_subscriber_is_not_notified_when_bulk_inserting_without_load():
data_source = MongoDataSource(User)
user1 = User(name='Anders', age=12)
user2 = User(name='Henrik', age=50)
notified = False
def notify(data_source, entity):
notified = True # NOQA
data_source.subscribe(notify)
User.objects.insert([user1, user2], load_bulk=False)
assert notified == False
data_source.disconnect()
def test_subscriber_is_notified_on_delete():
data_source = MongoDataSource(User)
user = User(name='Anders', age=12)
user.save()
notified = {'notified_data_source': None, 'notified_entity_id': None}
def notify(data_source, entity):
notified['notified_data_source'] = data_source
notified['notified_entity_id'] = entity.id
data_source.subscribe(notify)
user.delete()
assert notified == {
'notified_data_source': data_source,
'notified_entity_id': user.id
}
data_source.disconnect()
def test_subscriber_is_notified_on_bulk_insert_with_load():
data_source = MongoDataSource(User)
user1 = User(name='Anders', age=12)
user2 = User(name='Henrik', age=50)
notified = {'notified_data_sources': [], 'notified_entity_ids': []}
def notify(data_source, entity):
notified['notified_data_sources'].append(data_source)
notified['notified_entity_ids'].append(str(entity.id))
data_source.subscribe(notify)
users = User.objects.insert([user1, user2])
user_ids = [str(user.id) for user in users]
assert notified == {
'notified_data_sources': [data_source, data_source],
'notified_entity_ids': user_ids
}
data_source.disconnect()
def _is_mongo_data_source(self, data_source):
return hasattr(data_source, 'document') and \
MongoDataSource.is_instance(data_source.document)
def _is_mongo_data_source(self, data_source):
return hasattr(data_source, "document") and MongoDataSource.is_instance(data_source.document)
def test_data_source_id():
data_source = MongoDataSource(User)
assert data_source.data_source_id == 'User'
data_source.disconnect()
def test_document():
data_source = MongoDataSource(User)
assert data_source.document == User
data_source.disconnect()
def test_data_source_for_entity_class():
data_source = MongoDataSource(User)
assert data_source.for_entity_class(User) == True
assert data_source.for_entity_class(Handin) == False
def test_data_source_for_entity():
data_source = MongoDataSource(User)
assert data_source.for_entity(User()) == True
assert data_source.for_entity(Handin()) == False
def test_document():
data_source = MongoDataSource(User)
assert data_source.document == User
data_source.disconnect()
def test_data_source_id():
data_source = MongoDataSource(User)
assert data_source.data_source_id == 'User'
data_source.disconnect()
def test_data_source_for_entity_class():
data_source = MongoDataSource(User)
assert data_source.for_entity_class(User) == True
assert data_source.for_entity_class(Handin) == False
def test_data_source_for_entity():
data_source = MongoDataSource(User)
assert data_source.for_entity(User()) == True
assert data_source.for_entity(Handin()) == False
