Example #1
0
    def setUpClass(cls):
        cls.pod_folder = prepare_agent_test_env()
        _logger.info("Temp POD folder: %s", cls.pod_folder)

        from ava.runtime.config import settings
        from ava.core.agent import Agent

        settings['DEBUG'] = True
        settings['TEST'] = True
        os.environ.setdefault(AVA_SWARM_SECRET, swarm_secret)
        os.environ.setdefault(AVA_AGENT_SECRET, agent_secret)
        AgentTest.agent = Agent(None, None)
        gevent.spawn(AgentTest.agent.run)
        agent_running.wait(10)
Example #2
0
    def setUpClass(cls):
        cls.pod_folder = prepare_agent_test_env()
        _logger.info("Temp POD folder: %s", cls.pod_folder)

        from ava.runtime.config import settings
        from ava.core.agent import Agent

        settings['DEBUG'] = True
        settings['TEST'] = True
        os.environ.setdefault(AVA_SWARM_SECRET, swarm_secret)
        os.environ.setdefault(AVA_AGENT_SECRET, agent_secret)
        AgentTest.agent = Agent(None, None)
        gevent.spawn(AgentTest.agent.run)
        agent_running.wait(10)
Example #3
0
def agent(request):
    from ava.runtime.config import settings
    from ava.core.agent import Agent

    pod_folder = prepare_agent_test_env()
    settings['GENERAL']['TEST'] = True
    os.environ.setdefault(AVA_SWARM_SECRET, swarm_secret)
    os.environ.setdefault(AVA_AGENT_SECRET, agent_secret)
    _agent = Agent(None, None)
    agent_greenlet = gevent.spawn(_agent.run)
    agent_running.wait(10)

    def teardown_agent():
        if not agent:
            return
        _agent.interrupted = True
        agent_stopped.wait(10)
        shutil.rmtree(pod_folder)

    request.addfinalizer(teardown_agent)
    return _agent
Example #4
0
def agent(request):
    from ava.runtime.config import settings
    from ava.core.agent import Agent

    pod_folder = prepare_agent_test_env()
    settings['GENERAL']['TEST'] = True
    os.environ.setdefault(AVA_SWARM_SECRET, swarm_secret)
    os.environ.setdefault(AVA_AGENT_SECRET, agent_secret)
    _agent = Agent(None, None)
    gevent.spawn(_agent.run)
    agent_running.wait(10)

    def teardown_agent():
        if not agent:
            return
        _agent.interrupted = True
        agent_stopped.wait(10)
        shutil.rmtree(pod_folder)

    request.addfinalizer(teardown_agent)
    return _agent