def _handle_default_db_config(server):
server.db.create_all()
admin_user = get_admin_user()
fd, temp_auth_file = tempfile.mkstemp()
os.close(fd)
with open(temp_auth_file, 'w') as f:
yaml.dump(auth_dict, f)
try:
# We're mocking the AMQPManager, we aren't really using Rabbit here
default_tenant = create_default_user_tenant_and_roles(
admin_username=admin_user['username'],
admin_password=admin_user['password'],
amqp_manager=MagicMock(),
authorization_file_path=temp_auth_file)
default_tenant.rabbitmq_password = encrypt(
AMQPManager._generate_user_password())
finally:
os.remove(temp_auth_file)
utils.set_current_tenant(default_tenant)
def put_agent(self,
agent_name='agent_1',
instance_id='node_instance_1',
deployment_id='deployment_1'):
node_instance = self._get_or_create_node_instance(
instance_id, deployment_id)
agent = Agent(id=agent_name,
name=agent_name,
ip='127.0.0.1',
install_method='remote',
system='centos core',
version='4.5.5',
visibility='tenant',
state=AgentState.CREATING,
rabbitmq_username='******'.format(agent_name),
rabbitmq_password=encrypt(
AMQPManager._generate_user_password()),
rabbitmq_exchange=agent_name,
created_at=get_formatted_timestamp(),
updated_at=get_formatted_timestamp())
agent.node_instance = node_instance
return self.sm.put(agent)