Belle II Software  release-08-01-10
WorkerTestCase Class Reference
Inheritance diagram for WorkerTestCase:
Collaboration diagram for WorkerTestCase:

Public Member Functions

def setUp (self)
 
def start (self)
 
def tearDown (self)
 
def assertIsDown (self, name, timeout=5, minimum_delay=0.1)
 
def assertIsRunning (self, name)
 
def assertMonitoring (self, socket, search_key, search_value, timeout=10)
 
def assertIsAndGet (self, socket, message_type, final=True, router=False)
 
def assertIsMsgType (self, socket, message_type, final=True, router=False)
 
def assertNothingMore (self, socket)
 
def assertHasOutputFile (self, output_file, unlink=True, timeout=0.5, minimum_delay=0.1)
 
def assertNotHasOutputFile (self, output_file, timeout=0.5)
 

Static Public Member Functions

def get_free_port ()
 
def create_socket (port, socket_type=zmq.DEALER, identity="socket", bind=False)
 
def create_router_socket (port)
 
def send (socket, message_type, first_data=b"", second_data=b"", identity="")
 
def recv (socket)
 

Public Attributes

 needed_programs
 input_socket More...
 
 output_identity
 output_identity
 
 input_identity
 input_identity
 
 first_run_event_data
 some data
 
 second_run_event_data
 some data
 
 test_dir
 use a temporary folder for testing
 
 previous_dir
 remember current working directory
 
 started_programs
 dict for all started programs
 

Static Public Attributes

 event_data = open(basf2.find_file("daq/hbasf2/tests/out.raw"), "br").read()
 event_data
 
list extra_arguments = []
 extra arguments to pass to the worker script
 
 ctx = zmq.Context()
 The ZMQ context.
 

Private Member Functions

def _is_running (self, name)
 

Detailed Description

Test case baseclass to spawn a worker

Definition at line 16 of file test_worker.py.

Member Function Documentation

◆ _is_running()

def _is_running (   self,
  name 
)
privateinherited
Check if a given program is still running.

Definition at line 84 of file test_support.py.

◆ assertHasOutputFile()

def assertHasOutputFile (   self,
  output_file,
  unlink = True,
  timeout = 0.5,
  minimum_delay = 0.1 
)
inherited
Assert that - at least after the given timeout - the output file
is present. If unlink is set to True, remove the file after checking.

Definition at line 227 of file test_support.py.

◆ assertIsAndGet()

def assertIsAndGet (   self,
  socket,
  message_type,
  final = True,
  router = False 
)
inherited
Assert that the next message received on the socket has the given message type.
If final is set to True, also assert that there is no additional message on the socket.
Use router only for router sockets.

Definition at line 200 of file test_support.py.

◆ assertIsDown()

def assertIsDown (   self,
  name,
  timeout = 5,
  minimum_delay = 0.1 
)
inherited
Test helper to assert the given program has terminated - at least after timeout in seconds has passed.
Checks every "minimal_delay seconds.

Definition at line 93 of file test_support.py.

◆ assertIsMsgType()

def assertIsMsgType (   self,
  socket,
  message_type,
  final = True,
  router = False 
)
inherited
Deprecated copy of "assertIsAndGet".

Definition at line 215 of file test_support.py.

◆ assertIsRunning()

def assertIsRunning (   self,
  name 
)
inherited
Assert that a given program is still running.

Definition at line 108 of file test_support.py.

◆ assertMonitoring()

def assertMonitoring (   self,
  socket,
  search_key,
  search_value,
  timeout = 10 
)
inherited
Ask the given socket for a monitoring JSON and make sure, the value related to "search_key"
is set to "search_value" - at least after the given timeout.
The search key can should be in the form "<category>.<key>".

Definition at line 169 of file test_support.py.

◆ assertNotHasOutputFile()

def assertNotHasOutputFile (   self,
  output_file,
  timeout = 0.5 
)
inherited
Assert that after the timeout the given file is not present
(a.k.a. no process has created it)

Definition at line 245 of file test_support.py.

◆ assertNothingMore()

def assertNothingMore (   self,
  socket 
)
inherited
Assert that there is no pending message to be received on the socket.

Definition at line 221 of file test_support.py.

◆ create_router_socket()

def create_router_socket (   port)
staticinherited
Shortcut to create a ROUTER type socket with the typical parameters
binding to the given port.

Definition at line 140 of file test_support.py.

◆ create_socket()

def create_socket (   port,
  socket_type = zmq.DEALER,
  identity = "socket",
  bind = False 
)
staticinherited
Create and return a ZMQ socket with the given type and identity and
bind or connect it to localhost and the given port.

Definition at line 115 of file test_support.py.

◆ get_free_port()

def get_free_port ( )
staticinherited
Get a free port number by reusing ZMQ's function for this.

Definition at line 41 of file test_support.py.

◆ recv()

def recv (   socket)
staticinherited
Try to receive a message from the socket (or throw an assertion error if none comes after the set timeout
of the socket).

Definition at line 159 of file test_support.py.

◆ send()

def send (   socket,
  message_type,
  first_data = b"",
  second_data = b"",
  identity = "" 
)
staticinherited
Send a message consisting of the message type, the first and the second data
either to the identity if given or without identity if omitted.

Definition at line 148 of file test_support.py.

◆ setUp()

def setUp (   self)
Setup necessary sockets and programs

Reimplemented from HLTZMQTestCase.

Definition at line 24 of file test_worker.py.

24  def setUp(self):
25  """Setup necessary sockets and programs"""
26 
27  self.input_socket, input_port = self.create_router_socket(None)
28 
29  self.output_socket, output_port = self.create_router_socket(None)
30  # and set the list of necessary programs to use these sockets
31  self.needed_programs = {
32  "worker": [
33  "python3", basf2.find_file("daq/hbasf2/tests/passthrough.no_run_py"),
34  "--input", f"tcp://localhost:{input_port}",
35  "--output", f"tcp://localhost:{output_port}",
36  ] + self.extra_arguments
37  }
38  super().setUp()
39 

◆ start()

def start (   self)
start the needed sockets and send some hello messages

Definition at line 40 of file test_worker.py.

◆ tearDown()

def tearDown (   self)
inherited
Custom tearDown function to kill the started programs if still present
and remove the temporary folder again.

Definition at line 70 of file test_support.py.

Member Data Documentation

◆ needed_programs

needed_programs

input_socket

output_socket

Definition at line 31 of file test_worker.py.


The documentation for this class was generated from the following file: