From 2f0cf1db06269d05c8ad30f865c8f788078bf16c Mon Sep 17 00:00:00 2001
From: Alexander Kukushkin <alexander.kukushkin@zalando.de>
Date: Thu, 15 Oct 2015 09:08:16 +0200
Subject: [PATCH] Mock etcd client delete method

---
 tests/test_ha.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_ha.py b/tests/test_ha.py
index 38bb07a6..a5a816da 100644
--- a/tests/test_ha.py
+++ b/tests/test_ha.py
@@ -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()