# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
-# 0.0510-1301, USA.
+# 02110-1301, USA.
"""Script for unittesting the confd client module"""
+import socket
import unittest
from ganeti import confd
self.mytime += delta
-class TestClient(unittest.TestCase):
- """Client tests"""
+class _BaseClientTest:
+ """Base class for client tests"""
+ mc_list = None
+ new_peers = None
+ family = None
def setUp(self):
self.mock_time = MockTime()
confd.client.ConfdAsyncUDPClient = MockConfdAsyncUDPClient
self.logger = MockLogger()
hmac_key = "mykeydata"
- self.mc_list = ['10.0.0.1',
- '10.0.0.2',
- '10.0.0.3',
- '10.0.0.4',
- '10.0.0.5',
- '10.0.0.6',
- '10.0.0.7',
- '10.0.0.8',
- '10.0.0.9',
- ]
self.callback = MockCallback()
self.client = confd.client.ConfdClient(hmac_key, self.mc_list,
self.callback, logger=self.logger)
self.assertEquals(self.callback.call_count, 1)
def testUpdatePeerList(self):
- new_peers = ['1.2.3.4', '1.2.3.5']
- self.client.UpdatePeerList(new_peers)
- self.assertEquals(self.client._peers, new_peers)
+ self.client.UpdatePeerList(self.new_peers)
+ self.assertEquals(self.client._peers, self.new_peers)
req = confd.client.ConfdClientRequest(type=constants.CONFD_REQ_PING)
self.client.SendRequest(req)
- self.assertEquals(self.client._socket.send_count, len(new_peers))
- self.assert_(self.client._socket.last_address in new_peers)
+ self.assertEquals(self.client._socket.send_count, len(self.new_peers))
+ self.assert_(self.client._socket.last_address in self.new_peers)
+
+ def testSetPeersFamily(self):
+ self.client._SetPeersAddressFamily()
+ self.assertEquals(self.client._family, self.family)
+ mixed_peers = ["192.0.2.99", "2001:db8:beef::13"]
+ self.client.UpdatePeerList(mixed_peers)
+ self.assertRaises(errors.ConfdClientError,
+ self.client._SetPeersAddressFamily)
+
+
+class TestIP4Client(unittest.TestCase, _BaseClientTest):
+ """Client tests"""
+ mc_list = ["192.0.2.1",
+ "192.0.2.2",
+ "192.0.2.3",
+ "192.0.2.4",
+ "192.0.2.5",
+ "192.0.2.6",
+ "192.0.2.7",
+ "192.0.2.8",
+ "192.0.2.9",
+ ]
+ new_peers = ["198.51.100.1", "198.51.100.2"]
+ family = socket.AF_INET
+
+ def setUp(self):
+ unittest.TestCase.setUp(self)
+ _BaseClientTest.setUp(self)
+
+
+class TestIP6Client(unittest.TestCase, _BaseClientTest):
+ """Client tests"""
+ mc_list = ["2001:db8::1",
+ "2001:db8::2",
+ "2001:db8::3",
+ "2001:db8::4",
+ "2001:db8::5",
+ "2001:db8::6",
+ "2001:db8::7",
+ "2001:db8::8",
+ "2001:db8::9",
+ ]
+ new_peers = ["2001:db8:beef::11", "2001:db8:beef::12"]
+ family = socket.AF_INET6
+
+ def setUp(self):
+ unittest.TestCase.setUp(self)
+ _BaseClientTest.setUp(self)
if __name__ == '__main__':