@@ -160,12 +160,7 @@ def test_worker_single_service(self):
160160
161161 # expect a call for traces and services
162162 self ._wait_thread_flush ()
163- eq_ (self .api ._put .call_count , 2 )
164- # check and retrieve the right call
165- endpoint , payload = self ._get_endpoint_payload (self .api ._put .call_args_list , '/v0.3/services' )
166- eq_ (endpoint , '/v0.3/services' )
167- eq_ (len (payload .keys ()), 1 )
168- eq_ (payload ['client.service' ], {'app' : 'django' , 'app_type' : 'web' })
163+ eq_ (self .api ._put .call_count , 1 )
169164
170165 def test_worker_service_called_multiple_times (self ):
171166 # service must be sent correctly
@@ -176,13 +171,7 @@ def test_worker_service_called_multiple_times(self):
176171
177172 # expect a call for traces and services
178173 self ._wait_thread_flush ()
179- eq_ (self .api ._put .call_count , 2 )
180- # check and retrieve the right call
181- endpoint , payload = self ._get_endpoint_payload (self .api ._put .call_args_list , '/v0.3/services' )
182- eq_ (endpoint , '/v0.3/services' )
183- eq_ (len (payload .keys ()), 2 )
184- eq_ (payload ['backend' ], {'app' : 'django' , 'app_type' : 'web' })
185- eq_ (payload ['database' ], {'app' : 'postgres' , 'app_type' : 'db' })
174+ eq_ (self .api ._put .call_count , 1 )
186175
187176 def test_worker_http_error_logging (self ):
188177 # Tests the logging http error logic
0 commit comments