def setup(self): sample_config = DotDict() sample_config.configfile = os.path.join(os.path.dirname(__file__), '../../loginput/index.conf') OptionParser.parse_args = mock.Mock(return_value=(sample_config, {})) from loginput import index self.application = index.application super(LoginputTestSuite, self).setup()
def test_complex_init(self): original_dct = { 'details': { 'key1': 'value1' } } dct = DotDict(original_dct) assert dct.details == {'key1': 'value1'} assert dct.details.key1 == 'value1'
def onMessage(self, message, metadata): if 'source' not in message: return (message, metadata) if not message['source'] == 'guardduty': return (message, metadata) # reformat the date fields to iosformat for date_key in self.date_keys: if key_exists(date_key, message): message = self.convert_key_date_format(date_key, message) # convert the dict to a dot dict for saner deep key/value processing message=DotDict(message) # pull out the likely source IP address for ipaddress_key in self.ipaddress_keys: if 'sourceipaddress' not in message['details'].keys(): if key_exists(ipaddress_key,message): message.details.sourceipaddress = message.get(ipaddress_key) # recovert the message back to a plain dict return (dict(message), metadata)
def setup(self): super(TestEsworkerSNSSQS, self).setup() mq_conn = 'abc' task_queue = 'example-logs-mozdef' es_connection = self.es_client options = DotDict({ "esbulksize": 0, "mozdefhostname": "unittest.hostname", "taskexchange": task_queue, 'plugincheckfrequency': 120, }) self.consumer = taskConsumer(mq_conn, task_queue, es_connection, options)
def test_complex_get(self): original_dct = { 'details': { 'key1': 'value1', 'subkey': { 'subkey': 'subvalue' } } } dct = DotDict(original_dct) assert dct.get('does.not.exist') == None assert dct.get('details') == {'key1': 'value1','subkey': {'subkey': 'subvalue'}} assert dct.get('details.key1') == 'value1' assert dct.get('details.subkey') == {'subkey':'subvalue'} assert dct.get('details.subkey.subkey') == 'subvalue'
def parse_config(self): default_config = os.path.join(os.path.dirname(__file__), "config.conf") options = DotDict() options.configfile = default_config options.esservers = list( getConfig('esservers', 'http://localhost:9200', options.configfile).split(',')) options.alertExchange = getConfig('alertexchange', 'alerts', options.configfile) options.queueName = getConfig('alertqueuename', 'alertBot', options.configfile) options.alerttopic = getConfig('alerttopic', 'mozdef.*', options.configfile) options.mquser = getConfig('mquser', 'guest', options.configfile) options.mqalertserver = getConfig('mqalertserver', 'localhost', options.configfile) options.mqpassword = getConfig('mqpassword', 'guest', options.configfile) options.mqport = getConfig('mqport', 5672, options.configfile) options.mqack = getConfig('mqack', True, options.configfile) self.options = options
import sys import os sys.path.append(os.path.join(os.path.dirname(__file__), "../")) from http_test_suite import HTTPTestSuite sys.path.append(os.path.join(os.path.dirname(__file__), "../../lib")) from utilities.dot_dict import DotDict import mock from configlib import OptionParser sample_config = DotDict() sample_config.configfile = os.path.join(os.path.dirname(__file__), '../../rest/index.conf') OptionParser.parse_args = mock.Mock(return_value=(sample_config, {})) sys.path.append(os.path.join(os.path.dirname(__file__), "../../")) class RestTestSuite(HTTPTestSuite): def setup(self): from rest import index self.application = index.application super(RestTestSuite, self).setup()
def parse_config_file(): global CONFIG_FILE_CONTENTS try: CONFIG_FILE_CONTENTS except NameError: default_config = os.path.join(os.path.dirname(__file__), "config.conf") options = DotDict() options.configfile = default_config options.esservers = list( getConfig('esservers', 'http://localhost:9200', options.configfile).split(',')) options.alertExchange = getConfig('alertexchange', 'alerts', options.configfile) options.queueName = getConfig('alertqueuename', 'alertBot', options.configfile) options.alerttopic = getConfig('alerttopic', 'mozdef.*', options.configfile) options.mquser = getConfig('mquser', 'guest', options.configfile) options.mqalertserver = getConfig('mqalertserver', 'localhost', options.configfile) options.mqpassword = getConfig('mqpassword', 'guest', options.configfile) options.mqport = getConfig('mqport', 5672, options.configfile) options.mqack = getConfig('mqack', True, options.configfile) CONFIG_FILE_CONTENTS = options return CONFIG_FILE_CONTENTS
def test_basic_init(self): dct = DotDict({'key1': 'value1', 'key2': 'value2'}) assert sorted(dct.keys()) == sorted(['key1', 'key2']) assert dct.key1 == 'value1' assert dct.key2 == 'value2'
def test_nonexisting_key(self): dct = DotDict() with pytest.raises(KeyError): dct.abcd
def test_blank_init(self): dct = DotDict() assert dct.keys() == []