def setUp(self):
super(self.__class__, self).setUp()
conn = DBConnection()
conn.alias = 'test_connection'
conn.engine = settings.DATABASES['default']['ENGINE']
conn.name = settings.DATABASES['default']['NAME']
conn.user = settings.DATABASES['default']['USER']
conn.password = settings.DATABASES['default']['PASSWORD']
conn.host = settings.DATABASES['default']['HOST']
conn.port = settings.DATABASES['default']['PORT']
conn.save()
layer = DataBaseLayer()
layer.db_connection = conn
layer.slug = 'tests_location'
layer.name = 'tests_location'
layer.table = 'tests_location'
layer.pk_field = 'code'
layer.geom_field = 'geometry'
layer.anonymous_view = True
layer.anonymous_add = True
layer.anonymous_update = True
layer.anonymous_delete = True
layer.save()
self.layer = layer
def test_sequence_int(self):
sql = """
CREATE SEQUENCE tree_int_code_seq
START WITH 1
INCREMENT BY 1
NO MINVALUE
NO MAXVALUE
CACHE 1;
CREATE TABLE tree_int (
id SERIAL NOT NULL,
code integer NOT NULL
default (nextval('tree_int_code_seq'::regclass)),
name character varying(12) default 'Hell")o',
description character varying(250)
);
"""
with self.conn.get_connection().cursor() as cursor:
cursor.execute(sql)
self.sql_to_delete = [
'DROP TABLE tree_int;',
'DROP SEQUENCE tree_int_code_seq;',
]
layer = DataBaseLayer()
layer.db_connection = self.conn
layer.slug = 'tree_int'
layer.name = 'tree_int'
layer.table = 'tree_int'
layer.pk_field = 'id'
layer.anonymous_view = True
layer.anonymous_add = True
layer.anonymous_update = True
layer.anonymous_delete = True
layer.save()
self.layer = layer
field = self.layer.fields.filter(name='code').first()
field.blank = True
field.save()
Tree = create_dblayer_model(self.layer)
obj = Tree()
obj.description = 'test 1'
obj.save()
self.assertEqual(Tree.objects.last().code, 1)
data = {
'ADD': [
{
'description': 'test 2'
},
],
'UPDATE': [],
'DELETE': []
}
url = reverse('content-bulk', kwargs={'name': self.layer.name})
response = self.client.post(url, data, format='json')
# print(response.json())
self.assertEqual(response.status_code, 200)
self.assertEqual(Tree.objects.last().code, 2)