from ocs.base import OpCode from ocs.testing import ( create_agent_runner_fixture, create_client_fixture, ) from integration.util import ( create_crossbar_fixture ) pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture('../agents/host_master/host_master.py', 'master-host-1', args=['--log-dir', os.path.join(os.getcwd(), 'log/')]) client = create_client_fixture('master-host-1') @pytest.mark.integtest def test_host_master_agent_master(wait_for_crossbar, run_agent, client): # Startup is always true, so let's check it's running resp = client.master.status() print(resp) assert resp.session['op_code'] == OpCode.RUNNING.value
import pytest import ocs from ocs.base import OpCode from ocs.testing import ( create_agent_runner_fixture, create_client_fixture, ) from integration.util import (create_crossbar_fixture) pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/influxdb_publisher/influxdb_publisher.py', 'influxagent-local') client = create_client_fixture('influxagent-local') @pytest.mark.integtest def test_influxdb_publisher_agent_record(wait_for_crossbar, run_agent, client): resp = client.record.start(test_mode=True) print(resp) assert resp.status == ocs.OK resp = client.record.wait(timeout=20) assert resp.status == ocs.OK assert resp.session['op_code'] == OpCode.SUCCEEDED.value
import ocs from ocs.base import OpCode from ocs.testing import create_agent_runner_fixture, create_client_fixture from integration.util import create_crossbar_fixture from socs.testing.device_emulator import create_device_emulator pytest_plugins = ("docker_compose") # Set the OCS_CONFIG_DIR so we read the local default.yaml file always os.environ['OCS_CONFIG_DIR'] = os.getcwd() run_agent = create_agent_runner_fixture( '../agents/lakeshore372/LS372_agent.py', 'ls372') client = create_client_fixture('LSASIM') wait_for_crossbar = create_crossbar_fixture() def build_init_responses(): values = { '*IDN?': 'LSCI,MODEL372,LSASIM,1.3', 'SCAN?': '01,1', 'INSET? A': '0,010,003,00,1', 'INNAME? A': 'Input A', 'INTYPE? A': '1,04,0,15,0,2', 'TLIMIT? A': '+0000', 'OUTMODE? 0': '2,6,1,0,0,001', 'HTRSET? 0': '+120.000,8,+0000.00,1', 'OUTMODE? 2': '4,16,1,0,0,001', 'HTRSET? 2': '+120.000,8,+0000.00,1'
from ocs.testing import ( create_agent_runner_fixture, create_client_fixture, ) from integration.util import (create_crossbar_fixture) from socs.testing.device_emulator import create_device_emulator pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/lakeshore240/LS240_agent.py', 'ls240_agent') client = create_client_fixture('LSA240S') initial_responses = { '*IDN?': 'LSCI,MODEL240,LSA240S,1.3', 'MODNAME?': 'LSA240S', 'INTYPE? 1': '1,1,0,0,1,1', 'INNAME? 1': 'Channel 1', 'INTYPE? 2': '1,1,0,0,1,1', 'INNAME? 2': 'Channel 2', 'INTYPE? 3': '1,1,0,0,1,1', 'INNAME? 3': 'Channel 3', 'INTYPE? 4': '1,1,0,0,1,1', 'INNAME? 4': 'Channel 4', 'INTYPE? 5': '1,1,0,0,1,1', 'INNAME? 5': 'Channel 5', 'INTYPE? 6': '1,1,0,0,1,1',
import pytest import ocs from ocs.base import OpCode from ocs.testing import ( create_agent_runner_fixture, create_client_fixture, ) from integration.util import (create_crossbar_fixture) pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/aggregator/aggregator_agent.py', 'aggregator-local') client = create_client_fixture('aggregator-local') @pytest.mark.dependency(depends=["so3g"], scope='session') @pytest.mark.integtest def test_aggregator_agent_record(wait_for_crossbar, run_agent, client): resp = client.record.start(test_mode=True) assert resp.status == ocs.OK resp = client.record.wait(timeout=20) assert resp.status == ocs.OK assert resp.session['op_code'] == OpCode.SUCCEEDED.value
from ocs.testing import ( create_agent_runner_fixture, create_client_fixture, ) from integration.util import ( create_crossbar_fixture, ) pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/fake_data/fake_data_agent.py', 'fake_data') client = create_client_fixture('fake-data-local') @pytest.mark.integtest def test_fake_data_agent_delay_task(wait_for_crossbar, run_agent, client): resp = client.delay_task(delay=0.01) # print(resp) assert resp.status == ocs.OK # print(resp.session) assert resp.session['op_code'] == OpCode.SUCCEEDED.value @pytest.mark.integtest def test_fake_data_agent_set_heartbeat(wait_for_crossbar, run_agent, client): resp = client.set_heartbeat(heartbeat=True) assert resp.status == ocs.OK
from integration.util import (create_crossbar_fixture) from socs.testing.device_emulator import create_device_emulator pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/hwp_rotation/rotation_agent.py', 'hwp_rotation_agent', args=['--log-dir', './logs/']) run_agent_idle = create_agent_runner_fixture( '../agents/hwp_rotation/rotation_agent.py', 'hwp_rotation_agent', args=['--mode', 'idle', '--log-dir', './logs/']) client = create_client_fixture('rotator') kikusui_emu = create_device_emulator({'SYST:REM': ''}, relay_type='tcp', port=2000) pid_emu = create_device_emulator({'*W02400000': 'W02\r'}, relay_type='tcp', port=2001) @pytest.mark.integtest def test_testing(wait_for_crossbar): """Just a quick test to make sure we can bring up crossbar.""" assert True # This ends up hanging for some reason that I can't figure out at the moment.
create_client_fixture, ) from integration.util import ( create_crossbar_fixture ) from ocs.base import OpCode pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture('../agents/registry/registry.py', 'registry', args=['--log-dir', os.path.join(os.getcwd(), 'log/')]) client = create_client_fixture('registry') @pytest.mark.integtest def test_registry_agent_main(wait_for_crossbar, run_agent, client): # Startup is always true, so let's check it's running resp = client.main.status() print(resp) assert resp.session['op_code'] == OpCode.RUNNING.value client.main.stop() resp = client.main.status() assert resp.session['op_code'] == OpCode.STOPPING.value
Parameters: data (str): Data string to package into telegram. Returns: str: The full telegram string that is emulating the response from the TC400. """ return chksum_msg('001' + '10' + '010' + '{:02d}'.format(len(data)) + data) wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/pfeiffer_tc400/pfeiffer_tc400_agent.py', 'tc400_agent') client = create_client_fixture('pfeifferturboA') emulator = create_device_emulator({}, relay_type='tcp') @pytest.mark.integtest def test_pfeiffer_tc400_init_lakeshore(wait_for_crossbar, emulator, run_agent, client): resp = client.init() print(resp) assert resp.status == ocs.OK print(resp.session) assert resp.session['op_code'] == OpCode.SUCCEEDED.value @pytest.mark.integtest def test_pfeiffer_tc400_turn_turbo_on(wait_for_crossbar, emulator, run_agent,
) from integration.util import ( create_crossbar_fixture ) from socs.testing.device_emulator import create_device_emulator pytest_plugins = ("docker_compose") wait_for_crossbar = create_crossbar_fixture() run_agent = create_agent_runner_fixture( '../agents/lakeshore425/LS425_agent.py', 'ls425_agent') run_agent_acq = create_agent_runner_fixture( '../agents/lakeshore425/LS425_agent.py', 'ls425_agent', args=['--mode', 'acq']) client = create_client_fixture('LS425') emulator = create_device_emulator({'*IDN?': 'LSCI,MODEL425,LSA425T,1.3'}, relay_type='serial') @pytest.mark.integtest def test_testing(wait_for_crossbar): """Just a quick test to make sure we can bring up crossbar.""" assert True @pytest.mark.integtest def test_ls425_init_lakeshore(wait_for_crossbar, emulator, run_agent, client): resp = client.init_lakeshore() # print(resp) assert resp.status == ocs.OK