def test_remove(self):
def handle_remove(response, error=None):
self.assertIsNone(error)
self.assertEqual(response[0]['ok'], 1.0)
self.stop()
self.sync_db[self.collection_name].insert({"num": 1})
model = DBLayer(self.async_db, self.collection_name)
model.remove({"num": 1}, callback=handle_remove)
self.wait()
def test_remove(self):
def handle_remove(response, error=None):
self.assertIsNone(error)
self.assertEqual(response[0]['ok'], 1.0)
self.stop()
self.sync_db[self.collection_name].insert({"num": 1})
model = DBLayer(self.async_db, self.collection_name)
model.remove({"num": 1}, callback=handle_remove)
self.wait()
def test_delete(self):
def handle_delete(response, error=None):
self.assertIsNone(error)
self.assertEqual(response[0]["ok"], 1.0)
self.stop()
self.sync_db[self.collection_name].insert({"two": 2})
model = DBLayer(self.async_db, self.collection_name)
model.remove({"two": 2}, callback=handle_delete)
self.wait()
def test_delete(self):
response = [{u'connectionId': 1, u'ok': 1.0, u'err': None, u'n': 0}]
query = {"id": "1", "two": 2}
collection_name = "collection_remove"
collection = Mock(name=collection_name)
collection.remove.return_value = None
collection.remove.side_effect = lambda *args, **kwargs: kwargs['callback'](response, error=None)
client = {collection_name: collection}
callback = Mock(name="remove_callback")
callback.side_effect = lambda response, error : self.stop()
# Act
model = DBLayer(client, collection_name)
model.remove(query, callback=callback)
self.wait()
# Assert
collection.remove.assert_called_once_with(query, callback=callback)
callback.assert_called_once_with(response, error=None)
def test_delete(self):
response = [{u'connectionId': 1, u'ok': 1.0, u'err': None, u'n': 0}]
query = {"id": "1", "two": 2}
collection_name = "collection_remove"
collection = Mock(name=collection_name)
collection.remove.return_value = None
collection.remove.side_effect = lambda *args, **kwargs: kwargs[
'callback'](response, error=None)
client = {collection_name: collection}
callback = Mock(name="remove_callback")
callback.side_effect = lambda response, error: self.stop()
# Act
model = DBLayer(client, collection_name)
model.remove(query, callback=callback)
self.wait()
# Assert
collection.remove.assert_called_once_with(query, callback=callback)
callback.assert_called_once_with(response, error=None)
