From 3850d3dffba9ae83e1eb55a0e1b224b7c3a86ca4 Mon Sep 17 00:00:00 2001 From: Christophe de Vienne Date: Sat, 15 Oct 2011 21:53:11 +0200 Subject: [PATCH] Fix a test --- wsme/tests/test_controller.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/wsme/tests/test_controller.py b/wsme/tests/test_controller.py index 81182fa..d597951 100644 --- a/wsme/tests/test_controller.py +++ b/wsme/tests/test_controller.py @@ -75,10 +75,12 @@ class TestController(unittest.TestCase): assert len(r.protocols) == 0 r.addprotocol('dummy') - assert r.protocols['dummy'] + assert len(r.protocols) == 1 + assert r.protocols[0].__class__ == DummyProtocol r = WSRoot(['dummy']) - assert r.protocols['dummy'] + assert len(r.protocols) == 1 + assert r.protocols[0].__class__ == DummyProtocol def test_scan_api(self): class NS(object):