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