1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2015 Étienne Loks <etienne.loks_AT_peacefrogsDOTnet>
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# See the file COPYING for details.
"""
Unit tests
"""
import json
from django.conf import settings
from django.core.exceptions import ValidationError
from django.test import TestCase
from archaeological_operations.tests import OperationInitTest
from archaeological_operations.models import Parcel
from archaeological_context_records import models
class ContextRecordInit(OperationInitTest):
def create_context_record(self, user=None, data={}):
if not getattr(self, 'context_records', None):
self.context_records = []
default = {'label':"Context record"}
if not data.get('operation'):
data['operation'] = self.get_default_operation()
if not data.get('parcel'):
data['parcel'] = self.get_default_parcel()
if not data.get('history_modifier'):
data['history_modifier'] = self.get_default_user()
default.update(data)
self.context_records.append(models.ContextRecord.objects.create(
**default))
return self.context_records
def get_default_context_record(self):
return self.create_context_record()[0]
class RecordRelationsTest(TestCase, ContextRecordInit):
fixtures = [settings.ROOT_PATH +
'../fixtures/initial_data.json',
settings.ROOT_PATH +
'../ishtar_common/fixtures/initial_data.json',
settings.ROOT_PATH +
'../archaeological_files/fixtures/initial_data.json',
settings.ROOT_PATH +
'../archaeological_operations/fixtures/initial_data-fr.json']
model = models.ContextRecord
def setUp(self):
# two different context records
self.create_context_record({"label":u"CR 1"})
self.create_context_record({"label":u"CR 2"})
def testRelations(self):
sym_rel_type = models.RelationType.objects.create(symmetrical=True,
txt_idx='sym')
rel_type_1 = models.RelationType.objects.create(symmetrical=False,
txt_idx='rel_1')
# cannot be symmetrical and have an inverse_relation
with self.assertRaises(ValidationError):
rel_test = models.RelationType.objects.create(symmetrical=True,
inverse_relation=rel_type_1, txt_idx='rel_3')
rel_test.full_clean()
# auto fill inverse relations
rel_type_2 = models.RelationType.objects.create(symmetrical=False,
inverse_relation=rel_type_1, txt_idx='rel_2')
self.assertEqual(rel_type_1.inverse_relation, rel_type_2)
cr_1 = self.context_records[0]
cr_2 = self.context_records[1]
# inserting a new symmetrical relation automatically creates the same
# relation for the second context record
rel = models.RecordRelations.objects.create(left_record=cr_1,
right_record=cr_2, relation_type=sym_rel_type)
self.assertEqual(models.RecordRelations.objects.filter(left_record=cr_2,
right_record=cr_1, relation_type=sym_rel_type).count(), 1)
# removing one symmetrical relation removes the other
rel.delete()
self.assertEqual(models.RecordRelations.objects.filter(left_record=cr_2,
right_record=cr_1, relation_type=sym_rel_type).count(), 0)
# for non-symmetrical relation, adding one relation automatically
# adds the inverse
rel = models.RecordRelations.objects.create(left_record=cr_1,
right_record=cr_2, relation_type=rel_type_1)
self.assertEqual(models.RecordRelations.objects.filter(left_record=cr_2,
right_record=cr_1, relation_type=rel_type_2).count(), 1)
|
