0ea58cb3bc
Change resmoke.py to wait up to 5 minutes for a mongod/mongos process to start accepting connections. Waiting only 30 seconds for a connection to be established is insufficient on some build variants in Evergreen. This is likely due to file preallocation, disk speed, and other resource factors.
80 lines
2.0 KiB
Python
80 lines
2.0 KiB
Python
"""
|
|
Helper functions.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import os.path
|
|
|
|
import pymongo
|
|
import yaml
|
|
|
|
|
|
def default_if_none(value, default):
|
|
return value if value is not None else default
|
|
|
|
|
|
def is_string_list(lst):
|
|
"""
|
|
Returns true if 'lst' is a list of strings, and false otherwise.
|
|
"""
|
|
return isinstance(lst, list) and all(isinstance(x, basestring) for x in lst)
|
|
|
|
|
|
def is_string_set(value):
|
|
"""
|
|
Returns true if 'value' is a set of strings, and false otherwise.
|
|
"""
|
|
return isinstance(value, set) and all(isinstance(x, basestring) for x in value)
|
|
|
|
|
|
def is_js_file(filename):
|
|
"""
|
|
Returns true if 'filename' ends in .js, and false otherwise.
|
|
"""
|
|
return os.path.splitext(filename)[1] == ".js"
|
|
|
|
|
|
def is_yaml_file(filename):
|
|
"""
|
|
Returns true if 'filename' ends in .yml or .yaml, and false
|
|
otherwise.
|
|
"""
|
|
return os.path.splitext(filename)[1] in (".yaml", ".yml")
|
|
|
|
|
|
def load_yaml_file(filename):
|
|
"""
|
|
Attempts to read 'filename' as YAML.
|
|
"""
|
|
try:
|
|
with open(filename, "r") as fp:
|
|
return yaml.safe_load(fp)
|
|
except yaml.YAMLError as err:
|
|
raise ValueError("File '%s' contained invalid YAML: %s" % (filename, err))
|
|
|
|
|
|
def dump_yaml(value):
|
|
"""
|
|
Returns 'value' formatted as YAML.
|
|
"""
|
|
# Use block (indented) style for formatting YAML.
|
|
return yaml.safe_dump(value, default_flow_style=False).rstrip()
|
|
|
|
|
|
def new_mongo_client(port, read_preference=pymongo.ReadPreference.PRIMARY, timeout_millis=30000):
|
|
"""
|
|
Returns a pymongo.MongoClient connected on 'port' with a read
|
|
preference of 'read_preference'.
|
|
|
|
The PyMongo driver will wait up to 'timeout_millis' milliseconds
|
|
before concluding that the server is unavailable.
|
|
"""
|
|
|
|
kwargs = {"connectTimeoutMS": timeout_millis}
|
|
if pymongo.version_tuple[0] >= 3:
|
|
kwargs["serverSelectionTimeoutMS"] = timeout_millis
|
|
kwargs["connect"] = True
|
|
|
|
return pymongo.MongoClient(port=port, read_preference=read_preference, **kwargs)
|