コード例 #1
0
    def test_stop(self):
        self._subject._recv_thread = MagicMock()
        self._subject._send_thread = MagicMock()
        self._subject._expire_thread = MagicMock()

        self._subject.stop()

        assert_that(
            self._subject._stopped,
            equal_to(True)
        )

        assert_that(
            self._subject._recv_thread.mock_calls,
            equal_to([
                call.join()
            ])
        )

        assert_that(
            self._subject._send_thread.mock_calls,
            equal_to([
                call.join()
            ])
        )

        assert_that(
            self._subject._expire_thread.mock_calls,
            equal_to([
                call.join()
            ])
        )
コード例 #2
0
 def test_reparent(self, m_driver_cls, m_logging, m_socket, m_ppid):
     m_ppid.side_effect = iter([123, 123, 1])
     m_driver = m_driver_cls.return_value
     m_driver.join.return_value = False
     self._import_main()
     self.assertEqual(m_driver.mock_calls,
                      [call.start(),
                       call.join(timeout=1),
                       call.join(timeout=1),
                       call.stop()])
コード例 #3
0
ファイル: test_main.py プロジェクト: yekeqiang/calico
 def test_reparent(self, m_driver_cls, m_logging, m_socket, m_ppid):
     m_ppid.side_effect = iter([123, 123, 1])
     m_driver = m_driver_cls.return_value
     m_driver.join.return_value = False
     self._import_main()
     self.assertEqual(m_driver.mock_calls, [
         call.start(),
         call.join(timeout=1),
         call.join(timeout=1),
         call.stop()
     ])
コード例 #4
0
ファイル: test_main.py プロジェクト: yekeqiang/calico
 def test_mainline(self, m_driver_cls, m_logging, m_socket, m_ppid):
     m_ppid.return_value = 123
     m_driver = m_driver_cls.return_value
     m_driver.join.side_effect = iter([False, True])
     self._import_main()
     self.assertEqual(
         m_driver.mock_calls,
         [call.start(),
          call.join(timeout=1),
          call.join(timeout=1)])
     self.assertEqual(m_logging.mock_calls, [
         call(gevent_in_use=False,
              syslog_executable_name="calico-felix-etcd")
     ])
コード例 #5
0
 def test_mainline(self, m_driver_cls, m_logging, m_socket, m_ppid):
     m_ppid.return_value = 123
     m_driver = m_driver_cls.return_value
     m_driver.join.side_effect = iter([
         False,
         True
     ])
     self._import_main()
     self.assertEqual(m_driver.mock_calls,
                      [call.start(),
                       call.join(timeout=1),
                       call.join(timeout=1)])
     self.assertEqual(m_logging.mock_calls,
                      [call(gevent_in_use=False,
                            syslog_executable_name="calico-felix-etcd")])
コード例 #6
0
ファイル: test_slot.py プロジェクト: matthewstory/pg2kinesis
def test__exit__(slot):
    slot._keepalive_window = 10
    slot.__exit__(None, None, None)

    assert call.close() in slot._repl_cursor.method_calls
    assert call.close() in slot._repl_conn.method_calls
    assert call.close() in slot._normal_conn.method_calls
    assert call.join(timeout=13) in slot._keepalive_thread.method_calls

    slot._repl_cursor.close = Mock(side_effect=Exception)
    slot._repl_conn.close = Mock(side_effect=Exception)
    slot._normal_conn.close = Mock(side_effect=Exception)
    slot._keepalive_thread.join = Mock(side_effect=Exception)
    slot.__exit__(None, None, None)

    assert slot._keepalive_thread.join.called
    assert slot._repl_cursor.close.called, "Still called even thought call above raised"
    assert slot._repl_conn.close.called, "Still called even thought call above raised"
    assert slot._normal_conn.close.called, "Still called even thought call above raised"