def test_init_consistence(self, mock_spawn): test_driver = TestPollingDataSourceDriver.TestDriver() mock_spawn.assert_not_called() self.assertIsNone(test_driver.worker_greenthread) test_driver.start() mock_spawn.assert_called_once_with(test_driver.poll_loop, test_driver.poll_time) self.assertTrue(test_driver.initialized) self.assertIsNotNone(test_driver.worker_greenthread)
def test_cleanup(self, mock_spawn, mock_kill): dummy_thread = dict() mock_spawn.return_value = dummy_thread test_driver = TestPollingDataSourceDriver.TestDriver() test_driver.start() self.assertEqual(test_driver.worker_greenthread, dummy_thread) test_driver.stop() mock_kill.assert_called_once_with(dummy_thread) self.assertIsNone(test_driver.worker_greenthread)