def setUp(self):
self.wrapped = mock.Mock(Resource)
self.resource = TracingWrapperResource(self.wrapped)
self.request = mock.Mock(Request)
self.request.method = 'GET'
self.request.requestHeaders = mock.Mock(wraps=Headers({}))
self.request.getHost.return_value.host = '127.0.0.1'
self.request.getHost.return_value.port = 8080
def test_sets_endpoint_with_service_name(self, mock_trace):
resource = TracingWrapperResource(
self.wrapped, service_name='test-http')
resource.getChildWithDefault('foo', self.request)
set_endpoint = mock_trace.return_value.set_endpoint
self.assertEqual(set_endpoint.call_count, 1)
endpoint = set_endpoint.mock_calls[0][1][0]
self.assertEqual(endpoint.ipv4, '127.0.0.1')
self.assertEqual(endpoint.port, 8080)
self.assertEqual(endpoint.service_name, 'test-http')
