def test_dispatch(self): c = Channel(None, None, {}) frame = mock() frame.class_id = 32 klass = mock() c._class_map[32] = klass expect(klass.dispatch).args(frame) c.dispatch(frame) frame.class_id = 33 assert_raises(Channel.InvalidClass, c.dispatch, frame)