def _insert_agent(self, aname, tenant_uuid=None): tenant_uuid = tenant_uuid or self.default_tenant.uuid a = StatAgent(name=aname, tenant_uuid=tenant_uuid) self.add_me(a) return a.name, a.id
def add_stat_agent(self, **kwargs): kwargs.setdefault('name', self._random_name()) kwargs.setdefault('tenant_uuid', self.default_tenant.uuid) stat_agent = StatAgent(**kwargs) self.add_me(stat_agent) return stat_agent
def _insert_agent(self, aname): a = StatAgent(name=aname) self.session.begin() self.session.add(a) self.session.commit() return a.name, a.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_stat_agent(self, **kwargs): session = self.Session() agent = StatAgent(**kwargs) session.add(agent) session.flush() stat_agent_id = agent.id session.commit() return stat_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_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()