From dbba4acb25f405199845d0cf0b177cb53fdcf599 Mon Sep 17 00:00:00 2001
From: Graham Dumpleton <Graham.Dumpleton@gmail.com>
Date: Fri, 25 Apr 2014 21:51:34 +1000
Subject: [PATCH] Test for wrapping attribute access. #20

---
 tests/test_attribute_wrapper.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 tests/test_attribute_wrapper.py

diff --git a/tests/test_attribute_wrapper.py b/tests/test_attribute_wrapper.py
new file mode 100644
index 0000000..f68315a
--- /dev/null
+++ b/tests/test_attribute_wrapper.py
@@ -0,0 +1,24 @@
+from __future__ import print_function
+
+import unittest
+
+import wrapt
+
+class Class(object):
+    def __init__(self, value):
+        self.value = value
+
+class TestAttributeProxy(unittest.TestCase):
+
+    def test_wrap_attribute(self):
+        wrapt.wrap_object_attribute(__name__, 'Class.value', wrapt.ObjectProxy)
+
+        instance = Class(1)
+
+        self.assertEqual(instance.value, 1)
+        self.assertTrue(isinstance(instance.value, wrapt.ObjectProxy))
+
+        instance.value = 2
+
+        self.assertEqual(instance.value, 2)
+        self.assertTrue(isinstance(instance.value, wrapt.ObjectProxy))