blob: f19e7e654d48aa6031f1dde3901542dfe3f4ec1a [file] [log] [blame]
# Copyright 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Test the cbuildbot_failures module."""
import os
import sys
sys.path.insert(0, os.path.abspath('%s/../..' % os.path.dirname(__file__)))
from chromite.cbuildbot import cbuildbot_failures as failures_lib
from chromite.lib import cros_test_lib
class SetFailureTypeTest(cros_test_lib.TestCase):
"""Test that the SetFailureType decorator works."""
ERROR_MESSAGE = 'You failed!'
class TacoNotTasty(Exception):
"""Raised when the taco is not tasty."""
class NoGuacamole(TacoNotTasty):
"""Raised when no guacamole in the taco."""
class FooException(Exception):
"""A foo exception."""
def _GetFunction(self, set_type, raise_type):
"""Returns a function to test.
set_type: The exception type that the function is decorated with.
raise_type: The exception type that the function raises.
The function to test.
def f():
raise raise_type(self.ERROR_MESSAGE)
return f
def testReraiseAsNewException(self):
"""Tests that the pre-set exception type is raised."""
f = self._GetFunction(self.TacoNotTasty, self.FooException)
self.assertRaises(self.TacoNotTasty, f)
def testStepFailureRemainsPrintable(self):
"""Tests that the re-raised exception remains printable."""
self._GetFunction(failures_lib.StepFailure, self.FooException)()
except Exception as e:
self.assertTrue(isinstance(e, failures_lib.StepFailure))
self.assertTrue(isinstance(e.__str__(), str))
def testReraiseOriginalException(self):
"""Tests that the original exception is re-raised."""
f = self._GetFunction(self.TacoNotTasty, self.NoGuacamole)
self.assertRaises(self.NoGuacamole, f)
if __name__ == '__main__':