Skip to content

Commit 3047216

Browse files
committed
pr feedback
Signed-off-by: Konvalinka <[email protected]>
1 parent 692b4f2 commit 3047216

File tree

3 files changed

+20
-18
lines changed

3 files changed

+20
-18
lines changed

.github/workflows/build.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ jobs:
3737
- 'hooks/openfeature-hooks-opentelemetry/**'
3838
providers/openfeature-provider-env-var:
3939
- 'providers/openfeature-provider-env-var/**'
40-
providers/providers/openfeature-provider-flagd:
41-
- 'providers/providers/openfeature-provider-flagd/**'
40+
providers/openfeature-provider-flagd:
41+
- 'providers/openfeature-provider-flagd/**'
4242
providers/openfeature-provider-flipt:
4343
- 'providers/openfeature-provider-flipt/**'
4444
providers/openfeature-provider-ofrep:

providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd/resolvers/process/connector/grpc_watcher.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -215,10 +215,9 @@ def _fetch_metadata(self) -> typing.Optional[sync_pb2.GetMetadataResponse]:
215215
return None
216216

217217
context_values_request = sync_pb2.GetMetadataRequest()
218-
context_values_response: sync_pb2.GetMetadataResponse
219218
try:
220-
context_values_response = self.stub.GetMetadata(
221-
context_values_request, wait_for_ready=True
219+
context_values_response: sync_pb2.GetMetadataResponse = (
220+
self.stub.GetMetadata(context_values_request, wait_for_ready=True)
222221
)
223222
return context_values_response
224223
except grpc.RpcError as e:

providers/openfeature-provider-flagd/tests/test_grpc_watcher.py

Lines changed: 16 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -60,13 +60,6 @@ def setUp(self):
6060
self.mock_stub.GetMetadata = Mock(return_value=self.mock_metadata)
6161
self.grpc_watcher.stub = self.mock_stub
6262
self.grpc_watcher.active = True
63-
self.shutdown_thread = lambda: threading.Thread(
64-
target=self.shutdown_after_x_seconds
65-
)
66-
67-
def shutdown_after_x_seconds(self, seconds=1):
68-
time.sleep(seconds)
69-
self.grpc_watcher.shutdown()
7063

7164
def test_listen_with_sync_metadata_and_sync_context(self):
7265
sync_context = Struct()
@@ -81,9 +74,12 @@ def test_listen_with_sync_metadata_and_sync_context(self):
8174
)
8275
self.mock_stub.SyncFlags = Mock(return_value=mock_stream_with_sync_context)
8376

84-
self.shutdown_thread().start()
77+
listener = threading.Thread(target=self.grpc_watcher.listen)
78+
listener.start()
8579

86-
self.grpc_watcher.listen()
80+
time.sleep(0.5)
81+
self.grpc_watcher.shutdown()
82+
listener.join(timeout=1)
8783

8884
self.emit_provider_ready.assert_called_once_with(
8985
ProviderEventDetails(message="gRPC sync connection established"),
@@ -98,9 +94,12 @@ def test_listen_with_sync_metadata_only(self):
9894
)
9995
self.mock_stub.SyncFlags = Mock(return_value=mock_stream_no_sync_context)
10096

101-
self.shutdown_thread().start()
97+
listener = threading.Thread(target=self.grpc_watcher.listen)
98+
listener.start()
10299

103-
self.grpc_watcher.listen()
100+
time.sleep(0.5)
101+
self.grpc_watcher.shutdown()
102+
listener.join(timeout=1)
104103

105104
self.emit_provider_ready.assert_called_once_with(
106105
ProviderEventDetails(message="gRPC sync connection established"),
@@ -115,9 +114,13 @@ def test_listen_with_sync_metadata_disabled_in_config(self):
115114
]
116115
)
117116
self.mock_stub.SyncFlags = Mock(return_value=mock_stream_no_sync_context)
118-
self.shutdown_thread().start()
119117

120-
self.grpc_watcher.listen()
118+
listener = threading.Thread(target=self.grpc_watcher.listen)
119+
listener.start()
120+
121+
time.sleep(0.5)
122+
self.grpc_watcher.shutdown()
123+
listener.join(timeout=1)
121124

122125
self.mock_stub.GetMetadata.assert_not_called()
123126

0 commit comments

Comments
 (0)