def test_list(self):
qs = models.QuerySnippet(
trigger="x",
description="y",
snippet="z",
user=self.factory.user,
org=self.factory.org,
)
models.db.session.add(qs)
models.db.session.commit()
res = self.make_request("get",
"/api/query_snippets",
user=self.factory.user)
self.assertEqual(res.status_code, 200)
data = res.json
self.assertEqual(len(data), 1)
self.assertEqual(
project(data[0], ["id", "trigger", "description", "snippet"]),
{
"id": 1,
"trigger": "x",
"description": "y",
"snippet": "z"
},
)
self.assertEqual(qs.trigger, "x")
self.assertEqual(qs.description, "y")
self.assertEqual(qs.snippet, "z")
def test_edit(self):
qs = models.QuerySnippet(trigger='a',
description='b',
snippet='c',
user=self.factory.user,
org=self.factory.org)
models.db.session.add(qs)
models.db.session.commit()
res = self.make_request('post',
'/api/query_snippets/1',
data={
'trigger': 'x',
'description': 'y',
'snippet': 'z'
},
user=self.factory.user)
self.assertEqual(
project(res.json, ['id', 'trigger', 'description', 'snippet']), {
'id': 1,
'trigger': 'x',
'description': 'y',
'snippet': 'z',
})
self.assertEqual(qs.trigger, 'x')
self.assertEqual(qs.description, 'y')
self.assertEqual(qs.snippet, 'z')
def test_edit(self):
qs = models.QuerySnippet(
trigger="a",
description="b",
snippet="c",
user=self.factory.user,
org=self.factory.org,
)
models.db.session.add(qs)
models.db.session.commit()
res = self.make_request(
"post",
"/api/query_snippets/1",
data={
"trigger": "x",
"description": "y",
"snippet": "z"
},
user=self.factory.user,
)
self.assertEqual(
project(res.json, ["id", "trigger", "description", "snippet"]),
{
"id": 1,
"trigger": "x",
"description": "y",
"snippet": "z"
},
)
self.assertEqual(qs.trigger, "x")
self.assertEqual(qs.description, "y")
self.assertEqual(qs.snippet, "z")
def test_list(self):
qs = models.QuerySnippet(
trigger='x',
description='y',
snippet='z',
user=self.factory.user,
org=self.factory.org
)
models.db.session.add(qs)
models.db.session.commit()
res = self.make_request(
'get',
'/api/query_snippets',
user=self.factory.user)
self.assertEqual(res.status_code, 200)
data = res.json
self.assertEqual(len(data), 1)
self.assertEqual(
project(data[0], ['id', 'trigger', 'description', 'snippet']), {
'id': 1,
'trigger': 'x',
'description': 'y',
'snippet': 'z',
})
self.assertEqual(qs.trigger, 'x')
self.assertEqual(qs.description, 'y')
self.assertEqual(qs.snippet, 'z')
def test_delete(self):
qs = models.QuerySnippet(trigger='a',
description='b',
snippet='c',
user=self.factory.user,
org=self.factory.org)
models.db.session.add(qs)
models.db.session.commit()
self.make_request('delete',
'/api/query_snippets/1',
user=self.factory.user)
self.assertEqual(models.QuerySnippet.query.count(), 0)
def test_delete(self):
qs = models.QuerySnippet(
trigger="a",
description="b",
snippet="c",
user=self.factory.user,
org=self.factory.org,
)
models.db.session.add(qs)
models.db.session.commit()
self.make_request("delete", "/api/query_snippets/1", user=self.factory.user)
self.assertEqual(models.QuerySnippet.query.count(), 0)
def post(self):
req = request.get_json(True)
require_fields(req, ('trigger', 'description', 'snippet'))
snippet = models.QuerySnippet(trigger=req['trigger'],
description=req['description'],
snippet=req['snippet'],
user=self.current_user,
org=self.current_org)
models.db.session.add(snippet)
self.record_event({
'action': 'create',
'object_id': snippet.id,
'object_type': 'query_snippet'
})
models.db.session.commit()
return snippet.to_dict()
def post(self):
req = request.get_json(True)
require_fields(req, ("trigger", "description", "snippet"))
snippet = models.QuerySnippet(
trigger=req["trigger"],
description=req["description"],
snippet=req["snippet"],
user=self.current_user,
org=self.current_org,
)
models.db.session.add(snippet)
models.db.session.commit()
self.record_event({
"action": "create",
"object_id": snippet.id,
"object_type": "query_snippet",
})
return snippet.to_dict()
