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
|
#!/usr/bin/env python2
'''
Unit tests for oc scale
'''
# To run
# python -m unittest version
#
# .
# Ran 1 test in 0.597s
#
# OK
import os
import sys
import unittest
import mock
# Removing invalid variable names for tests so that I can
# keep them brief
# pylint: disable=invalid-name,no-name-in-module
# Disable import-error b/c our libraries aren't loaded in jenkins
# pylint: disable=import-error
# place class in our python path
module_path = os.path.join('/'.join(os.path.realpath(__file__).split('/')[:-4]), 'library') # noqa: E501
sys.path.insert(0, module_path)
from oc_scale import OCScale # noqa: E402
class OCScaleTest(unittest.TestCase):
'''
Test class for OCVersion
'''
def setUp(self):
''' setup method will create a file and set to known configuration '''
pass
@mock.patch('oc_scale.OCScale.openshift_cmd')
def test_state_list(self, mock_openshift_cmd):
''' Testing a get '''
params = {'name': 'router',
'namespace': 'default',
'replicas': 2,
'state': 'list',
'kind': 'dc',
'kubeconfig': '/etc/origin/master/admin.kubeconfig',
'debug': False}
dc = '''{"kind": "DeploymentConfig",
"apiVersion": "v1",
"metadata": {
"name": "router",
"namespace": "default",
"selfLink": "/oapi/v1/namespaces/default/deploymentconfigs/router",
"uid": "a441eedc-e1ae-11e6-a2d5-0e6967f34d42",
"resourceVersion": "6558",
"generation": 8,
"creationTimestamp": "2017-01-23T20:58:07Z",
"labels": {
"router": "router"
}
},
"spec": {
"replicas": 2,
}
}'''
mock_openshift_cmd.side_effect = [
{"cmd": '/usr/bin/oc get dc router -n default',
'results': dc,
'returncode': 0}]
results = OCScale.run_ansible(params, False)
self.assertFalse(results['changed'])
self.assertEqual(results['result'][0], 2)
@mock.patch('oc_scale.OCScale.openshift_cmd')
def test_scale(self, mock_openshift_cmd):
''' Testing a get '''
params = {'name': 'router',
'namespace': 'default',
'replicas': 3,
'state': 'list',
'kind': 'dc',
'kubeconfig': '/etc/origin/master/admin.kubeconfig',
'debug': False}
dc = '''{"kind": "DeploymentConfig",
"apiVersion": "v1",
"metadata": {
"name": "router",
"namespace": "default",
"selfLink": "/oapi/v1/namespaces/default/deploymentconfigs/router",
"uid": "a441eedc-e1ae-11e6-a2d5-0e6967f34d42",
"resourceVersion": "6558",
"generation": 8,
"creationTimestamp": "2017-01-23T20:58:07Z",
"labels": {
"router": "router"
}
},
"spec": {
"replicas": 3,
}
}'''
mock_openshift_cmd.side_effect = [
{"cmd": '/usr/bin/oc get dc router -n default',
'results': dc,
'returncode': 0},
{"cmd": '/usr/bin/oc create -f /tmp/router -n default',
'results': '',
'returncode': 0}
]
results = OCScale.run_ansible(params, False)
self.assertFalse(results['changed'])
self.assertEqual(results['result'][0], 3)
def tearDown(self):
'''TearDown method'''
pass
if __name__ == "__main__":
unittest.main()
|
