class RunServerPipelineTest(unittest.TestCase):
def setUp(self):
self.receive = Receive()
self.j = Job([], [Edge(self.receive, MockPipelineElement())])
self.data = {'a': 1, 'b' : 2}
self.j.receive = lambda: self.data
def test_receive(self):
self.assertDictEqual(self.receive.out, {})
self.j.run_server_pipeline()
self.assertDictEqual(self.receive.out, {'results' : self.data})
def test_no_receivers(self):
j = Job([], [DacapoHarness() >> Print()])
with self.assertRaises(ValueError):
j.run_server_pipeline()
def test_empty_flow(self):
j = Job([], [])
self.assertEqual(j.run_server_pipeline(), None)
def test_empty_flow(self):
j = Job([], [])
self.assertEqual(j.run_server_pipeline(), None)
def test_no_receivers(self):
j = Job([], [DacapoHarness() >> Print()])
with self.assertRaises(ValueError):
j.run_server_pipeline()
