def test_reduce(self):
class X(object):
def __reduce__(self):
return 123
x = Proxy(lambda: X())
self.assertEqual(x.__reduce__(), 123)
def test_reduce(self):
class X(object):
def __reduce__(self):
return 123
x = Proxy(lambda: X())
self.assertEqual(x.__reduce__(), 123)
def test_dir(self):
class X(object):
def __dir__(self):
return ['a', 'b', 'c']
x = Proxy(lambda: X())
self.assertListEqual(dir(x), ['a', 'b', 'c'])
class Y(object):
def __dir__(self):
raise RuntimeError()
y = Proxy(lambda: Y())
self.assertListEqual(dir(y), [])
def test_listproxy(self):
v = []
x = Proxy(lambda: v)
x.append(1)
x.extend([2, 3, 4])
self.assertEqual(x[0], 1)
self.assertEqual(x[:-1], [1, 2, 3])
del(x[-1])
self.assertEqual(x[:-1], [1, 2])
x[0] = 10
self.assertEqual(x[0], 10)
self.assertIn(10, x)
self.assertEqual(len(x), 3)
self.assertTrue(iter(x))
def test_slots(self):
class X(object):
__slots__ = ()
x = Proxy(X)
with self.assertRaises(AttributeError):
x.__dict__
def test_name(self):
def real():
"""real function"""
return 'REAL'
x = Proxy(lambda: real, name='xyz')
self.assertEqual(x.__name__, 'xyz')
y = Proxy(lambda: real)
self.assertEqual(y.__name__, 'real')
self.assertEqual(x.__doc__, 'real function')
self.assertEqual(x.__class__, type(real))
self.assertEqual(x.__dict__, real.__dict__)
self.assertEqual(repr(x), repr(real))
def test_bool(self):
class X(object):
def __bool__(self):
return False
__nonzero__ = __bool__
x = Proxy(lambda: X())
self.assertFalse(x)
def test_dictproxy(self):
v = {}
x = Proxy(lambda: v)
x['foo'] = 42
self.assertEqual(x['foo'], 42)
self.assertEqual(len(x), 1)
self.assertIn('foo', x)
del (x['foo'])
with self.assertRaises(KeyError):
x['foo']
self.assertTrue(iter(x))
def test_unicode(self):
class X(object):
def __unicode__(self):
return 'UNICODE'
__str__ = __unicode__
def __repr__(self):
return 'REPR'
x = Proxy(lambda: X())
self.assertEqual(str(x), 'UNICODE')
del (X.__unicode__)
del (X.__str__)
self.assertEqual(str(x), 'REPR')
def test_context(self):
class X(object):
entered = exited = False
def __enter__(self):
self.entered = True
return 1234
def __exit__(self, *exc_info):
self.exited = True
v = X()
x = Proxy(lambda: v)
with x as val:
self.assertEqual(val, 1234)
self.assertTrue(x.entered)
self.assertTrue(x.exited)
def test_listproxy(self):
v = []
x = Proxy(lambda: v)
x.append(1)
x.extend([2, 3, 4])
self.assertEqual(x[0], 1)
self.assertEqual(x[:-1], [1, 2, 3])
del (x[-1])
self.assertEqual(x[:-1], [1, 2])
x[0] = 10
self.assertEqual(x[0], 10)
self.assertIn(10, x)
self.assertEqual(len(x), 3)
self.assertTrue(iter(x))
def test_getsetdel_attr(self):
class X(object):
a = 1
b = 2
c = 3
def __dir__(self):
return ['a', 'b', 'c']
v = X()
x = Proxy(lambda: v)
self.assertListEqual(x.__members__, ['a', 'b', 'c'])
self.assertEqual(x.a, 1)
self.assertEqual(x.b, 2)
self.assertEqual(x.c, 3)
setattr(x, 'a', 10)
self.assertEqual(x.a, 10)
del (x.a)
self.assertEqual(x.a, 1)
self.stack = LocalStack()
def add(self, handler, method, args, kwargs, service_name):
"""Register new request."""
request = self.Request(handler, method, args, kwargs, service_name)
self.stack.push(request)
return request
@property
def current(self):
"""Return current request."""
return self.stack.top
def to_dict(self):
return {
ident: [(request.service_name, request.method.__name__,
request.args, request.kwargs)
for request in d.get('stack', [])]
for ident, d in self.stack
}
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, traceback):
self.stack.pop()
#: Get current thrift request.
current_request = Proxy(lambda: thriftpool.request_stack.current)
"""Handle global package state."""
from __future__ import absolute_import
from thriftworker.utils.proxy import Proxy
default_app = None
def set_current_app(app):
global default_app
default_app = app
def get_current_app():
global default_app
if default_app is None:
# creates the default app, but we want to defer that.
from thriftpool.app.base import ThriftPool
set_current_app(ThriftPool())
return default_app
current_app = Proxy(get_current_app)
def test_int(self):
self.assertEqual(Proxy(lambda: 10) + 1, Proxy(lambda: 11))
self.assertEqual(Proxy(lambda: 10) - 1, Proxy(lambda: 9))
self.assertEqual(Proxy(lambda: 10) * 2, Proxy(lambda: 20))
self.assertEqual(Proxy(lambda: 10)**2, Proxy(lambda: 100))
self.assertEqual(Proxy(lambda: 20) / 2, Proxy(lambda: 10))
self.assertEqual(Proxy(lambda: 20) // 2, Proxy(lambda: 10))
self.assertEqual(Proxy(lambda: 11) % 2, Proxy(lambda: 1))
self.assertEqual(Proxy(lambda: 10) << 2, Proxy(lambda: 40))
self.assertEqual(Proxy(lambda: 10) >> 2, Proxy(lambda: 2))
self.assertEqual(Proxy(lambda: 10) ^ 7, Proxy(lambda: 13))
self.assertEqual(Proxy(lambda: 10) | 40, Proxy(lambda: 42))
self.assertEqual(~Proxy(lambda: 10), Proxy(lambda: -11))
self.assertEqual(-Proxy(lambda: 10), Proxy(lambda: -10))
self.assertEqual(+Proxy(lambda: -10), Proxy(lambda: -10))
self.assertTrue(Proxy(lambda: 10) < Proxy(lambda: 20))
self.assertTrue(Proxy(lambda: 20) > Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 10) >= Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 10) <= Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 10) == Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 20) != Proxy(lambda: 10))
x = Proxy(lambda: 10)
x -= 1
self.assertEqual(x, 9)
x = Proxy(lambda: 9)
x += 1
self.assertEqual(x, 10)
x = Proxy(lambda: 10)
x *= 2
self.assertEqual(x, 20)
x = Proxy(lambda: 20)
x /= 2
self.assertEqual(x, 10)
x = Proxy(lambda: 10)
x %= 2
self.assertEqual(x, 0)
x = Proxy(lambda: 10)
x <<= 3
self.assertEqual(x, 80)
x = Proxy(lambda: 80)
x >>= 4
self.assertEqual(x, 5)
x = Proxy(lambda: 5)
x ^= 1
self.assertEqual(x, 4)
x = Proxy(lambda: 4)
x **= 4
self.assertEqual(x, 256)
x = Proxy(lambda: 256)
x //= 2
self.assertEqual(x, 128)
x = Proxy(lambda: 128)
x |= 2
self.assertEqual(x, 130)
x = Proxy(lambda: 130)
x &= 10
self.assertEqual(x, 2)
x = Proxy(lambda: 10)
self.assertEqual(type(x.__float__()), float)
self.assertEqual(type(x.__int__()), int)
self.assertEqual(type(x.__long__()), long)
self.assertTrue(hex(x))
self.assertTrue(oct(x))
def test_hash(self):
class X(object):
def __hash__(self):
return 1234
self.assertEqual(hash(Proxy(lambda: X())), 1234)
def test_int(self):
self.assertEqual(Proxy(lambda: 10) + 1, Proxy(lambda: 11))
self.assertEqual(Proxy(lambda: 10) - 1, Proxy(lambda: 9))
self.assertEqual(Proxy(lambda: 10) * 2, Proxy(lambda: 20))
self.assertEqual(Proxy(lambda: 10) ** 2, Proxy(lambda: 100))
self.assertEqual(Proxy(lambda: 20) / 2, Proxy(lambda: 10))
self.assertEqual(Proxy(lambda: 20) // 2, Proxy(lambda: 10))
self.assertEqual(Proxy(lambda: 11) % 2, Proxy(lambda: 1))
self.assertEqual(Proxy(lambda: 10) << 2, Proxy(lambda: 40))
self.assertEqual(Proxy(lambda: 10) >> 2, Proxy(lambda: 2))
self.assertEqual(Proxy(lambda: 10) ^ 7, Proxy(lambda: 13))
self.assertEqual(Proxy(lambda: 10) | 40, Proxy(lambda: 42))
self.assertEqual(~Proxy(lambda: 10), Proxy(lambda: -11))
self.assertEqual(-Proxy(lambda: 10), Proxy(lambda: -10))
self.assertEqual(+Proxy(lambda: -10), Proxy(lambda: -10))
self.assertTrue(Proxy(lambda: 10) < Proxy(lambda: 20))
self.assertTrue(Proxy(lambda: 20) > Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 10) >= Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 10) <= Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 10) == Proxy(lambda: 10))
self.assertTrue(Proxy(lambda: 20) != Proxy(lambda: 10))
x = Proxy(lambda: 10)
x -= 1
self.assertEqual(x, 9)
x = Proxy(lambda: 9)
x += 1
self.assertEqual(x, 10)
x = Proxy(lambda: 10)
x *= 2
self.assertEqual(x, 20)
x = Proxy(lambda: 20)
x /= 2
self.assertEqual(x, 10)
x = Proxy(lambda: 10)
x %= 2
self.assertEqual(x, 0)
x = Proxy(lambda: 10)
x <<= 3
self.assertEqual(x, 80)
x = Proxy(lambda: 80)
x >>= 4
self.assertEqual(x, 5)
x = Proxy(lambda: 5)
x ^= 1
self.assertEqual(x, 4)
x = Proxy(lambda: 4)
x **= 4
self.assertEqual(x, 256)
x = Proxy(lambda: 256)
x //= 2
self.assertEqual(x, 128)
x = Proxy(lambda: 128)
x |= 2
self.assertEqual(x, 130)
x = Proxy(lambda: 130)
x &= 10
self.assertEqual(x, 2)
x = Proxy(lambda: 10)
self.assertEqual(type(x.__float__()), float)
self.assertEqual(type(x.__int__()), int)
self.assertEqual(type(x.__long__()), long)
self.assertTrue(hex(x))
self.assertTrue(oct(x))
def test_call(self):
class X(object):
def __call__(self):
return 1234
self.assertEqual(Proxy(lambda: X())(), 1234)
