def test_update_block_list(self): p = Process("proc", MagicMock()) p.q.reset_mock() p.update_block_list("cc", ["myblock"]) request = BlockList(client_comms="cc", blocks=["myblock"]) p.q.put.assert_called_once_with(request) self.assertEqual(p._client_comms, {}) p._handle_block_list(request) self.assertEqual(p._client_comms, dict(cc=["myblock"])) self.assertEqual(p.process_block.remoteBlocks.value, ["myblock"]) self.assertEqual(p.get_client_comms("myblock"), "cc")