-
Notifications
You must be signed in to change notification settings - Fork 0
/
tests_integration_failing.py
137 lines (114 loc) · 4.79 KB
/
tests_integration_failing.py
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
"""Integration Tests for the generic service discovery API
In this case they do not use a Mock, they access the real Consul K/V store
"""
import unittest
import time
import kvstore
import registry
URL = 'http://10.112.0.101:8500/v1/kv'
BASEDN = '__registrytests__'
with open('service-template.json') as jsontemplate:
TEMPLATE = jsontemplate.read()
with open('service-template.yaml') as yamltemplate:
TEMPLATEYAML = yamltemplate.read()
with open('options.json') as optionsfile:
OPTIONS = optionsfile.read()
class RegistryTemplatesTestCase(unittest.TestCase):
def setUp(self):
registry.connect(URL)
self.PREFIX = registry.TMPLPREFIX
self.servicename = "__unittests__"
self.start_time = time.time()
def tearDown(self):
registry._kv.delete('{}/{}'.format(self.PREFIX, self.servicename), recursive=True)
duration = time.time() - self.start_time
print '{} took {} seconds'.format(self.id(), duration)
def test_set_node_disks_one_disk(self):
cluster = self._add_sample_service_instance()
node = cluster.nodes[0]
expected = [{
'name': 'disk1',
'origin': '/data/1/instances-jlopez-cdh-5.7.0-1',
'destination': '/data/1',
'mode': 'rw'}, ]
#node.disks = expected
node.set_disks(expected)
disk = node.disks[0]
print 'DEBUG: node.disks: {}'.format(node.disks)
print 'DEBUG: disk: {}'.format(disk)
self.assertEqual(disk.origin, expected['origin'])
def _add_sample_service_instance(self):
servicename = self.servicename
version = "0.1.0"
description = "Unit test"
template = TEMPLATE
templateopts = OPTIONS
registry.register(servicename, version, description,
template, templateopts)
user = 'testuser'
options = {'slaves.number': 1}
cluster = registry.instantiate(user, servicename, version, options)
return cluster
def test_set_node_disks_two_disks(self):
basedn = BASEDN + '/cluster1/nodes/master0'
basedn_disks = basedn + '/disks'
expected = [
{
'name': 'disk1',
'origin': '/data/1/instances-jlopez-cdh-5.7.0-1',
'destination': '/data/1',
'mode': 'rw'
},
{
'name': 'disk2',
'origin': '/data/2/instances-jlopez-cdh-5.7.0-1',
'destination': '/data/2',
'mode': 'rw'
},
]
node = registry.Node(basedn)
node.disks = expected
self.assertEqual(node.disks, sorted(expected))
def test_register_new_cluster_instance_returns_dn(self):
nodes = REGISTRY['clusters']['cluster1']['nodes']
services = REGISTRY['clusters']['cluster1']['services']
dn = registry.instantiate(user='jlopez', product='a', version='1.0.0', nodes=nodes,
services=services)
expected = registry.PREFIX + '/jlopez/a/1.0.0/1'
self.assertEqual(dn, expected)
def test_register_two_cluster_instances(self):
nodes = REGISTRY['clusters']['cluster1']['nodes']
services = REGISTRY['clusters']['cluster1']['services']
dn = registry.instantiate(user='jlopez', product='a', version='1.0.0', nodes=nodes,
services=services)
expected = registry.PREFIX + '/jlopez/a/1.0.0/1'
self.assertEqual(dn, expected)
dn = registry.instantiate(user='jlopez', product='a', version='1.0.0', nodes=nodes,
services=services)
expected = registry.PREFIX + '/jlopez/a/1.0.0/2'
self.assertEqual(dn, expected)
def test_get_cluster_instance(self):
nodes = REGISTRY['clusters']['cluster1']['nodes']
services = REGISTRY['clusters']['cluster1']['services']
dn = registry.instantiate(user='jlopez', product='a', version='1.0.0', nodes=nodes,
services=services)
instance = registry.get_cluster(dn=dn)
expected_dn = registry.PREFIX + '/jlopez/a/1.0.0/1'
expected_nodes = [
registry.Node('{}/nodes/{}'.format(expected_dn, e)) for e in nodes
]
expected_services = [
registry.Service('{}/services/{}'.format(expected_dn, e)) for e in services
]
self.assertEqual(sorted(instance.nodes), sorted(expected_nodes))
self.assertEqual(sorted(instance.services), sorted(expected_services))
def test_get_node_networks(self):
raise NotImplemented
def test_set_node_networks(self):
raise NotImplemented
def test_get_node_tags(self):
raise NotImplemented
def test_set_node_tags(self):
raise NotImplemented
if __name__ == '__main__':
unittest.main()