def test_store_existing_agent(self, data_svc):
agent = Agent(paw='123', sleep_min=2, group='red', sleep_max=8, watchdog=0, executors=['cmd', 'test'],
platform='windows', trusted=True, pending_contact='HTML')
agent.store(data_svc.ram)
assert len(data_svc.ram['agents']) == 1
agent.group = "blue"
agent.trusted = False
agent.sleep_min = 1
agent.sleep_max = 5
agent.watchdog = 1
agent.pending_contact = "HTTP"
stored_agent = agent.store(data_svc.ram)
assert len(data_svc.ram['agents']) == 1
assert stored_agent.schema.dump(stored_agent) == agent.schema.dump(agent)
def test_store_new_agent(self, data_svc):
agent = Agent(paw='123',
sleep_min=2,
sleep_max=8,
watchdog=0,
executors=['cmd', 'test'],
platform='windows')
stored_agent = agent.store(data_svc.ram)
assert len(data_svc.ram['agents']) == 1
assert agent in data_svc.ram['agents']
assert stored_agent == agent
