def test_invoke_service_bytes_data(self): dapr = DaprClient(f'localhost:{self.server_port}') resp = dapr.invoke_service( id='targetId', method='bytes', data=b'haha', content_type="text/plain", metadata=( ('key1', 'value1'), ('key2', 'value2'), ), ) self.assertEqual(b'haha', resp.data) self.assertEqual("text/plain", resp.content_type) self.assertEqual(3, len(resp.headers)) self.assertEqual(['value1'], resp.headers['hkey1'])
def test_dapr_api_token_insertion(self): dapr = DaprClient(f'localhost:{self.server_port}') resp = dapr.invoke_service( id='targetId', method='bytes', data=b'haha', content_type="text/plain", metadata=( ('key1', 'value1'), ('key2', 'value2'), ), ) self.assertEqual(b'haha', resp.content) self.assertEqual("text/plain", resp.content_type) self.assertEqual(4, len(resp.headers)) self.assertEqual(['value1'], resp.headers['hkey1']) self.assertEqual(['test-token'], resp.headers['hdapr-api-token']) self.assertEqual(['value1'], resp.trailers['tkey1'])
def test_invoke_service_proto_data(self): dapr = DaprClient(f'localhost:{self.server_port}') req = common_v1.StateItem(key='test') resp = dapr.invoke_service( id='targetId', method='proto', data=req, metadata=( ('key1', 'value1'), ('key2', 'value2'), ), ) self.assertEqual(3, len(resp.headers)) self.assertEqual(['value1'], resp.headers['hkey1']) self.assertTrue(resp.is_proto()) # unpack to new protobuf object new_resp = common_v1.StateItem() resp.unpack(new_resp) self.assertEqual('test', new_resp.key)