async def test_can_send_frame(self): stomp = StompReader(None, self.loop) stomp._transport = Mock() stomp.send_frame('SUBSCRIBE', {'ack': 'auto'}, 'ç') stomp._transport.write.assert_called_with(b'SUBSCRIBE\n' b'ack:auto\n' b'\n' b'\xc3\xa7\x00')
async def test_can_send_frame(self): stomp = StompReader(None, self.loop) stomp._transport = Mock() stomp.send_frame("SUBSCRIBE", {"ack": "auto"}, "ç") stomp._transport.write.assert_called_with(b"SUBSCRIBE\n" b"ack:auto\n" b"\n" b"\xc3\xa7\x00")
async def test_can_connect(self): stomp = StompReader(None, self.loop, heartbeat={ "enabled": True, "cx": 1000, "cy": 1000 }) stomp._transport = Mock() stomp.connect() stomp._transport.write.assert_called_with( b"CONNECT\naccept-version:1.1\nheart-beat:1000,1000\n\n\x00")
async def test_can_connect_with_password(self): stomp = StompReader(None, self.loop, heartbeat={ 'enabled': True, 'cx': 1000, 'cy': 1000 }, password='******') stomp._transport = Mock() stomp.connect() stomp._transport.write.assert_called_with( b'CONNECT\naccept-version:1.1\nheart-beat:1000,1000\npasscode:pass\n\n\x00' ) # noqa
async def test_can_connect_with_username(self): stomp = StompReader(None, self.loop, heartbeat={ 'enabled': True, 'cx': 1000, 'cy': 1000 }, username='******') stomp._transport = Mock() stomp.connect() stomp._transport.write.assert_called_with( b'CONNECT\naccept-version:1.1\nheart-beat:1000,1000\nlogin:pkiefer\n\n\x00' ) # noqa
async def test_can_handle_connected_frame_with_heartbeat_disabled( self, heartbeater_klass_mock): frame = Frame("CONNECTED", {"heart-beat": "1000,1000"}, "{}") heartbeater_mock = heartbeater_klass_mock.return_value heartbeater_mock.start = CoroutineMock() stomp = StompReader(None, self.loop, heartbeat={ "enabled": False, "cx": 0, "cy": 0 }) stomp._transport = Mock() await stomp._handle_connect(frame) heartbeater_klass_mock.assert_not_called
async def test_can_connect_with_login_pass(self): stomp = StompReader( None, self.loop, heartbeat={ "enabled": True, "cx": 1000, "cy": 1000 }, username="******", password="******", ) stomp._transport = Mock() stomp.connect() stomp._transport.write.assert_called_with( b"CONNECT\naccept-version:1.1\nheart-beat:1000,1000\nlogin:pkiefer\npasscode:pass\n\n\x00" ) # noqa
async def test_can_handle_connected_frame_with_heartbeat_disabled( self, heartbeater_klass_mock): frame = Frame('CONNECTED', {'heart-beat': '1000,1000'}, '{}') heartbeater_mock = heartbeater_klass_mock.return_value heartbeater_mock.start = CoroutineMock() stomp = StompReader(None, self.loop, heartbeat={ 'enabled': False, 'cx': 0, 'cy': 0 }) stomp._transport = Mock() await stomp._handle_connect(frame) heartbeater_klass_mock.assert_not_called
async def test_send_frame_can_raise_error(self): stomp = StompReader(None, self.loop) stomp._transport = None with self.assertRaises(StompDisconnectedError): stomp.send_frame("SUBSCRIBE", {"ack": "auto"}, "ç")
async def test_send_frame_can_raise_error(self): stomp = StompReader(None, self.loop) stomp._transport = None with self.assertRaises(StompDisconnectedError): stomp.send_frame('SUBSCRIBE', {'ack': 'auto'}, 'ç')