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
|
# vim: set fileencoding=utf-8 :
import responses
import unittest
from foreman_ansible_inventory import ForemanInventory
class TestGetJson(unittest.TestCase):
def setUp(self):
self.inv = ForemanInventory()
self.inv.foreman_url = 'http://localhost:3000'
self.inv.foreman_user = 'doesnot'
self.inv.foreman_pw = 'mastter'
self.inv.foreman_ssl_verify = True
@responses.activate
def test_get_hosts(self):
url = 'http://localhost:3000/api/v2/hosts'
responses.add(responses.GET,
url,
json={'results': [{'name': 'foo'},
{'name': 'bar'}],
'total': 4},
status=200)
ret = self.inv._get_hosts()
self.assertEqual(sorted(ret),
sorted([{u'name': u'foo'},
{u'name': u'bar'},
{u'name': u'foo'},
{u'name': u'bar'}]))
self.assertEqual(len(responses.calls), 2)
self.assertEqual(responses.calls[0].request.url,
'%s?per_page=250&page=1' % url)
self.assertEqual(responses.calls[1].request.url,
'%s?per_page=250&page=2' % url)
@responses.activate
def test_get_facts(self):
self.inv.want_facts = True
url = 'http://localhost:3000/api/v2/hosts/10/facts'
responses.add(responses.GET,
url,
json={'results': {'facts': {'fact1': 'val1',
'fact2': 'val2',
}
},
},
status=200)
ret = self.inv._get_facts({'id': 10})
self.assertEqual(ret, {u'fact2': u'val2', u'fact1': u'val1'})
self.assertEqual(len(responses.calls), 1)
self.assertEqual(responses.calls[0].request.url,
'%s?per_page=250&page=1' % url)
@responses.activate
def test_resolve_params(self):
url = 'http://localhost:3000/api/v2/hosts/10'
responses.add(responses.GET,
url,
json={'all_parameters':
[{'name': 'param1',
'value': 'value1'},
{'name': 'param2',
'value': 'value2'}]},
status=200)
ret = self.inv._resolve_params({'id': 10})
self.assertEqual(sorted(ret.items()),
sorted({'param1': 'value1',
'param2': 'value2'}.items()))
self.assertEqual(len(responses.calls), 1)
self.assertEqual(responses.calls[0].request.url,
'%s?per_page=250&page=1' % url)
|