def test_new_query_gets_created_by_logged_in_user(self, staff_user, staff_client):
query = SimpleQueryFactory.build(created_by_user=staff_user)
data = model_to_dict(query)
del data["id"]
staff_client.post(reverse("explorer:query_create"), {**data, "action": "save"})
query = Query.objects.first()
assert query.created_by_user_id == staff_user.id
def test_valid_query(self, staff_user, staff_client):
query = SimpleQueryFactory.build(sql="SELECT 1;", created_by_user=staff_user)
data = model_to_dict(query)
data["action"] = "save"
del data["id"]
del data["created_by_user"]
staff_client.post(reverse("explorer:query_create"), data)
assert Query.objects.all()[0].sql == "SELECT 1;"
def test_invalid_query_saved(self, staff_user, staff_client):
query = SimpleQueryFactory.build(sql='SELECT foo; DELETE FROM foo;',
created_by_user=staff_user)
data = model_to_dict(query)
data['action'] = "save"
del data['id']
del data['created_by_user']
staff_client.post(reverse("explorer:query_create"), data)
assert Query.objects.all()[0].sql == 'SELECT foo; DELETE FROM foo;'