def _insert_agent_to_stat_agent(self):
agent = StatAgent()
agent.name = 'test_agent'
self.add_me(agent)
return agent.name, agent.id
def _insert_agent_to_stat_agent(self, agent_name=None):
agent_name = agent_name if agent_name else 'Agent/1234'
agent = StatAgent()
agent.name = agent_name
self.add_me(agent)
return agent.name, agent.id
def _insert_agent_to_stat_agent(self):
agent = StatAgent()
agent.name = 'test_agent'
agent.tenant_uuid = self.default_tenant.uuid
self.add_me(agent)
return agent.name, agent.id
def _insert_agent_to_stat_agent(self):
agent = StatAgent()
agent.name = 'test_agent'
self.session.begin()
self.session.add(agent)
self.session.commit()
return agent.name, agent.id
def _insert_agent_to_stat_agent(self):
agent = StatAgent()
agent.name = 'test_agent'
self.session.begin()
self.session.add(agent)
self.session.commit()
return agent.name, agent.id
def _insert_agent_to_stat_agent(self, agent_name=None, tenant_uuid=None):
agent_name = agent_name if agent_name else 'Agent/1234'
agent = StatAgent()
agent.name = agent_name
agent.tenant_uuid = tenant_uuid or self.default_tenant.uuid
self.add_me(agent)
return agent.name, agent.id
def _insert_agent(self, name):
agent = StatAgent()
agent.name = name
self.session.begin()
self.session.add(agent)
self.session.commit()
return agent
def _insert_agent(self,
name,
tenant_uuid=None,
agent_id=None,
deleted=False):
agent = StatAgent()
agent.name = name
agent.tenant_uuid = tenant_uuid or self.default_tenant.uuid
agent.agent_id = agent_id
agent.deleted = deleted
self.add_me(agent)
return agent
def _create_missing_agents(session, confd_agents_by_name):
new_agent_names = set(confd_agents_by_name.keys())
db_agent_query = session.query(StatAgent).filter(
StatAgent.deleted.is_(False))
old_agent_names = set(agent.name for agent in db_agent_query.all())
missing_agents = list(new_agent_names - old_agent_names)
for agent_name in missing_agents:
agent = confd_agents_by_name[agent_name]
new_agent = StatAgent()
new_agent.name = agent_name
new_agent.tenant_uuid = agent['tenant_uuid']
new_agent.agent_id = agent['id']
new_agent.deleted = False
session.add(new_agent)
session.flush()
