VOL-405: Update the membership test to use twisted timeout for the watch
Change-Id: Idf608d42ac3ba8e431c7c0fd4ba1eb331928463e
diff --git a/voltha/coordinator.py b/voltha/coordinator.py
index a82ebea..9bad7d0 100644
--- a/voltha/coordinator.py
+++ b/voltha/coordinator.py
@@ -270,11 +270,14 @@
def _assert_membership_record_valid(self):
try:
log.info('membership-record-before')
- (_, record) = yield self._retry('GET',
- self.membership_record_key,
- wait='5s',
- index=0
- )
+ is_timeout, (_, record) = yield \
+ self.consul_get_with_timeout(
+ key=self.membership_record_key,
+ index=0,
+ timeout=5)
+ if is_timeout:
+ returnValue(False)
+
log.info('membership-record-after', record=record)
if record is None or \
'Session' not in record or \