fix schema and primary key

* version is an additional primary key so we need to indicate that
* schema no longer relies on Template model, and uses nested user
This commit is contained in:
Leo Hemsted
2016-08-08 16:57:39 +01:00
parent 049514d4b2
commit c820938ced
5 changed files with 18 additions and 19 deletions

View File

@@ -29,6 +29,7 @@ from app.models import (
VerifyCode,
ApiKey,
Template,
TemplateHistory,
Job,
Notification,
NotificationHistory,

View File

@@ -6,7 +6,7 @@ from app.dao.templates_dao import (
dao_update_template,
dao_get_template_versions)
from tests.app.conftest import sample_template as create_sample_template
from app.models import Template
from app.models import Template, TemplateHistory
import pytest
@@ -261,12 +261,16 @@ def test_get_template_versions(sample_template):
sample_template.content = 'new version'
dao_update_template(sample_template)
versions = dao_get_template_versions(service_id=sample_template.service_id, template_id=sample_template.id)
assert versions.__len__() == 2
for x in versions:
if x.version == 2:
assert x.content == 'new version'
else:
assert x.content == original_content
assert len(versions) == 2
versions = sorted(versions, key=lambda x: x.version)
assert versions[0].content == original_content
assert versions[1].content == 'new version'
assert versions[0].created_at == versions[1].created_at
assert versions[0].updated_at is None
assert versions[1].updated_at is not None
from app.schemas import template_history_schema
v = template_history_schema.load(versions, many=True)
assert v.__len__() == 2
assert len(v) == 2