def test_replace_string_in_file(self):
test_name = 'gbk.txt'
ori_data = '这是一个文件，用于进行测试。what are you doing.'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
expect_data = ori_data
file_data = myfile.read_file_content(file_name)
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expect_data, file_data, msg_format.format(expect_data, file_data))
enc_result = myfile.detect_file_enc(file_name)
expect_data = 'gb2312'
encoding = enc_result['encoding'].lower()
self.assertEqual(expect_data, encoding, msg_format.format(expect_data, encoding))
src_dst_list = list(zip(['what'], ['F**k']))
enc_result = myfile.replace_string_in_file(file_name, src_dst_list)
expect_data = '这是一个文件，用于进行测试。F**k are you doing.'
file_data = myfile.read_file_content(file_name)
self.assertEqual(expect_data, file_data, msg_format.format(expect_data, file_data))
enc_result = myfile.detect_file_enc(file_name)
expect_data = 'gb2312'
encoding = enc_result['encoding'].lower()
self.assertEqual(expect_data, encoding, msg_format.format(expect_data, encoding))
myfile.write_to_file(file_name, ori_data, encoding)
def test_read_file_first_line(self):
test_name = r'upgrade.xml'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
first_line = myfile.read_file_first_line(file_name)
expect_data = '<?xml version="1.0" encoding="utf-8"?>\r\n'
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expect_data, first_line, msg_format.format(expect_data, first_line))
def test_read_file_content(self):
# GB2312 编码格式文件验证
ori_content = '这只是一个测试文件。'
test_name = r'for_test.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
readed_content = myfile.read_file_content(file_name)
msg_format = 'expecte {0}, but {1}'
self.assertEqual(ori_content, readed_content, msg_format.format(ori_content, readed_content))
# UTF8 with BOM 编码格式文件验证
ori_content = 'utf-8-sig及中文验证;'
test_name = r'utf_8_sig.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
readed_content = myfile.read_file_content(file_name)
msg_format = 'expecte {0}, but {1}'
self.assertEqual(ori_content, readed_content, msg_format.format(ori_content, readed_content))
def test_get_file_sha1(self):
test_name = r'gbk.txt'
file_name = myfile.get_real_dir(__modpath__) + '\\' + test_name
file_sha1 = myhash.get_file_sha1(file_name)
ori_sha1 = '6B497D040B40DDA36C53E7F96BE12390602523B1'
expected = ori_sha1.lower()
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expected, file_sha1.lower(),
msg_format.format(expected, file_sha1.lower()))
def test_get_file_md5(self):
test_name = r'gbk.txt'
file_name = myfile.get_real_dir(__modpath__) + '\\' + test_name
file_md5 = myhash.get_file_md5(file_name)
ori_md5 = '71BC7781C1AF31B3F50FFF631FA80852'
expected = ori_md5.lower()
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expected, file_md5.lower(),
msg_format.format(expected, file_md5.lower()))
def test_write_to_file(self):
content = "info to write"
dst_name = r'file_output.txt'
dst_file = myfile.get_real_dir(__modpath__) + os.sep + dst_name
myfile.write_to_file(dst_file, content)
readed_content = myfile.read_file_content(dst_file)
msg_format = 'expecte {0}, but {1}'
self.assertEqual(content, readed_content, msg_format.format(content, readed_content))
def test_read_valid_string_list_from_file():
test_name = r'AndroidManifest.xml'
file_name = myfile.get_real_dir(__modpath__) + '\\' + test_name
updater = apk_builder.VersionConfigUpdater(file_name, False)
print('original version code: ' + updater.version_code)
print('original version name: ' + updater.version_name)
print()
updater.update_version_config()
print('new version code: ' + updater.version_code)
print('new version name: ' + updater.version_name)
def test_filter_unique_item(self):
test_name = r'filter_unique_intput.txt'
file_name = myfile.get_real_dir(__modpath__) + '\\' + test_name
readed_lines = myfile.read_file_lines(file_name)
src_list = []
for item in readed_lines:
src_list.append(item.strip())
dst_list = myutility.filter_unique_item(src_list)
expected = 5
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expected, len(dst_list),
msg_format.format(expected, len(dst_list)))
def test_detect_file_enc(self):
test_name = r'for_test.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
enc_result = myfile.detect_file_enc(file_name)
expecte = 'GB2312'
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expecte, enc_result['encoding'], msg_format.format(expecte, enc_result['encoding']))
test_name = r'alias_example.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
enc_result = myfile.detect_file_enc(file_name)
expecte = 'utf-8'
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expecte, enc_result['encoding'], msg_format.format(expecte, enc_result['encoding']))
test_name = r'utf_8_sig.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
enc_result = myfile.detect_file_enc(file_name)
expecte = 'utf-8-sig'
msg_format = 'expecte {0}, but {1}'
self.assertEqual(expecte, enc_result['encoding'], msg_format.format(expecte, enc_result['encoding']))
def test_get_dict_from_file(self):
alias_file = 'alias_example.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + alias_file
dict_ = myfile.get_dict_from_file(file_name)
key = 'BTV文艺'
value = '北京文艺'
msg_format = 'expecte {0}, but {1}'
self.assertEqual(dict_[key], value, msg_format.format(value, dict_[key]))
key = 'BTV科教'
value = '北京科教'
msg_format = 'expecte {0}, but {1}'
self.assertEqual(dict_[key], value, msg_format.format(value, dict_[key]))
key = 'what'
msg_format = 'expecte {0}, but {1}'
self.assertFalse(key in dict_, msg_format.format(False, True))
key = 'BTV科教'
msg_format = 'expecte {0}, but {1}'
self.assertTrue(key in dict_, msg_format.format(False, True))
def test_read_valid_string_list_from_file():
test_name = r'test_valid_string_list.txt'
file_name = myfile.get_real_dir(__modpath__) + os.sep + test_name
valid_list = myfile.read_valid_string_list_from_file(file_name)
pprint.pprint(valid_list)