From 43e17688bbdebaa20d3ea3e1f2bbdee9f84866d6 Mon Sep 17 00:00:00 2001 From: "T. Dampier" Date: Wed, 11 Apr 2012 17:45:27 -0700 Subject: [PATCH] Update test_replica_set.py to reflect new ability to make direct connection to an arbiter. --- test/replica/test_replica_set.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/test/replica/test_replica_set.py b/test/replica/test_replica_set.py index 97239e617..ca13f15cb 100644 --- a/test/replica/test_replica_set.py +++ b/test/replica/test_replica_set.py @@ -94,8 +94,11 @@ class TestSecondaryConnection(unittest.TestCase): # Test direct connection to an arbiter secondary_host = replset_tools.get_arbiters()[0] - self.assertRaises( - ConnectionFailure, Connection, secondary_host, use_greenlets=use_greenlets) + host, port = replset_tools.arbiters()[0].split(':') + port = int(port) + conn = Connection(host, port) + self.assertEqual(host, conn.host) + self.assertEqual(port, conn.port) def tearDown(self): self.c.close()