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)])