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]}