def validate_auth_mechanism(dummy, value): """Validate the authMechanism URI option. """ try: return MECHANISMS.index(value) except ValueError: raise ConfigurationError("%s is not a supported " "value for authMechanism" % (value,))
def test_split_options(self): self.assertRaises(ConfigurationError, split_options, 'foo') self.assertRaises(ConfigurationError, split_options, 'foo=bar') self.assertRaises(ConfigurationError, split_options, 'foo=bar;foo') self.assertRaises(ConfigurationError, split_options, 'socketTimeoutMS=foo') self.assertRaises(ConfigurationError, split_options, 'socketTimeoutMS=0.0') self.assertRaises(ConfigurationError, split_options, 'connectTimeoutMS=foo') self.assertRaises(ConfigurationError, split_options, 'connectTimeoutMS=0.0') self.assertRaises(ConfigurationError, split_options, 'connectTimeoutMS=1e100000') self.assertRaises(ConfigurationError, split_options, 'connectTimeoutMS=-1e100000') # On most platforms float('inf') and float('-inf') represent # +/- infinity, although on Python 2.4 and 2.5 on Windows those # expressions are invalid if not (sys.platform == "win32" and sys.version_info <= (2, 5)): self.assertRaises(ConfigurationError, split_options, 'connectTimeoutMS=inf') self.assertRaises(ConfigurationError, split_options, 'connectTimeoutMS=-inf') self.assertTrue(split_options('socketTimeoutMS=300')) self.assertTrue(split_options('connectTimeoutMS=300')) self.assertEqual({'sockettimeoutms': 0.3}, split_options('socketTimeoutMS=300')) self.assertEqual({'sockettimeoutms': 0.0001}, split_options('socketTimeoutMS=0.1')) self.assertEqual({'connecttimeoutms': 0.3}, split_options('connectTimeoutMS=300')) self.assertEqual({'connecttimeoutms': 0.0001}, split_options('connectTimeoutMS=0.1')) self.assertTrue(split_options('connectTimeoutMS=300')) self.assertTrue(isinstance(split_options('w=5')['w'], int)) self.assertTrue(isinstance(split_options('w=5.5')['w'], basestring)) self.assertTrue(split_options('w=foo')) self.assertTrue(split_options('w=majority')) self.assertRaises(ConfigurationError, split_options, 'wtimeoutms=foo') self.assertRaises(ConfigurationError, split_options, 'wtimeoutms=5.5') self.assertTrue(split_options('wtimeoutms=500')) self.assertRaises(ConfigurationError, split_options, 'fsync=foo') self.assertRaises(ConfigurationError, split_options, 'fsync=5.5') self.assertEqual({'fsync': True}, split_options('fsync=true')) self.assertEqual({'fsync': False}, split_options('fsync=false')) self.assertEqual({'authmechanism': MECHANISMS.index('GSSAPI')}, split_options('authMechanism=GSSAPI')) self.assertEqual({'authmechanism': MECHANISMS.index('MONGO-CR')}, split_options('authMechanism=MONGO-CR')) self.assertEqual({'authsource': 'foobar'}, split_options('authSource=foobar')) # maxPoolSize isn't yet a documented URI option. self.assertRaises(ConfigurationError, split_options, 'maxpoolsize=50')
def test_parse_uri(self): self.assertRaises(InvalidURI, parse_uri, "http://foobar.com") self.assertRaises(InvalidURI, parse_uri, "http://[email protected]") self.assertRaises(ConfigurationError, parse_uri, "mongodb://::1", 27017) orig = { 'nodelist': [("localhost", 27017)], 'username': None, 'password': None, 'database': None, 'collection': None, 'options': {} } res = copy.deepcopy(orig) self.assertEqual(res, parse_uri("mongodb://localhost")) res.update({'username': '******', 'password': '******'}) self.assertEqual(res, parse_uri("mongodb://*****:*****@localhost")) res.update({'database': 'baz'}) self.assertEqual(res, parse_uri("mongodb://*****:*****@localhost/baz")) res = copy.deepcopy(orig) res['nodelist'] = [("example1.com", 27017), ("example2.com", 27017)] self.assertEqual(res, parse_uri("mongodb://example1.com:27017," "example2.com:27017")) res = copy.deepcopy(orig) res['nodelist'] = [("localhost", 27017), ("localhost", 27018), ("localhost", 27019)] self.assertEqual(res, parse_uri("mongodb://localhost," "localhost:27018,localhost:27019")) res = copy.deepcopy(orig) res['database'] = 'foo' self.assertEqual(res, parse_uri("mongodb://localhost/foo")) res = copy.deepcopy(orig) self.assertEqual(res, parse_uri("mongodb://localhost/")) res.update({'database': 'test', 'collection': 'yield_historical.in'}) self.assertEqual(res, parse_uri("mongodb://" "localhost/test.yield_historical.in")) res.update({'username': '******', 'password': '******'}) self.assertEqual(res, parse_uri("mongodb://*****:*****@localhost/" "test.yield_historical.in")) res = copy.deepcopy(orig) res['nodelist'] = [("example1.com", 27017), ("example2.com", 27017)] res.update({'database': 'test', 'collection': 'yield_historical.in'}) self.assertEqual(res, parse_uri("mongodb://example1.com:27017,example2.com" ":27017/test.yield_historical.in")) res = copy.deepcopy(orig) res['nodelist'] = [("::1", 27017)] res['options'] = {'slaveok': True} self.assertEqual(res, parse_uri("mongodb://[::1]:27017/?slaveOk=true")) res = copy.deepcopy(orig) res['nodelist'] = [("2001:0db8:85a3:0000:0000:8a2e:0370:7334", 27017)] res['options'] = {'slaveok': True} self.assertEqual(res, parse_uri( "mongodb://[2001:0db8:85a3:0000:0000" ":8a2e:0370:7334]:27017/?slaveOk=true")) res = copy.deepcopy(orig) res['nodelist'] = [("/tmp/mongodb-27017.sock", None)] self.assertEqual(res, parse_uri("mongodb:///tmp/mongodb-27017.sock")) res = copy.deepcopy(orig) res['nodelist'] = [("example2.com", 27017), ("/tmp/mongodb-27017.sock", None)] self.assertEqual(res, parse_uri("mongodb://example2.com," "/tmp/mongodb-27017.sock")) res = copy.deepcopy(orig) res['nodelist'] = [("shoe.sock.pants.co.uk", 27017), ("/tmp/mongodb-27017.sock", None)] res['database'] = "nethers_db" self.assertEqual(res, parse_uri("mongodb://shoe.sock.pants.co.uk," "/tmp/mongodb-27017.sock/nethers_db")) res = copy.deepcopy(orig) res['nodelist'] = [("/tmp/mongodb-27017.sock", None), ("example2.com", 27017)] res.update({'database': 'test', 'collection': 'yield_historical.in'}) self.assertEqual(res, parse_uri("mongodb:///tmp/mongodb-27017.sock," "example2.com:27017" "/test.yield_historical.in")) res = copy.deepcopy(orig) res['nodelist'] = [("/tmp/mongodb-27017.sock", None), ("example2.com", 27017)] res.update({'database': 'test', 'collection': 'yield_historical.sock'}) self.assertEqual(res, parse_uri("mongodb:///tmp/mongodb-27017.sock," "example2.com:27017" "/test.yield_historical.sock")) res = copy.deepcopy(orig) res['nodelist'] = [("example2.com", 27017)] res.update({'database': 'test', 'collection': 'yield_historical.sock'}) self.assertEqual(res, parse_uri("mongodb://example2.com:27017" "/test.yield_historical.sock")) res = copy.deepcopy(orig) res['nodelist'] = [("/tmp/mongodb-27017.sock", None)] res.update({'database': 'test', 'collection': 'mongodb-27017.sock'}) self.assertEqual(res, parse_uri("mongodb:///tmp/mongodb-27017.sock" "/test.mongodb-27017.sock")) res = copy.deepcopy(orig) res['nodelist'] = [('/tmp/mongodb-27020.sock', None), ("::1", 27017), ("2001:0db8:85a3:0000:0000:8a2e:0370:7334", 27018), ("192.168.0.212", 27019), ("localhost", 27018)] self.assertEqual(res, parse_uri("mongodb:///tmp/mongodb-27020.sock," "[::1]:27017,[2001:0db8:" "85a3:0000:0000:8a2e:0370:7334]," "192.168.0.212:27019,localhost", 27018)) res = copy.deepcopy(orig) res.update({'username': '******', 'password': '******'}) res.update({'database': 'test', 'collection': 'yield_historical.in'}) res['options'] = {'slaveok': True} self.assertEqual(res, parse_uri("mongodb://*****:*****@localhost/" "test.yield_historical.in?slaveok=true")) res = copy.deepcopy(orig) res['options'] = {'readpreference': ReadPreference.SECONDARY} self.assertEqual(res, parse_uri("mongodb://localhost/?readPreference=secondary")) # Various authentication tests res = copy.deepcopy(orig) res['options'] = {'authmechanism': MECHANISMS.index('MONGO-CR')} res['username'] = '******' res['password'] = '******' self.assertEqual(res, parse_uri("mongodb://*****:*****@localhost/" "?authMechanism=MONGO-CR")) res = copy.deepcopy(orig) res['options'] = {'authmechanism': MECHANISMS.index('MONGO-CR'), 'authsource': 'bar'} res['username'] = '******' res['password'] = '******' res['database'] = 'foo' self.assertEqual(res, parse_uri("mongodb://*****:*****@localhost/foo" "?authSource=bar;authMechanism=MONGO-CR")) res = copy.deepcopy(orig) res['options'] = {'authmechanism': MECHANISMS.index('MONGO-CR')} res['username'] = '******' res['password'] = '' self.assertEqual(res, parse_uri("mongodb://user:@localhost/" "?authMechanism=MONGO-CR")) res = copy.deepcopy(orig) res['username'] = '******' res['password'] = '******' res['database'] = 'foo' self.assertEqual(res, parse_uri("mongodb://user%40domain.com:password" "@localhost/foo")) res = copy.deepcopy(orig) res['options'] = {'authmechanism': MECHANISMS.index('GSSAPI')} res['username'] = '******' res['password'] = '******' res['database'] = 'foo' self.assertEqual(res, parse_uri("mongodb://user%40domain.com:password" "@localhost/foo?authMechanism=GSSAPI")) res = copy.deepcopy(orig) res['options'] = {'authmechanism': MECHANISMS.index('GSSAPI')} res['username'] = '******' res['password'] = '' res['database'] = 'foo' self.assertEqual(res, parse_uri("mongodb://user%40domain.com" "@localhost/foo?authMechanism=GSSAPI"))