class TestAsyncDataStore(unittest.TestCase):
def setUp(self):
self.walker = mock.Mock()
self.register_func = mock.Mock()
self.data_store = AsyncDataStore(self.register_func)
self.data_store.register(self.walker)
def test_setup(self):
self.data_store.setup()
self.register_func.assert_called_with(
self.data_store.on_message)
def setUp(self):
self.walker = mock.Mock()
self.register_func = mock.Mock()
self.data_store = AsyncDataStore(self.register_func)
self.data_store.register(self.walker)
