From 7b6af1f6c0d5338d0492d5897be65d4433248f14 Mon Sep 17 00:00:00 2001 From: Tom Cobb Date: Tue, 18 Oct 2016 13:25:46 +0100 Subject: [PATCH] Added process get_block test --- tests/test_core/test_process.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_core/test_process.py b/tests/test_core/test_process.py index 66c6d7460..e802e7557 100644 --- a/tests/test_core/test_process.py +++ b/tests/test_core/test_process.py @@ -37,6 +37,16 @@ def test_add_block(self): p.add_block(b) self.assertEqual(p._blocks["name"], b) + def test_get_block(self): + p = Process("proc", MagicMock()) + b1 = p.get_block("name1") + self.assertEqual(b1.status, "Waiting for connection...") + self.assertEqual(p.get_block("name1"), b1) + b2 = Block() + b2.set_parent(p, "name2") + p.add_block(b2) + self.assertEqual(p.get_block("name2"), b2) + def test_add_block_calls_handle(self): s = SyncFactory("sched") p = Process("proc", s)