0b1723ed77
There's a lot of methods implementing the common watcher tools that are inherited from all drivers whereas there is no need for that. The mixin is actually not correctly designed, as it mixes the hooks and the cache code. This splits out the common implementation into its common class that can be inherited to implement this model of driver. The Kazoo and Zake drivers are also merged to simplify code, as the base class was not used. Change-Id: I1acbf1983165d00e68b96cbca2bc5c1e2acadfa8
86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (c) 2015 OpenStack Foundation
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import socket
|
|
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
from testtools import testcase
|
|
|
|
from tooz import coordination
|
|
from tooz import tests
|
|
|
|
|
|
class TestMemcacheDriverFailures(testcase.TestCase):
|
|
FAKE_URL = "memcached://mocked-not-really-localhost"
|
|
|
|
@mock.patch('pymemcache.client.PooledClient')
|
|
def test_client_failure_start(self, mock_client_cls):
|
|
mock_client_cls.side_effect = socket.timeout('timed-out')
|
|
member_id = tests.get_random_uuid()
|
|
coord = coordination.get_coordinator(self.FAKE_URL, member_id)
|
|
self.assertRaises(coordination.ToozConnectionError, coord.start)
|
|
|
|
@mock.patch('pymemcache.client.PooledClient')
|
|
def test_client_failure_join(self, mock_client_cls):
|
|
mock_client = mock.MagicMock()
|
|
mock_client_cls.return_value = mock_client
|
|
member_id = tests.get_random_uuid()
|
|
coord = coordination.get_coordinator(self.FAKE_URL, member_id)
|
|
coord.start()
|
|
mock_client.gets.side_effect = socket.timeout('timed-out')
|
|
fut = coord.join_group(tests.get_random_uuid())
|
|
self.assertRaises(coordination.ToozConnectionError, fut.get)
|
|
|
|
@mock.patch('pymemcache.client.PooledClient')
|
|
def test_client_failure_leave(self, mock_client_cls):
|
|
mock_client = mock.MagicMock()
|
|
mock_client_cls.return_value = mock_client
|
|
member_id = tests.get_random_uuid()
|
|
coord = coordination.get_coordinator(self.FAKE_URL, member_id)
|
|
coord.start()
|
|
mock_client.gets.side_effect = socket.timeout('timed-out')
|
|
fut = coord.leave_group(tests.get_random_uuid())
|
|
self.assertRaises(coordination.ToozConnectionError, fut.get)
|
|
|
|
@mock.patch('pymemcache.client.PooledClient')
|
|
def test_client_failure_heartbeat(self, mock_client_cls):
|
|
mock_client = mock.MagicMock()
|
|
mock_client_cls.return_value = mock_client
|
|
member_id = tests.get_random_uuid()
|
|
coord = coordination.get_coordinator(self.FAKE_URL, member_id)
|
|
coord.start()
|
|
mock_client.set.side_effect = socket.timeout('timed-out')
|
|
self.assertRaises(coordination.ToozConnectionError, coord.heartbeat)
|
|
|
|
@mock.patch(
|
|
'tooz.coordination.CoordinationDriverCachedRunWatchers.run_watchers',
|
|
autospec=True)
|
|
@mock.patch('pymemcache.client.PooledClient')
|
|
def test_client_run_watchers_mixin(self, mock_client_cls,
|
|
mock_run_watchers):
|
|
mock_client = mock.MagicMock()
|
|
mock_client_cls.return_value = mock_client
|
|
member_id = tests.get_random_uuid()
|
|
coord = coordination.get_coordinator(self.FAKE_URL, member_id)
|
|
coord.start()
|
|
coord.run_watchers()
|
|
self.assertTrue(mock_run_watchers.called)
|