Refator server tests to use flow.FlowMaster and flow.State

1 files changed, 12 insertions, 14 deletions

diff --git a/test/test_proxy.py b/test/test_proxy.py
index 08b3634f..d4d275b5 100644
--- a/test/test_proxy.py
+++ b/test/test_proxy.py
@@ -5,21 +5,19 @@ from libmproxy import proxy, flow
 import tutils
 
 
-class TestProxyError:
-    def test_simple(self):
-        p = proxy.ProxyError(111, "msg")
-        assert repr(p)
+def test_proxy_error():
+    p = proxy.ProxyError(111, "msg")
+    assert str(p)
 
 
-class TestAppRegistry:
-    def test_add_get(self):
-        ar = proxy.AppRegistry()
-        ar.add("foo", "domain", 80)
+def test_app_registry():
+    ar = proxy.AppRegistry()
+    ar.add("foo", "domain", 80)
 
-        r = tutils.treq()
-        r.host = "domain"
-        r.port = 80
-        assert ar.get(r)
+    r = tutils.treq()
+    r.host = "domain"
+    r.port = 80
+    assert ar.get(r)
 
-        r.port = 81
-        assert not ar.get(r)
+    r.port = 81
+    assert not ar.get(r)