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() ]) )
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()])
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() ])
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") ])
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")])
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"