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' ] )