def test_editfile_suffix(self):
"""
fl.editfile('legit', 's', 'foo', 'bar', 'old')
"""
fp = U.pj(self.testdir, U.function_name())
forig = fp + ".old"
tdata = ["foo bar",
"bar foo",
"barfoo",
"foobar foo",
"loofafool"]
xdata = [z.replace('foo', 'bar') for z in tdata]
U.writefile(fp, tdata, newlines=True)
fl.editfile(fp, 's', 'foo', 'bar', 'old')
self.assertEq(True, U.exists(fp))
self.assertEq(True, U.exists(forig))
self.assertEq(xdata, U.contents(fp))
self.assertEq(tdata, U.contents(forig))
def test_editfile_delete(self):
"""
fl.editfile('legit', 's', 'foo', '', None)
"""
fp = U.pj(self.testdir, U.function_name())
forig = fp + ".original"
tdata = ["foo bar",
"bar foo",
"barfoo",
"foobar foo",
"loofafool"]
xdata = [re.sub('^foo', '', z) for z in tdata]
U.writefile(fp, tdata, newlines=True)
fl.editfile(fp, 's', '^foo', '', None)
self.assertEq(True, U.exists(fp))
self.assertEq(True, U.exists(forig))
self.assertEq(xdata, U.contents(fp))
self.assertEq(tdata, U.contents(forig))
def test_function_name(self):
"""
Routine function_name() should return the name of its caller
"""
self.assertEqual(U.function_name(), "test_function_name")
