Commits

Anonymous committed 2afbca1

introduce_factory: added get_name()

Comments (0)

Files changed (2)

rope/refactor/introduce_factory.py

                             global_factory, job_set)
         return changes
 
+    def get_name(self):
+        """Return the name of the class"""
+        return self.old_name
+
     def _change_module(self, resources, changes,
                        factory_name, global_, job_set):
         if global_:

ropetest/refactor/__init__.py

                                         'create', global_factory=True)
         self.assertEquals(expected, mod.read())
 
+    def test_get_name_for_factories(self):
+        code = 'class C(object):\n    pass\n'
+        mod = testutils.create_module(self.project, 'mod')
+        mod.write(code)
+        factory = IntroduceFactory(self.project, mod,
+                                   mod.read().index('C') + 1)
+        self.assertEquals('C', factory.get_name())
+
     @testutils.assert_raises(RefactoringError)
     def test_raising_exception_for_global_factory_for_nested_classes(self):
         code = 'def create_var():\n'\