[deprecation/refactoring] simplify class_moved

1 files changed, 17 insertions, 0 deletions

diff --git a/test/unittest_deprecation.py b/test/unittest_deprecation.py
index eb75444..5432823 100644
--- a/test/unittest_deprecation.py
+++ b/test/unittest_deprecation.py
@@ -61,6 +61,23 @@ class RawInputTC(TestCase):
         self.assertEqual(self.messages,
                          ['[logilab.common] OldClass is deprecated, use AnyClass instead'])
 
+    def test_class_moved(self):
+        class AnyClass(object):
+            pass
+
+        OldClass = deprecation.class_moved(new_class=AnyClass, old_name="OldName")
+        OldClass()
+        self.assertEqual(self.messages,
+                         ['[logilab.common] class OldName is now available as unittest_deprecation.AnyClass'])
+
+        self.messages = []
+
+        AnyClass = deprecation.class_moved(new_class=AnyClass)
+
+        AnyClass()
+        self.assertEqual(self.messages,
+                         ['[logilab.common] class AnyClass is now available as unittest_deprecation.AnyClass'])
+
     def test_deprecated_func(self):
         any_func = deprecation.deprecated()(self.mk_func())
         any_func()