def setUp(self):
self.m_sck = Mock(spec=socket.socket)
self.driver = EtcdDriver(self.m_sck)
self.msg_reader = StubMessageReader(self.m_sck)
self.msg_writer = StubMessageWriter(self.m_sck)
self.driver._msg_reader = self.msg_reader
self.driver._msg_writer = self.msg_writer
def setUp(self):
sck = Mock()
self.watcher_etcd = StubEtcd()
self.resync_etcd = StubEtcd()
self.driver = EtcdDriver(sck)
self.orig_next_watcher_event = self.driver._next_watcher_event
self.driver._next_watcher_event = self._next_watcher_event
self.msg_reader = StubMessageReader(sck)
self.msg_writer = StubMessageWriter(sck)
self.driver._msg_reader = self.msg_reader
self.driver._msg_writer = self.msg_writer
self.driver._issue_etcd_request = Mock(
spec=self.driver._issue_etcd_request,
side_effect=self.mock_etcd_request)
self._logging_patch = patch(
"calico.etcddriver.driver."
"complete_logging", autospec=True)
self._logging_patch.start()
