Fix a test

2011-10-15 21:53:11 +02:00
parent 95d99bce68
commit 3850d3dffb
1 changed files with 4 additions and 2 deletions
--- 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):