def test_client_options(self):
repl_set = ReplicaSet().start()
try:
config_def = {
'mainAddress': repl_set.uri,
'oplogFile': from_here('lib', 'dummy.timestamp'),
'docManagers': [
{
'docManager': 'solr_doc_manager',
'targetURL': solr_url,
'args': {
'clientOptions': {
'timeout': 100
}
}
}
]
}
config_obj = config.Config(get_config_options())
config_obj.load_json(json.dumps(config_def))
config_obj.parse_args(argv=[])
conn = connector.Connector.from_config(config_obj)
self.assertEqual(100, conn.doc_managers[0].solr.timeout)
finally:
repl_set.stop()
def test_ssl_options(self):
config_def = {
"mainAddress": "localhost:27017",
"oplogFile": from_here("lib", "dummy.timestamp"),
"ssl": {
"sslCertfile": "certfile.pem",
"sslKeyfile": "certfile.key",
"sslCACerts": "ca.pem",
},
}
for cert_policy, expected_ssl_cert_req in [
("ignored", ssl.CERT_NONE),
("optional", ssl.CERT_OPTIONAL),
("required", ssl.CERT_REQUIRED),
(None, None),
]:
config_def["ssl"]["sslCertificatePolicy"] = cert_policy
config_obj = config.Config(get_config_options())
config_obj.load_json(json.dumps(config_def))
config_obj.parse_args(argv=[])
mc = connector.Connector.from_config(config_obj)
self.assertEqual("certfile.pem", mc.ssl_kwargs.get("ssl_certfile"))
self.assertEqual("ca.pem", mc.ssl_kwargs.get("ssl_ca_certs"))
self.assertEqual("certfile.key", mc.ssl_kwargs.get("ssl_keyfile"))
self.assertEqual(expected_ssl_cert_req, mc.ssl_kwargs.get("ssl_cert_reqs"))
def test_client_options(self):
config_def = {
"mainAddress": "localhost:27017",
"oplogFile": from_here("lib", "dummy.timestamp"),
"docManagers": [
{
"docManager": "mongo_doc_manager",
"targetURL": "dummyhost:27017",
"args": {"clientOptions": {"maxPoolSize": 50, "connect": False}},
}
],
}
config_obj = config.Config(get_config_options())
config_obj.load_json(json.dumps(config_def))
config_obj.parse_args(argv=[])
conn = connector.Connector.from_config(config_obj)
self.assertEqual(50, conn.doc_managers[0].mongo.max_pool_size)
def test_client_options(self):
config_def = {
'mainAddress':
'localhost:27017',
'oplogFile':
from_here('lib', 'dummy.timestamp'),
'docManagers': [{
'docManager': 'mongo_doc_manager',
'targetURL': 'dummyhost:27017',
'args': {
'clientOptions': {
'maxPoolSize': 50,
'connect': False
}
}
}]
}
config_obj = config.Config(get_config_options())
config_obj.load_json(json.dumps(config_def))
config_obj.parse_args(argv=[])
conn = connector.Connector.from_config(config_obj)
self.assertEqual(50, conn.doc_managers[0].mongo.max_pool_size)
def test_client_options(self):
config_def = {
'mainAddress': 'localhost:27017',
'oplogFile': from_here('lib', 'dummy.timestamp'),
'docManagers': [
{
'docManager': 'mongo_doc_manager',
'targetURL': 'dummyhost:27017',
'args': {
'clientOptions': {
'maxPoolSize': 50,
'connect': False
}
}
}
]
}
config_obj = config.Config(get_config_options())
config_obj.load_json(json.dumps(config_def))
config_obj.parse_args(argv=[])
conn = connector.Connector.from_config(config_obj)
self.assertEqual(50, conn.doc_managers[0].mongo.max_pool_size)
def test_ssl_options(self):
config_def = {
'mainAddress': 'localhost:27017',
'oplogFile': from_here('lib', 'dummy.timestamp'),
'ssl': {
'sslCertfile': 'certfile.pem',
'sslKeyfile': 'certfile.key',
'sslCACerts': 'ca.pem'
},
}
for cert_policy, expected_ssl_cert_req in [
('ignored', ssl.CERT_NONE), ('optional', ssl.CERT_OPTIONAL),
('required', ssl.CERT_REQUIRED), (None, None)
]:
config_def['ssl']['sslCertificatePolicy'] = cert_policy
config_obj = config.Config(get_config_options())
config_obj.load_json(json.dumps(config_def))
config_obj.parse_args(argv=[])
mc = connector.Connector.from_config(config_obj)
self.assertEqual('certfile.pem', mc.ssl_kwargs.get('ssl_certfile'))
self.assertEqual('ca.pem', mc.ssl_kwargs.get('ssl_ca_certs'))
self.assertEqual('certfile.key', mc.ssl_kwargs.get('ssl_keyfile'))
self.assertEqual(expected_ssl_cert_req,
mc.ssl_kwargs.get('ssl_cert_reqs'))
def reset_config(self):
self.options = get_config_options()
self.conf = config.Config(self.options)
def setUp(self):
self.config = config.Config(get_config_options())
# Remove all logging Handlers, since tests may create Handlers.
logger = logging.getLogger()
for handler in logger.handlers:
logger.removeHandler(handler)
def reset_config(self):
self.options = get_config_options()
self.conf = config.Config(self.options)