def _assert_source_tags(self, flag):
# one tag
options, _ = _get_options([flag, "tag1"])
eq_(options["source_tags"], ["tag1"])
# many tags
options, _ = _get_options([flag, "tag1,tag2"])
eq_(options["source_tags"], ["tag1", "tag2"])
def _assert_source_tags(self, flag):
# one tag
options = _get_options([flag, 'tag1'])
eq_(options['source_tags'], ['tag1'])
# many tags
options = _get_options([flag, 'tag1,tag2'])
eq_(options['source_tags'], ['tag1', 'tag2'])
def test_defaults(self):
options, _ = _get_options([])
expected = {
"data_file": None,
"parameters": [],
"parameter_file": None,
"force": False,
"profile": False,
"debug": False,
"disco_debug": False,
"server": None,
"settings_file": None,
"source_tags": None,
"source_urls": None,
"just_query": False,
"result_tag": None,
"day_start": None,
"day_range": None,
"day_offset": None,
"immediate_rule": None,
"rules_directory": None,
"start_paused": False,
"example_rules": None,
"process_map": None,
"process_results": None,
"run_daemon": False,
}
assert_dicts_equal(options, expected)
def test_defaults(self):
options = _get_options([])
expected = {
'data_file': None,
'parameters': [],
'parameter_file': None,
'force': False,
'profile': False,
'debug': False,
'disco_debug': False,
'server': None,
'settings_file': None,
'source_tags': None,
'source_urls': None,
'just_query': False,
'result_tag': None,
'day_start': None,
'day_range': None,
'day_offset': None,
'immediate_rule': None,
'rules_directory': None,
'start_paused': False,
'example_rules': None,
'process_map': None,
'process_results': None
}
assert_dicts_equal(options, expected)
def _assert_parameters(self, flag):
# one param
options, _ = _get_options([flag, "some_param: some_value"])
eq_(options["some_param"], "some_value")
# many params
options, _ = _get_options([flag, "some_param_1: some_value_1", flag, "some_param_2: some_value_2"])
eq_(options["some_param_1"], "some_value_1")
eq_(options["some_param_2"], "some_value_2")
# integer
options, _ = _get_options([flag, "some_int: 100"])
eq_(options["some_int"], 100)
# list of integers
options, _ = _get_options([flag, "some_int_list: [100, 200, 300,]"])
eq_(options["some_int_list"], [100, 200, 300])
def _assert_parameters(self, flag):
# one param
options = _get_options([flag, 'some_param: some_value'])
eq_(options['some_param'], 'some_value')
# many params
options = _get_options([
flag, 'some_param_1: some_value_1',
flag, 'some_param_2: some_value_2'])
eq_(options['some_param_1'], 'some_value_1')
eq_(options['some_param_2'], 'some_value_2')
# integer
options = _get_options([flag, 'some_int: 100'])
eq_(options['some_int'], 100)
# list of integers
options = _get_options([flag, 'some_int_list: [100, 200, 300,]'])
eq_(options['some_int_list'], [100, 200, 300])
def test_get_settings(self):
# settings default
options = _get_options(['-e', 'some_unknown_settings_file'])
settings = _get_settings(options)
eq_(settings['server'], 'localhost')
# options override
options = _get_options(['-s', 'some_server'])
settings = _get_settings(options)
eq_(settings['server'], 'some_server')
# file override
settings_file = self._create_settings_file()
options = _get_options(['-e', settings_file])
settings = _get_settings(options)
eq_(settings['server'], 'another_server')
# server from -s trumps -e
settings_file = self._create_settings_file()
options = _get_options(['-e', settings_file, '-s', 'some_server'])
settings = _get_settings(options)
eq_(settings['server'], 'some_server')
def test_get_settings(self):
# settings default
options, _ = _get_options(["-e", "some_unknown_settings_file"])
settings = _get_settings(options)
eq_(settings["server"], "localhost")
# options override
options, _ = _get_options(["-s", "some_server"])
settings = _get_settings(options)
eq_(settings["server"], "some_server")
# file override
settings_file = self._create_settings_file()
options, _ = _get_options(["-e", settings_file])
settings = _get_settings(options)
eq_(settings["server"], "another_server")
# server from -s trumps -e
settings_file = self._create_settings_file()
options, _ = _get_options(["-e", settings_file, "-s", "some_server"])
settings = _get_settings(options)
eq_(settings["server"], "some_server")
def _assert_day_offset(self, flag):
options = _get_options([flag, '5'])
eq_(options['day_offset'], 5)
def _assert_day_range(self, flag):
options = _get_options([flag, '10'])
eq_(options['day_range'], 10)
def _assert_rules_directory(self, flag):
options = _get_options([flag, 'path_to_rules_directory'])
eq_(options['rules_directory'], 'path_to_rules_directory')
def _assert_settings(self, flag):
options = _get_options([flag, 'path_to_settings_file'])
eq_(options['settings_file'], 'path_to_settings_file')
def _assert_disco_debug(self, flag):
options = _get_options([flag])
eq_(options['disco_debug'], True)
def _assert_day_range(self, flag):
options, _ = _get_options([flag, "10"])
eq_(options["day_range"], 10)
def _assert_immediate_rule(self, flag):
options, _ = _get_options([flag, "some_module.some_rule"])
eq_(options["immediate_rule"], "some_module.some_rule")
def _assert_result_tag(self, flag):
options = _get_options([flag, 'some_result_tag'])
eq_(options['result_tag'], 'some_result_tag')
def _assert_server(self, flag):
options, _ = _get_options([flag, "some_server"])
eq_(options["server"], "some_server")
def _assert_settings(self, flag):
options, _ = _get_options([flag, "path_to_settings_file"])
eq_(options["settings_file"], "path_to_settings_file")
def _assert_day_offset(self, flag):
options, _ = _get_options([flag, "5"])
eq_(options["day_offset"], 5)
def _assert_result_tag(self, flag):
options, _ = _get_options([flag, "some_result_tag"])
eq_(options["result_tag"], "some_result_tag")
def _assert_day_start(self, flag):
options = _get_options([flag, '2012-12-01'])
eq_(options['day_start'], date(2012, 12, 1))
def _assert_disco_debug(self, flag):
options, _ = _get_options([flag])
eq_(options["disco_debug"], True)
def _assert_immediate_rule(self, flag):
options = _get_options([flag, 'some_module.some_rule'])
eq_(options['immediate_rule'], 'some_module.some_rule')
def _assert_day_start(self, flag):
options, _ = _get_options([flag, "2012-12-01"])
eq_(options["day_start"], date(2012, 12, 1))
def _assert_profile(self, flag):
options, _ = _get_options([flag])
eq_(options["profile"], True)
def _assert_profile(self, flag):
options = _get_options([flag])
eq_(options['profile'], True)
def _assert_server(self, flag):
options = _get_options([flag, 'some_server'])
eq_(options['server'], 'some_server')
def _assert_rules_directory(self, flag):
options, _ = _get_options([flag, "path_to_rules_directory"])
eq_(options["rules_directory"], "path_to_rules_directory")
def _assert_force(self, flag):
options = _get_options([flag])
eq_(options['force'], True)
def _assert_force(self, flag):
options, _ = _get_options([flag])
eq_(options["force"], True)