def test_models(self):
new_ticket = Ticket.from_json(self.ticket_json)
created_ticket = self.db.add_ticket(new_ticket)
db_ticket = Ticket.from_db(self.db.get_ticket(created_ticket.id))
ticket_id = created_ticket.id
self.assertEqual(created_ticket.created_at, db_ticket.created_at)
db_ticket.state = TicketState.get_next_state(db_ticket.state)
self.db.save_ticket(db_ticket)
updated_ticket = Ticket.from_db(self.db.get_ticket(db_ticket.id))
self.assertEqual(db_ticket.state, updated_ticket.state)
self.assertEqual(TicketState.string_state(updated_ticket.state), 'отвечен')
for ticket in self.db.get_tickets():
self.assertTrue(isinstance(ticket, Ticket))
new_comment = Comment.from_json(self.comment_json % ticket_id)
created_comment = self.db.add_comment(new_comment)
db_comment = Comment.from_db(self.db.get_comment(created_comment.id))
self.assertEqual(created_comment.created_at, db_comment.created_at)
self.db.delete_comments()
for item in range(10):
self.db.add_comment(new_comment)
comments_count = self.db.get_comments_count(ticket_id)
for comment in self.db.get_comments(ticket_id):
self.assertTrue(isinstance(comment, Comment))
self.assertEqual(comments_count, 10)
self.db.delete_comments()
self.db.delete_tickets()
def new_ticket_comment(id):
try:
ticket = Ticket.from_db(g.db.get_ticket(id))
except ModelNotFound as e:
return not_found(str(e))
else:
if ticket.state == TicketState.CLOSED:
return ticket_closed(ticket_id=id)
else:
comment = Comment.from_json(request.get_json(), ticket_id=id)
g.db.add_comment(comment)
return jsonify(comment.get_attrs()), 201, {
'Location':
url_for('api.get_comment', id=comment.id, _external=True)
}