diff --git a/tests/helper.py b/tests/helper.py index 9f0358f..280e72c 100644 --- a/tests/helper.py +++ b/tests/helper.py @@ -1,3 +1,4 @@ +import configparser from tornado.ioloop import IOLoop import tornado.testing from nvchecker.get_version import get_version @@ -11,5 +12,12 @@ class ExternalVersionTestCase(tornado.testing.AsyncTestCase): def get_version_callback(name, version): self.stop(version) - get_version(name, config, get_version_callback) + if isinstance(config, dict): + _config = configparser.ConfigParser(dict_type=dict, allow_no_value=True) + _config.read_dict({name: config}) + _config = _config[name] + else: + _config = config + + get_version(name, _config, get_version_callback) return self.wait()