Switch to a custom NotImplemented error

Some code in the drivers might actually raise the standard
NotImplemented error, and that would be a bug to skip a test if that was
the case. In that case we want the test to fail.

So let's switch to a custom exception that is used to skip the test if
it's raised.

Change-Id: Ideafee0b1f008ff32724fb98d6a477bd3976104d

1 files changed, 8 insertions, 7 deletions

diff --git a/tooz/drivers/zake.py b/tooz/drivers/zake.py
index 4e43ae1..5ca244e 100644
--- a/tooz/drivers/zake.py
+++ b/tooz/drivers/zake.py
@@ -17,6 +17,7 @@ from __future__ import absolute_import
 from zake import fake_client
 from zake import fake_storage
 
+import tooz
 from tooz.drivers import zookeeper
 
 
@@ -35,28 +36,28 @@ class ZakeDriver(zookeeper.BaseZooKeeperDriver):
 
     @staticmethod
     def watch_join_group(group_id, callback):
-        raise NotImplementedError
+        raise tooz.NotImplemented
 
     @staticmethod
     def unwatch_join_group(group_id, callback):
-        raise NotImplementedError
+        raise tooz.NotImplemented
 
     @staticmethod
     def watch_leave_group(group_id, callback):
-        raise NotImplementedError
+        raise tooz.NotImplemented
 
     @staticmethod
     def unwatch_leave_group(group_id, callback):
-        raise NotImplementedError
+        raise tooz.NotImplemented
 
     @staticmethod
     def watch_elected_as_leader(group_id, callback):
-        raise NotImplementedError
+        raise tooz.NotImplemented
 
     @staticmethod
     def unwatch_elected_as_leader(group_id, callback):
-        raise NotImplementedError
+        raise tooz.NotImplemented
 
     @staticmethod
     def run_watchers():
-        raise NotImplementedError
+        raise tooz.NotImplemented