Mock etcd client delete method

This commit is contained in:
Alexander Kukushkin
2015-10-15 09:08:16 +02:00
parent 5c86b60cd2
commit 2f0cf1db06
+2
View File
@@ -1,3 +1,4 @@
import etcd
import unittest
from mock import Mock, patch
@@ -98,6 +99,7 @@ class TestHa(unittest.TestCase):
self.e = Etcd('foo', {'ttl': 30, 'host': 'ok:2379', 'scope': 'test'})
self.e.client.read = etcd_read
self.e.client.write = etcd_write
self.e.client.delete = Mock(side_effect=etcd.EtcdException())
self.ha = Ha(MockPatroni(self.p, self.e))
self.ha._async_executor.run_async = run_async
self.ha.old_cluster = self.e.get_cluster()