/usr/lib/python2.7/dist-packages/pyvows/decorators.py is in python-pyvows 2.0.6-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | # -*- coding: utf-8 -*-
'''This module is the foundation that allows users to write PyVows-style tests.
'''
# pyVows testing engine
# https://github.com/heynemann/pyvows
# Licensed under the MIT license:
# http://www.opensource.org/licenses/mit-license
# Copyright (c) 2011 Bernardo Heynemann heynemann@gmail.com
from functools import wraps
import re
from pyvows.async_topic import VowsAsyncTopic
#-------------------------------------------------------------------------------------------------
def _batch(klass):
# This is underscored-prefixed because the only intended use (via
# `@Vows.batch`) expands on this core functionality
def klass_name(*args, **kw):
klass(*args, **kw)
return klass_name
def async_topic(topic):
'''Topic decorator. Allows PyVows testing of asynchronous topics.
Use `@Vows.async_topic` on your `topic` method to mark it as
asynchronous. This allows PyVows to test topics which use callbacks
instead of return values.
'''
def wrapper(*args, **kw):
return VowsAsyncTopic(topic, args, kw)
wrapper._original = topic
wrapper._wrapper_type = 'async_topic'
wrapper.__name__ = topic.__name__
return wrapper
def capture_error(topic_func):
'''Topic decorator. Allows any errors raised to become the topic value.
By default, errors raised in topic functions are reported as
errors. But sometimes you want the error to be the topic value, in
which case decorate the topic function with this decorator.'''
def wrapper(*args, **kw):
try:
return topic_func(*args, **kw)
except Exception as e:
return e
wrapper._original = topic_func
wrapper._wrapper_type = 'capture_error'
wrapper.__name__ = topic_func.__name__
return wrapper
#-------------------------------------------------------------------------------------------------
class FunctionWrapper(object):
'''Function decorator. Simply calls the decorated function when all
the wrapped functions have been called.
'''
def __init__(self, func):
self.waiting = 0
self.func = func
def wrap(self, method):
self.waiting += 1
@wraps(method)
def wrapper(*args, **kw):
try:
ret = method(*args, **kw)
return ret
finally:
self.waiting -= 1
self()
wrapper._original = method
return wrapper
def __call__(self):
if self.waiting == 0:
self.func()
|