Exemplos de MockSTStmgrClient em Python

Exemplo n.º 1

Arquivo: st_stmgr_client_unittest.py Projeto: yangxu1362850636/heron

 def setUp(self):
   self.stmgr_client = MockSTStmgrClient()

Exemplo n.º 2

Arquivo: st_stmgr_client_unittest.py Projeto: ashvina/heron

 def setUp(self):
   self.stmgr_client = MockSTStmgrClient()

Exemplo n.º 3

Arquivo: st_stmgr_client_unittest.py Projeto: yangxu1362850636/heron

class STStmgrClientTest(unittest.TestCase):
  def setUp(self):
    self.stmgr_client = MockSTStmgrClient()

  def tearDown(self):
    self.stmgr_client = None

  def test_on_connect_ok(self):
    self.stmgr_client.on_connect(StatusCode.OK)
    self.assertTrue(self.stmgr_client.register_msg_called)
    # timeout_task should be included in timer_tasks
    self.assertEqual(len(self.stmgr_client.looper.timer_tasks), 1)

  def test_on_connect_error(self):
    self.stmgr_client.on_connect(StatusCode.CONNECT_ERROR)
    self.assertEqual(len(self.stmgr_client.looper.timer_tasks), 1)
    self.assertEqual(self.stmgr_client.looper.timer_tasks[0][1].__name__, 'start_connect')

  def test_on_response(self):
    with self.assertRaises(RuntimeError):
      self.stmgr_client.on_response(StatusCode.INVALID_PACKET, None, None)

    with self.assertRaises(RuntimeError):
      self.stmgr_client.on_response(StatusCode.OK, None, mock_protobuf.get_mock_instance())

    self.stmgr_client.on_response(StatusCode.OK, None, mock_protobuf.get_mock_register_response())
    self.assertTrue(self.stmgr_client.handle_register_response_called)

  def test_on_error(self):
    self.stmgr_client.on_error()
    self.assertEqual(len(self.stmgr_client.looper.timer_tasks), 1)
    self.assertEqual(self.stmgr_client.looper.timer_tasks[0][1].__name__, 'start_connect')

Exemplo n.º 4

Arquivo: st_stmgr_client_unittest.py Projeto: ashvina/heron

class STStmgrClientTest(unittest.TestCase):
  def setUp(self):
    self.stmgr_client = MockSTStmgrClient()

  def tearDown(self):
    self.stmgr_client = None

  def test_on_connect_ok(self):
    self.stmgr_client.on_connect(StatusCode.OK)
    self.assertTrue(self.stmgr_client.register_msg_called)
    # timeout_task should be included in timer_tasks
    self.assertEqual(len(self.stmgr_client.looper.timer_tasks), 1)

  def test_on_connect_error(self):
    self.stmgr_client.on_connect(StatusCode.CONNECT_ERROR)
    self.assertEqual(len(self.stmgr_client.looper.timer_tasks), 1)
    self.assertEqual(self.stmgr_client.looper.timer_tasks[0][1].__name__, 'start_connect')

  def test_on_response(self):
    with self.assertRaises(RuntimeError):
      self.stmgr_client.on_response(StatusCode.INVALID_PACKET, None, None)

    with self.assertRaises(RuntimeError):
      self.stmgr_client.on_response(StatusCode.OK, None, mock_protobuf.get_mock_instance())

    self.stmgr_client.on_response(StatusCode.OK, None, mock_protobuf.get_mock_register_response())
    self.assertTrue(self.stmgr_client.handle_register_response_called)

  def test_on_error(self):
    self.stmgr_client.on_error()
    self.assertEqual(len(self.stmgr_client.looper.timer_tasks), 1)
    self.assertEqual(self.stmgr_client.looper.timer_tasks[0][1].__name__, 'start_connect')