def test_layer_bulk_insert():
"""Writing multiple documents should save correctly. They can be
modified"""
dml = DMLayers()
layers = [{
'111-22': [],
'111-22-a': [],
'doc_id': 'verver/111-22'
}, {
'111-23': [],
'doc_id': 'verver/111-23'
}]
dml.bulk_insert(layers, 'name', 'cfr')
assert Layer.objects.count() == 2
assert dml.get('name', 'cfr', 'verver/111-22') == {
'111-22': [],
'111-22-a': []
}
assert dml.get('name', 'cfr', 'verver/111-23') == {'111-23': []}
layers[1] = {'111-23': [1], 'doc_id': 'verver/111-23'}
dml.bulk_delete('name', 'cfr', 'verver/111')
dml.bulk_insert(layers, 'name', 'cfr')
assert Layer.objects.count() == 2
assert dml.get('name', 'cfr', 'verver/111-23') == {'111-23': [1]}
class DMLayersTest(TestCase):
def setUp(self):
self.dml = DMLayers()
def test_get_404(self):
self.assertIsNone(self.dml.get('namnam', 'cfr', 'verver/lablab'))
def test_get_success(self):
Layer(name='namnam',
doc_type='cfr',
doc_id='verver/lablab',
layer={
"some": "body"
}).save()
self.assertEqual({"some": 'body'},
self.dml.get('namnam', 'cfr', 'verver/lablab'))
def test_bulk_insert(self):
"""Writing multiple documents should save correctly. They can be
modified"""
layers = [{
'111-22': [],
'111-22-a': [],
'doc_id': 'verver/111-22'
}, {
'111-23': [],
'doc_id': 'verver/111-23'
}]
self.dml.bulk_insert(layers, 'name', 'cfr')
self.assertEqual(Layer.objects.count(), 2)
self.assertEqual(self.dml.get('name', 'cfr', 'verver/111-22'), {
'111-22': [],
'111-22-a': []
})
self.assertEqual(self.dml.get('name', 'cfr', 'verver/111-23'),
{'111-23': []})
layers[1] = {'111-23': [1], 'doc_id': 'verver/111-23'}
self.dml.bulk_delete('name', 'cfr', 'verver/111')
self.dml.bulk_insert(layers, 'name', 'cfr')
self.assertEqual(Layer.objects.count(), 2)
self.assertEqual(self.dml.get('name', 'cfr', 'verver/111-23'),
{'111-23': [1]})
def test_layer_bulk_insert():
"""Writing multiple documents should save correctly. They can be
modified"""
dml = DMLayers()
layers = [{'111-22': [], '111-22-a': [], 'doc_id': 'verver/111-22'},
{'111-23': [], 'doc_id': 'verver/111-23'}]
dml.bulk_insert(layers, 'name', 'cfr')
assert Layer.objects.count() == 2
assert dml.get('name', 'cfr', 'verver/111-22') == {'111-22': [],
'111-22-a': []}
assert dml.get('name', 'cfr', 'verver/111-23') == {'111-23': []}
layers[1] = {'111-23': [1], 'doc_id': 'verver/111-23'}
dml.bulk_delete('name', 'cfr', 'verver/111')
dml.bulk_insert(layers, 'name', 'cfr')
assert Layer.objects.count() == 2
assert dml.get('name', 'cfr', 'verver/111-23') == {'111-23': [1]}
