def test_get_account_skey(self): config = {'account': {'skey': 'PASSWORD'}} Config.set_config(config) skey = Config.get_account_skey() self.assertEqual(skey, 'PASSWORD')
def create_tasks(server_to_writer): """ Create a pair of Producer-Consumer objects for each endpoint enabled within the account defined in config, or retrieve child accounts and do the same if the account is MSP. Return a list containing the asyncio tasks for running those objects. @param writer Dictionary mapping server ids to writer objects @return list of asyncio tasks for running the Producer and Consumer objects """ tasks = [] # Object with functions needed to utilize log API calls admin = create_admin( Config.get_account_ikey(), Config.get_account_skey(), Config.get_account_hostname()) # This is where functionality would be added to check if an account is MSP # (Config.account_is_msp), and then retrieve child accounts (ignoring those # in a blocklist) if the account is indeed MSP for mapping in Config.get_account_endpoint_server_mappings(): # Get the writer to be used for this set of endpoints writer = server_to_writer[mapping.get('server')] for endpoint in mapping.get('endpoints'): new_tasks = create_consumer_producer_pair(endpoint, writer, admin) tasks.extend(new_tasks) return tasks