def _make_agent_data(): """Create generate data to post to API server""" # Initialize key variables config = Config() polling_interval = 60 pattoo_agent_program = 1 pattoo_agent_polled_target = 2 pattoo_key = '3' pattoo_value = 4 # We want to make sure we get a different AgentID each time filename = files.agent_id_file(pattoo_agent_program, config) if os.path.isfile(filename) is True: os.remove(filename) # Setup AgentPolledData apd = AgentPolledData(pattoo_agent_program, polling_interval) # Initialize TargetDataPoints ddv = TargetDataPoints(pattoo_agent_polled_target) # Setup DataPoint data_type = DATA_INT variable = DataPoint(pattoo_key, pattoo_value, data_type=data_type) # Add data to TargetDataPoints ddv.add(variable) # Create a result apd.add(ddv) # Return agent data return apd
def test_agent_id_file(self): """Testing function agent_id_file.""" # Test filename = files._File(self.config) expected = filename.agent_id(self.prefix) result = files.agent_id_file(self.prefix, self.config) self.assertEqual(result, expected)
def test_get_agent_id(self): """Testing method or function named get_agent_id.""" # Test. Agent_id shouldn't change agent_name = random() expected = files.get_agent_id(agent_name, self.config) result = files.get_agent_id(agent_name, self.config) self.assertEqual(result, expected) # Delete the file, make sure a new ID is generated filename = files.agent_id_file(agent_name, self.config) os.remove(filename) _expected = files.get_agent_id(agent_name, self.config) _result = files.get_agent_id(agent_name, self.config) self.assertEqual(_result, _expected) self.assertNotEqual(expected, _expected)
def create_cache(): """Testing method / function records.""" # Initialize key variables config = ServerConfig() polling_interval = 20 cache_directory = config.agent_cache_directory(PATTOO_API_AGENT_NAME) result = { 'pattoo_agent_program': data.hashstring(str(random())), 'pattoo_agent_polled_target': socket.getfqdn(), 'pattoo_key': data.hashstring(str(random())), 'pattoo_value': round(uniform(1, 100), 5), 'pattoo_agent_hostname': socket.getfqdn() } # We want to make sure we get a different AgentID each time filename = files.agent_id_file( result['pattoo_agent_program'], config) if os.path.isfile(filename) is True: os.remove(filename) result['pattoo_agent_id'] = files.get_agent_id( result['pattoo_agent_program'], config) # Setup AgentPolledData apd = AgentPolledData(result['pattoo_agent_program'], polling_interval) # Initialize TargetDataPoints ddv = TargetDataPoints(result['pattoo_agent_hostname']) # Setup DataPoint data_type = DATA_INT variable = DataPoint( result['pattoo_key'], result['pattoo_value'], data_type=data_type) # Add data to TargetDataPoints ddv.add(variable) # Write data to cache apd.add(ddv) cache_dict = converter.posting_data_points( converter.agentdata_to_post(apd)) cache_file = '{}{}cache_test.json'.format(cache_directory, os.sep) with open(cache_file, 'w') as _fp: json.dump(cache_dict, _fp) return result