commit 87b56850d18a39dcb020d3c3e3d45dce2a374d0b Author: Sean Robinson <seankrobinson@gmail.com> Date: Sat Jan 5 14:16:04 2013 -0700 Add unit tests for Controller.get_version() Signed-off-by: Sean Robinson <seankrobinson@gmail.com> --- test/unit/control/controller.py | 45 +++++++++++++++++++++++++++++++++++++++ 1 files changed, 45 insertions(+), 0 deletions(-) diff --git a/test/unit/control/controller.py b/test/unit/control/controller.py index 3b8a52d..44b7543 100644 --- a/test/unit/control/controller.py +++ b/test/unit/control/controller.py @@ -21,6 +21,51 @@ class TestControl(unittest.TestCase): def tearDown(self): mocking.revert_mocking() + def test_get_version(self): + """ + Exercises the get_version() method. + """ + + try: + # Use one version for first check. + version_2_1 = "0.2.1.32" + version_2_1_object = stem.version.Version(version_2_1) + mocking.mock_method(Controller, "get_info", mocking.return_value(version_2_1)) + + # Return a version with a cold cache. + self.assertEqual(version_2_1_object, self.controller.get_version()) + + # Use a different version for second check. + version_2_2 = "0.2.2.39" + version_2_2_object = stem.version.Version(version_2_2) + mocking.mock_method(Controller, "get_info", mocking.return_value(version_2_2)) + + # Return a version with a hot cache, so it will be the old version. + self.assertEqual(version_2_1_object, self.controller.get_version()) + + # Turn off caching. + self.controller._is_caching_enabled = False + # Return a version without caching, so it will be the new version. + self.assertEqual(version_2_2_object, self.controller.get_version()) + + # Raise an exception in the get_info() call. + mocking.mock_method(Controller, "get_info", mocking.raise_exception(InvalidArguments)) + + # Get a default value when the call fails. + self.assertEqual("default returned", + self.controller.get_version(default = "default returned")) + + # No default value, accept the error. + self.assertRaises(InvalidArguments, self.controller.get_version) + + # Give a bad version. The stem.version.Version ValueError should bubble up. + version_A_42 = "0.A.42.spam" + mocking.mock_method(Controller, "get_info", mocking.return_value(version_A_42)) + self.assertRaises(ValueError, self.controller.get_version) + finally: + # Turn caching back on before we leave. + self.controller._is_caching_enabled = True + def test_event_listening(self): """ Exercises the add_event_listener and remove_event_listener methods.