From 35204ecb1cb6221e0baee8b81e2165a0458fae60 Mon Sep 17 00:00:00 2001
From: Shadab Zafar <dufferzafar0@gmail.com>
Date: Wed, 9 Mar 2016 23:51:29 +0530
Subject: Move har_extractor tests to the examples file

---
 test/mitmproxy/test_examples.py      | 24 +++++++++++++++++++++++
 test/mitmproxy/test_har_extractor.py | 37 ------------------------------------
 2 files changed, 24 insertions(+), 37 deletions(-)
 delete mode 100644 test/mitmproxy/test_har_extractor.py

(limited to 'test')

diff --git a/test/mitmproxy/test_examples.py b/test/mitmproxy/test_examples.py
index a6fed38b..2f29f9dd 100644
--- a/test/mitmproxy/test_examples.py
+++ b/test/mitmproxy/test_examples.py
@@ -1,4 +1,5 @@
 import glob
+import json
 import os
 from contextlib import contextmanager
 
@@ -114,3 +115,26 @@ def test_redirect_requests():
     with example("redirect_requests.py") as ex:
         ex.run("request", flow)
         assert flow.request.host == "mitmproxy.org"
+
+
+def test_har_extractor():
+    with tutils.raises(script.ScriptException):
+        with example("har_extractor.py") as ex:
+            pass
+
+    times = dict(
+        timestamp_start=746203272,
+        timestamp_end=746203272,
+    )
+
+    flow = tutils.tflow(
+        req=netutils.treq(**times),
+        resp=netutils.tresp(**times)
+    )
+
+    with example("har_extractor.py -") as ex:
+        ex.run("response", flow)
+
+        with open(tutils.test_data.path("data/har_extractor.har")) as fp:
+            test_data = json.load(fp)
+            assert json.loads(ex.ctx.HARLog.json()) == test_data["test_response"]
diff --git a/test/mitmproxy/test_har_extractor.py b/test/mitmproxy/test_har_extractor.py
deleted file mode 100644
index 7838f713..00000000
--- a/test/mitmproxy/test_har_extractor.py
+++ /dev/null
@@ -1,37 +0,0 @@
-import json
-import netlib.tutils
-from . import tutils
-
-from examples import har_extractor
-
-
-class Context(object):
-    pass
-
-
-trequest = netlib.tutils.treq(
-    timestamp_start=746203272,
-    timestamp_end=746203272,
-)
-
-tresponse = netlib.tutils.tresp(
-    timestamp_start=746203272,
-    timestamp_end=746203272,
-)
-
-
-def test_start():
-    tutils.raises(ValueError, har_extractor.start, Context(), [])
-
-
-def test_response():
-    ctx = Context()
-    ctx.HARLog = har_extractor._HARLog([])
-    ctx.seen_server = set()
-
-    fl = tutils.tflow(req=trequest, resp=tresponse)
-    har_extractor.response(ctx, fl)
-
-    with open(tutils.test_data.path("data/har_extractor.har")) as fp:
-        test_data = json.load(fp)
-        assert json.loads(ctx.HARLog.json()) == test_data["test_response"]
-- 
cgit v1.2.3