def test_random_string(self):
length = 10
first = get_random_string(length)
second = get_random_string(length)
self.assertTrue(len(first) == len(second) == length)
self.assertTrue(first != second)
def test_modify_file(self):
temp_dir = tempfile.gettempdir()
temp_name = get_random_string(20)
modify_file(temp_name, temp_dir, ['Test'])
self.assertTrue(os.path.isfile(os.path.join(temp_dir, temp_name)))
tmp = open(os.path.join(temp_dir, temp_name), 'r').readlines()
self.assertTrue(tmp == ['Test\n'])
modify_file(
temp_name,
temp_dir,
['And another test', 'with a', '', 'blank line']
)
tmp = open(os.path.join(temp_dir, temp_name), 'r').readlines()
self.assertTrue(
tmp == [
'Test\n',
'And another test\n',
'with a\n',
'\n',
'blank line\n'
]
)
