[tor-commits] [ooni-probe/master] Add unittests for some methods of runner

[art at torproject.org]

commit 2dc8ae91713fdd73ac1e835dcb191714c2e93593
Author: Arturo Filastò <art at fuffa.org>
Date:   Fri Nov 23 19:44:56 2012 +0100

    Add unittests for some methods of runner
---
 tests/test_runner.py |   78 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 78 insertions(+), 0 deletions(-)

diff --git a/tests/test_runner.py b/tests/test_runner.py
new file mode 100644
index 0000000..fac4cd7
--- /dev/null
+++ b/tests/test_runner.py
@@ -0,0 +1,78 @@
+from twisted.trial import unittest
+
+
+from ooni.inputunit import InputUnit
+from ooni.nettest import NetTestCase
+from ooni.reporter import OReporter
+
+from ooni.runner import loadTestsAndOptions, runTestCasesWithInputUnit
+
+
+class DummyTestCase(NetTestCase):
+    def test_a(self):
+        self.report['bar'] = 'bar'
+    def test_b(self):
+        self.report['foo'] = 'foo'
+
+class DummyTestCasePP(DummyTestCase):
+    def postProcessor(self, report):
+        self.report['antani'] = 'sblinda'
+
+class DummyReporter(OReporter):
+    dummy_report = []
+    def createReport(self, options):
+        pass
+
+    def writeReportEntry(self, entry):
+        self.dummy_report.append(entry)
+
+class TestRunner(unittest.TestCase):
+    def test_load_test_and_options(self):
+        input_unit = InputUnit([0,1,2,3,4])
+        cmd_line_options = {}
+        test_cases, options = loadTestsAndOptions([DummyTestCase],
+                cmd_line_options)
+        self.assertEqual(test_cases[0][1], 'test_b')
+        self.assertEqual(test_cases[1][1], 'test_a')
+
+    def test_run_testcase_with_input_unit(self):
+        oreporter = DummyReporter()
+        oreporter.dummy_report = []
+        def done(result):
+            report = oreporter.dummy_report
+            self.assertEqual(len(report), 5*2)
+            for idx, entry in enumerate(oreporter.dummy_report):
+                if idx % 2 == 0:
+                    self.assertEqual(entry['report']['foo'], 'foo')
+                else:
+                    self.assertEqual(entry['report']['bar'], 'bar')
+
+        input_unit = InputUnit([0,1,2,3,4])
+        cmd_line_options = {'collector': None}
+
+        test_cases, options = loadTestsAndOptions([DummyTestCase],
+                cmd_line_options)
+
+        d = runTestCasesWithInputUnit(test_cases, input_unit, oreporter)
+        d.addBoth(done)
+        return d
+
+    def test_with_post_processing(self):
+        oreporter = DummyReporter()
+        oreporter.dummy_report = []
+        def done(result):
+            report = oreporter.dummy_report
+            self.assertEqual(len(report), 3)
+            for entry in report:
+                if entry['test_name'] == 'summary':
+                    self.assertEqual(entry['report'], {'antani': 'sblinda'})
+
+        input_unit = InputUnit([None])
+        cmd_line_options = {'collector': None}
+
+        test_cases, options = loadTestsAndOptions([DummyTestCasePP],
+                cmd_line_options)
+
+        d = runTestCasesWithInputUnit(test_cases, input_unit, oreporter)
+        d.addBoth(done)
+        return d