|
1 | 1 | import json |
2 | 2 | import pytest |
3 | | -from unittest.mock import ANY, MagicMock, patch |
4 | | -from google.protobuf import wrappers_pb2 |
5 | | - |
6 | | -from durabletask.client import AsyncTaskHubGrpcClient, TaskHubGrpcClient |
7 | | -from durabletask.grpc_options import GrpcChannelOptions, GrpcRetryPolicyOptions |
8 | 3 | from datetime import datetime, timezone |
9 | 4 | from unittest.mock import ANY, AsyncMock, MagicMock, patch |
10 | 5 |
|
| 6 | +from google.protobuf import wrappers_pb2 |
| 7 | + |
11 | 8 | import durabletask.history as history |
12 | 9 | import durabletask.internal.orchestrator_service_pb2 as pb |
13 | 10 | from durabletask.client import AsyncTaskHubGrpcClient, OrchestrationStatus, TaskHubGrpcClient |
| 11 | +from durabletask.grpc_options import GrpcChannelOptions, GrpcRetryPolicyOptions |
14 | 12 | from durabletask.payload.store import LargePayloadStorageOptions, PayloadStore |
15 | 13 |
|
16 | 14 | from durabletask.internal.grpc_interceptor import ( |
@@ -290,6 +288,8 @@ def test_async_client_uses_provided_channel_directly(): |
290 | 288 | client = AsyncTaskHubGrpcClient(channel=provided_channel, host_address=HOST_ADDRESS) |
291 | 289 | assert client._channel is provided_channel |
292 | 290 | mock_get_channel.assert_not_called() |
| 291 | + |
| 292 | + |
293 | 293 | def test_get_orchestration_history_aggregates_chunks_and_deexternalizes_payloads(): |
294 | 294 | store = FakePayloadStore() |
295 | 295 | token = store.upload(b'history payload') |
|
0 commit comments