|
1 import json,os |
|
2 |
|
3 from django.test import TestCase |
|
4 from renkanmanager.models import content_copy |
|
5 from django.contrib.auth import get_user_model |
|
6 |
|
7 User = get_user_model() |
|
8 |
|
9 class ModelsTest(TestCase): |
|
10 |
|
11 def setUp(self): |
|
12 test_json_path = os.path.join(os.path.dirname(__file__), 'test_revision_copy.json') |
|
13 with open(test_json_path) as json_file: |
|
14 self.test_json = json_file.read() |
|
15 |
|
16 def test_content_copy_call(self): |
|
17 c_copy = content_copy(self.test_json) |
|
18 self.assertIsNotNone(c_copy) |
|
19 |
|
20 def test_content_copy_id_key(self): |
|
21 c_copy = content_copy(self.test_json) |
|
22 self.assertIn('id', c_copy) |
|
23 self.assertNotIn('_id', c_copy) |
|
24 for n in c_copy['nodes']: |
|
25 self.assertIn('id', n) |
|
26 self.assertNotIn('_id', n) |
|
27 for u in c_copy['users']: |
|
28 self.assertIn('id', u) |
|
29 self.assertNotIn('_id', u) |
|
30 for v in c_copy['views']: |
|
31 self.assertIn('id', v) |
|
32 self.assertNotIn('_id', v) |
|
33 for e in c_copy['edges']: |
|
34 self.assertIn('id', e) |
|
35 self.assertNotIn('_id', e) |
|
36 |
|
37 def test_content_copy_content(self): |
|
38 c_copy = content_copy(self.test_json) |
|
39 c_json = json.loads(self.test_json) |
|
40 |
|
41 for k in (c_copy.keys() - ['nodes', 'users', 'views', 'edges', 'id']): |
|
42 self.assertEquals(c_copy[k], c_json[k]) |
|
43 |
|
44 self.assertEquals(len(c_copy['nodes']), len(c_json['nodes'])) |
|
45 for node in c_copy['nodes']: |
|
46 original_nodes = [ n for n in c_json['nodes'] if n['title'] == node['title'] ] |
|
47 self.assertEquals(1,len(original_nodes)) |
|
48 for k in (node.keys() - ['id']): |
|
49 self.assertEquals(node[k], original_nodes[0][k]) |
|
50 |
|
51 self.assertEquals(len(c_copy['edges']), len(c_json['edges'])) |
|
52 for edge in c_copy['edges']: |
|
53 original_edges = [ e for e in c_json['edges'] if e['title'] == edge['title'] ] |
|
54 self.assertEquals(1,len(original_edges)) |
|
55 orig_edge = original_edges[0] |
|
56 for k in (edge.keys() - ['id', 'from', 'to']): |
|
57 self.assertEquals(edge[k], orig_edge[k]) |
|
58 from_copy = [e for e in c_copy['nodes'] if e['id'] == edge['from']][0] |
|
59 from_orig = [e for e in c_json['nodes'] if e['_id'] == orig_edge['from']][0] |
|
60 self.assertEquals(from_copy['title'], from_orig['title']) |
|
61 |
|
62 to_copy = [e for e in c_copy['nodes'] if e['id'] == edge['to']][0] |
|
63 to_orig = [e for e in c_json['nodes'] if e['_id'] == orig_edge['to']][0] |
|
64 self.assertEquals(to_copy['title'], to_orig['title']) |
|
65 |
|
66 self.assertEquals(len(c_copy['users']), len(c_json['users'])) |
|
67 for user in c_copy['users']: |
|
68 original_users = [ u for u in c_json['users'] if u['title'] == user['title'] ] |
|
69 self.assertEquals(1,len(original_users)) |
|
70 for k in (user.keys() - ['id']): |
|
71 self.assertEquals(user[k], original_users[0][k]) |
|
72 |
|
73 self.assertEquals(len(c_copy['views']), len(c_json['views'])) |
|
74 for view in c_copy['views']: |
|
75 original_views = [ v for v in c_json['views'] if v['title'] == view['title'] ] |
|
76 self.assertEquals(1,len(original_views)) |
|
77 orig_view = original_views[0] |
|
78 for k in (view.keys() - ['id', 'hidden_nodes']): |
|
79 self.assertEquals(view[k], orig_view[k]) |
|
80 self.assertEquals(len(view['hidden_nodes']), len(orig_view['hidden_nodes'])) |
|
81 copy_hidden_nodes_titles = sorted([ [n for n in c_copy['nodes'] if n['id'] == n_id][0]['title'] for n_id in view['hidden_nodes']]) |
|
82 orig_hidden_nodes_titles = sorted([ [n for n in c_json['nodes'] if n['_id'] == n_id][0]['title'] for n_id in orig_view['hidden_nodes']]) |
|
83 |
|
84 self.assertEquals(copy_hidden_nodes_titles, orig_hidden_nodes_titles) |