def _disconnect(self): """Send disconnect command """ self.log.debug("Sending disconnect command") cmd = stomper.disconnect() # self.log.debug("Writing cmd: %s" % cmd) self.transport.write(cmd)
def test_frames_with_optional_newlines_succeeds(self): parser = StompParser() frame = '\n%s\n' % stomper.disconnect() parser.add(2 * frame) for _ in xrange(2): self.assertEqual(parser.getMessage(), {'cmd': 'DISCONNECT', 'headers': {}, 'body': ''}) self.assertEqual(parser.getMessage(), None)
def test_dataReceived_unexpected_exception(self): class MyError(Exception): pass stomp = StompClient() stomp.cmdMap['DISCONNECT'] = mock.Mock(side_effect=MyError) self.assertRaises(MyError, stomp.dataReceived, stomper.disconnect())
def test_frame_without_header_or_body_succeeds(self): cmd = stomper.disconnect() lines = cmd.split('\n')[:-1] parser = StompFrameLineParser() for line in lines: parser.processLine(line) msg = parser.getMessage() self.assertEqual(msg, {'cmd': 'DISCONNECT', 'headers': {}, 'body': ''})
def testDisconnect(self): correct = "DISCONNECT\n\n\x00\n" self.assertEquals(stomper.disconnect(), correct)
def test_frame_without_header_or_body_succeeds(self): parser = StompParser() parser.add(stomper.disconnect()) msg = parser.getMessage() self.assertEqual(msg, {'cmd': 'DISCONNECT', 'headers': {}, 'body': ''})
def disconnect(self): self._write(stomper.disconnect()) self._socketDisconnect()