commit 50b7b702be664795fb1c5699c52f1f0d9322e58e
Author: aagbsn <aagbsn(a)extc.org>
Date: Fri Nov 22 17:39:37 2013 +0000
Move exceptions into ooni/errors.py
---
ooni/errors.py | 9 +++++++++
ooni/nettest.py | 23 +++++++----------------
ooni/oonicli.py | 2 +-
3 files changed, 17 insertions(+), 17 deletions(-)
diff --git a/ooni/errors.py b/ooni/errors.py
index a124487..c167bbe 100644
--- a/ooni/errors.py
+++ b/ooni/errors.py
@@ -195,6 +195,15 @@ class NetTestNotFound(Exception):
class MissingRequiredOption(Exception):
pass
+class FailureToLoadNetTest(Exception):
+ pass
+
+class NoPostProcessor(Exception):
+ pass
+
+class InvalidOption(Exception):
+ pass
+
def get_error(error_key):
if error_key == 'test-helpers-key-missing':
return CouldNotFindTestHelper
diff --git a/ooni/nettest.py b/ooni/nettest.py
index b6dac29..c813c37 100644
--- a/ooni/nettest.py
+++ b/ooni/nettest.py
@@ -58,7 +58,7 @@ def loadNetTestString(net_test_string):
test_cases.extend(get_test_methods(item))
if not test_cases:
- raise NoTestCasesFound
+ raise e.NoTestCasesFound
return test_cases
@@ -72,7 +72,7 @@ def loadNetTestFile(net_test_file):
test_cases.extend(get_test_methods(item))
if not test_cases:
- raise NoTestCasesFound
+ raise e.NoTestCasesFound
return test_cases
@@ -351,7 +351,7 @@ class NetTestLoader(object):
test_cases.extend(self._get_test_methods(item))
if not test_cases:
- raise NoTestCasesFound
+ raise e.NoTestCasesFound
self.setupTestCases(test_cases)
@@ -365,7 +365,7 @@ class NetTestLoader(object):
test_cases.extend(self._get_test_methods(item))
if not test_cases:
- raise NoTestCasesFound
+ raise e.NoTestCasesFound
self.setupTestCases(test_cases)
@@ -678,7 +678,7 @@ class NetTestCase(object):
anything that gets written to the object self.report[] will be added to
the final test report.
"""
- raise NoPostProcessor
+ raise e.NoPostProcessor
def inputProcessor(self, filename):
"""
@@ -758,23 +758,14 @@ class NetTestCase(object):
for option in self.localOptions:
if option not in self.usageOptions():
if not self.inputFile or option not in self.inputFile:
- raise InvalidOption
+ raise e.InvalidOption
def _checkRequiredOptions(self):
for required_option in self.requiredOptions:
log.debug("Checking if %s is present" % required_option)
if required_option not in self.localOptions or \
self.localOptions[required_option] == None:
- raise MissingRequiredOption(required_option)
+ raise e.MissingRequiredOption(required_option)
def __repr__(self):
return "<%s inputs=%s>" % (self.__class__, self.inputs)
-
-class FailureToLoadNetTest(Exception):
- pass
-class NoPostProcessor(Exception):
- pass
-class InvalidOption(Exception):
- pass
-class MissingRequiredOption(Exception):
- pass
diff --git a/ooni/oonicli.py b/ooni/oonicli.py
index 8622e09..c722c52 100644
--- a/ooni/oonicli.py
+++ b/ooni/oonicli.py
@@ -16,7 +16,7 @@ from ooni.settings import config
from ooni.director import Director
from ooni.deck import Deck, nettest_to_path
from ooni.reporter import YAMLReporter, OONIBReporter
-from ooni.nettest import NetTestLoader, MissingRequiredOption
+from ooni.nettest import NetTestLoader
from ooni.utils import log, checkForRoot