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;'