def test_changing_the_module_and_class_methods(self):
code = dedent("""\
class C(object):
def a_func(self):
return 1
def another_func(self):
pass
""")
self.mod.write(code)
replacer = MethodObject(self.project, self.mod, code.index("func"))
self.project.do(replacer.get_changes("_New"))
expected = dedent("""\
class C(object):
def a_func(self):
return _New(self)()
def another_func(self):
pass
class _New(object):
def __init__(self, host):
self.self = host
def __call__(self):
return 1
""")
self.assertEqual(expected, self.mod.read())
def test_changing_the_module(self):
code = 'def func():\n return 1\n'
self.mod.write(code)
replacer = MethodObject(self.project, self.mod, code.index('func'))
self.project.do(replacer.get_changes('_New'))
expected = 'def func():\n' \
' return _New()()\n\n\n' \
'class _New(object):\n\n' \
' def __call__(self):\n' \
' return 1\n'
self.assertEquals(expected, self.mod.read())
def test_changing_the_module(self):
code = 'def func():\n return 1\n'
self.mod.write(code)
replacer = MethodObject(self.project, self.mod, code.index('func'))
self.project.do(replacer.get_changes('_New'))
expected = 'def func():\n' \
' return _New()()\n\n\n' \
'class _New(object):\n\n' \
' def __call__(self):\n' \
' return 1\n'
self.assertEquals(expected, self.mod.read())
def test_changing_the_module(self):
code = dedent("""\
def func():
return 1
""")
self.mod.write(code)
replacer = MethodObject(self.project, self.mod, code.index("func"))
self.project.do(replacer.get_changes("_New"))
expected = dedent("""\
def func():
return _New()()
class _New(object):
def __call__(self):
return 1
""")
self.assertEqual(expected, self.mod.read())
def test_changing_the_module_and_class_methods(self):
code = 'class C(object):\n\n' \
' def a_func(self):\n' \
' return 1\n\n' \
' def another_func(self):\n' \
' pass\n'
self.mod.write(code)
replacer = MethodObject(self.project, self.mod, code.index('func'))
self.project.do(replacer.get_changes('_New'))
expected = 'class C(object):\n\n' \
' def a_func(self):\n' \
' return _New(self)()\n\n' \
' def another_func(self):\n' \
' pass\n\n\n' \
'class _New(object):\n\n' \
' def __init__(self, host):\n' \
' self.self = host\n\n' \
' def __call__(self):\n' \
' return 1\n'
self.assertEquals(expected, self.mod.read())
def test_changing_the_module_and_class_methods(self):
code = 'class C(object):\n\n' \
' def a_func(self):\n' \
' return 1\n\n' \
' def another_func(self):\n' \
' pass\n'
self.mod.write(code)
replacer = MethodObject(self.project, self.mod, code.index('func'))
self.project.do(replacer.get_changes('_New'))
expected = 'class C(object):\n\n' \
' def a_func(self):\n' \
' return _New(self)()\n\n' \
' def another_func(self):\n' \
' pass\n\n\n' \
'class _New(object):\n\n' \
' def __init__(self, host):\n' \
' self.self = host\n\n' \
' def __call__(self):\n' \
' return 1\n'
self.assertEquals(expected, self.mod.read())
