mirror of
https://github.com/paulscherrerinstitute/sf_daq_broker.git
synced 2026-05-03 21:04:14 +02:00
Dmitry Ozerov
103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
import unittest
|
|
from multiprocessing import Process, Manager
|
|
from time import sleep
|
|
|
|
import bottle
|
|
import json
|
|
from bsread import PUB
|
|
|
|
from sf_databuffer_writer import config
|
|
from sf_databuffer_writer.broker_manager import StreamRequestSender
|
|
from sf_databuffer_writer.utils import get_writer_request
|
|
|
|
|
|
class TestStreamRequestSender(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.manager = Manager()
|
|
self.data_container = self.manager.dict()
|
|
|
|
def http_server(data):
|
|
|
|
app = bottle.Bottle()
|
|
|
|
@app.put("/notify")
|
|
def print_request():
|
|
data["request"] = bottle.request.json
|
|
|
|
bottle.run(app=app, host="localhost", port=10200)
|
|
|
|
self.http_process = Process(target=http_server, args=(self.data_container,))
|
|
self.http_process.start()
|
|
sleep(0.5)
|
|
|
|
def tearDown(self):
|
|
self.http_process.terminate()
|
|
|
|
def test_write_request(self):
|
|
|
|
request_sender = StreamRequestSender(12000, 100, 100, PUB, config.DEFAULT_EPICS_WRITER_URL)
|
|
|
|
channels = ["channel_1", "channel_2", "channel_3:FPICTURE"]
|
|
|
|
start_pulse_id = 100
|
|
stop_pulse_id = 120
|
|
|
|
parameters = {"general/created": "test",
|
|
"general/user": "tester",
|
|
"general/process": "test_process",
|
|
"general/instrument": "mac",
|
|
"output_file": "test.h5"}
|
|
|
|
writer_request = get_writer_request(channels, parameters, start_pulse_id, stop_pulse_id)
|
|
|
|
request_sender.send(writer_request)
|
|
|
|
sleep(0.5)
|
|
|
|
self.assertTrue("request" in self.data_container)
|
|
data = self.data_container["request"]
|
|
|
|
self.assertTrue("range" in data)
|
|
self.assertTrue("startPulseId" in data["range"])
|
|
self.assertTrue("endPulseId" in data["range"])
|
|
|
|
self.assertTrue("parameters" in data)
|
|
self.assertTrue("output_file" in data["parameters"])
|
|
|
|
self.assertTrue("data_api_request" in writer_request)
|
|
data_api_request = json.loads(writer_request["data_api_request"])
|
|
|
|
self.assertTrue("channels" in data_api_request)
|
|
for channel_name, backend in ((x["name"], x["backend"])
|
|
for x in data_api_request["channels"]):
|
|
|
|
if channel_name.endswith(":FPICTURE"):
|
|
self.assertEqual(backend, config.IMAGE_BACKEND)
|
|
else:
|
|
self.assertEqual(backend, config.DATA_BACKEND)
|
|
|
|
self.assertListEqual(channels, [x["name"] for x in data_api_request["channels"]])
|
|
|
|
def test_write_request_with_channels(self):
|
|
channels = ["channel_1", "channel_2", "channel_3:FPICTURE"]
|
|
override_channels = ["override"]
|
|
|
|
parameters = {"general/created": "test",
|
|
"general/user": "tester",
|
|
"general/process": "test_process",
|
|
"general/instrument": "mac",
|
|
"output_file": "test.h5",
|
|
"channels": ["override"]}
|
|
|
|
writer_request = get_writer_request(channels, parameters, 0, 100)
|
|
|
|
self.assertTrue("data_api_request" in writer_request)
|
|
data_api_request = json.loads(writer_request["data_api_request"])
|
|
|
|
self.assertListEqual(override_channels, [x["name"] for x in data_api_request["channels"]])
|
|
|
|
|
|
|
|
|