def test_render_GET_valid_parameters(self):
# Test data:
entries = [
{
'fqdn': 'blah',
'city': 'Somewheresville',
'country': 'US',
'ip': ['127.2.3.4', '::1'],
'port': 8421,
'site': 'mlab01',
'tool_extra': {
'collector_onion': 'testfakenotreal.onion',
},
},
{
'fqdn': 'blorp',
'city': 'Elsewhereton',
'country': 'US',
'ip': ['127.2.3.8', '::7'],
'port': 1248,
'site': 'mlab02',
'tool_extra': {
'collector_onion': 'yetanother.onion',
},
},
]
# Mocks:
db = dict((entry['fqdn'], entry) for entry in entries)
m_request = MagicMock()
# Fake a request with sufficient parameters:
m_request.args = {
'format': ['json'],
'policy': ['all'],
}
# Execute the code under test:
lsr = lookup.LookupSimulatorResource(db)
retval = lsr.render_GET(m_request)
# Verifications:
self.assertEqual(server.NOT_DONE_YET, retval)
# Verify that a 200 response was sent:
self.assertEqual(m_request.mock_calls,
[call.sendJsonResponse(entries)])
def test_render_PUT_valid_parameters(self):
self.m_request.content.read.return_value = self.entryjson
# Execute the code under test:
retval = self.ur.render_PUT(self.m_request)
# Verifications:
self.assertEqual(server.NOT_DONE_YET, retval)
# Verify that m_db now stores the expected entry:
self.assertEqual(
self.m_db.mock_calls,
[call.__setitem__(self.fqdn, self.expectedentry)])
# Verify that a 200 response was sent:
self.assertEqual(
self.m_request.mock_calls,
[call.content.read(),
call.sendJsonResponse('Ok.'),
])
def test_render_GET_valid_parameters(self):
# Test data:
entries = [
{
"fqdn": "blah",
"city": "Somewheresville",
"country": "US",
"ip": ["127.2.3.4", "::1"],
"port": 8421,
"site": "mlab01",
"tool_extra": {"collector_onion": "testfakenotreal.onion"},
},
{
"fqdn": "blorp",
"city": "Elsewhereton",
"country": "US",
"ip": ["127.2.3.8", "::7"],
"port": 1248,
"site": "mlab02",
"tool_extra": {"collector_onion": "yetanother.onion"},
},
]
# Mocks:
db = dict((entry["fqdn"], entry) for entry in entries)
m_request = MagicMock()
# Fake a request with sufficient parameters:
m_request.args = {"format": ["json"], "policy": ["all"]}
# Execute the code under test:
lsr = lookup.LookupSimulatorResource(db)
retval = lsr.render_GET(m_request)
# Verifications:
self.assertEqual(server.NOT_DONE_YET, retval)
# Verify that a 200 response was sent:
self.assertEqual(m_request.mock_calls, [call.sendJsonResponse(entries)])
