__init__.py
 1  class RpcTestException(Exception):
 2      """Superclass for an exception generated from the RPC test code."""
 3  
 4  
 5  class FatalException(Exception):
 6      """An exception indicating that we need to stop the unit tests immediately."""
 7  
 8  
 9  def arti_test(func):
10      """
11      Decorator: Marks a function as a test.
12      """
13      # TODO: Later, expand this to take arguments to list specific requirements
14      # or attributes for the test.
15      func.arti_rpc_test = True
16      return func