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))