def test_maybe_model():
class X(Record):
x: int
y: int
assert maybe_model("foo") == "foo"
assert maybe_model(1) == 1
assert maybe_model(1.01) == 1.01
x1 = X(10, 20)
assert maybe_model(json.loads(x1.dumps(serializer="json"))) == x1
def test_maybe_model():
class X(Record):
x: int
y: int
assert maybe_model('foo') == 'foo'
assert maybe_model(1) == 1
assert maybe_model(1.01) == 1.01
x1 = X(10, 20)
assert maybe_model(json.loads(x1.dumps(serializer='json'))) == x1
async def _modify_set(self, stream: StreamT[SetManagerOperation]) -> None:
async for set_key, set_operation in stream.items():
try:
action = SetAction(set_operation.action)
except ValueError:
self.log.exception(
'Unknown set operation: %r', set_operation.action)
else:
member_obj = maybe_model(set_operation.member)
handler = self.actions[action]
handler(set_key, member_obj)
async def wait(self, *, key: Any = None, timeout: Seconds = None) -> VT:
"""Wait for signal to be completed."""
# wait for key to arrive in consumer
runner = self.case.current_execution
if runner is None:
raise RuntimeError("No test executing.")
test = runner.test
assert test
k: Any = test.id if key is None else key
timeout_s = want_seconds(timeout)
await runner.on_signal_wait(self, timeout=timeout_s)
time_start = monotonic()
event = await self._wait_for_message_by_key(key=k, timeout=timeout_s)
time_end = monotonic()
await runner.on_signal_received(
self,
time_start=time_start,
time_end=time_end,
)
self._verify_event(event, k, self.name, self.case.name)
return cast(VT, maybe_model(event.value))
