58 lines
2.9 KiB
Python
58 lines
2.9 KiB
Python
import copy
|
|
import importlib.util
|
|
import io
|
|
from pathlib import Path
|
|
import unittest
|
|
import zipfile
|
|
|
|
spec = importlib.util.spec_from_file_location('probe', Path(__file__).with_name('client-xlsx-address-probe.py'))
|
|
probe = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(probe)
|
|
|
|
|
|
class AddressProbeTests(unittest.TestCase):
|
|
def response(self):
|
|
return {'code': '200', 'data': {'mode': 'PARTIAL', 'sourceFileId': 's', 'errorReportFileId': 'e',
|
|
'result': {'total': 1, 'succeeded': 0, 'failed': 1, 'items': [
|
|
{'success': False, 'clientRowId': 'excel-row-3', 'errorCode': 'A0422',
|
|
'errorMessage': '取货地址不存在、已停用或不属于当前客户'}]}}}
|
|
|
|
def test_precise_address_rejection(self):
|
|
probe.assert_rejection(self.response(), '取货地址')
|
|
for change in [{'success': True}, {'orderId': 'created'}, {'errorCode': 'A0404'},
|
|
{'clientRowId': 'excel-row-4'}, {'errorMessage': '国内收货仓不能为空'}]:
|
|
changed = copy.deepcopy(self.response())
|
|
changed['data']['result']['items'][0].update(change)
|
|
with self.assertRaises(AssertionError):
|
|
probe.assert_rejection(changed, '取货地址')
|
|
|
|
def test_not_setup_or_partial_success(self):
|
|
for part, change in [('root', {'code': 'A0422'}), ('data', {'errorReportFileId': None}),
|
|
('result', {'succeeded': 1}), ('result', {'total': 2})]:
|
|
changed = self.response()
|
|
value = changed if part == 'root' else changed['data'] if part == 'data' else changed['data']['result']
|
|
value.update(change)
|
|
with self.assertRaises(AssertionError):
|
|
probe.assert_rejection(changed, '取货地址')
|
|
|
|
def test_workbook_preserves_hidden_parts_and_replaces_only_data(self):
|
|
raw = io.BytesIO()
|
|
sheet = '<worksheet><sheetData><row r="1"><c r="A1"/></row><row r="2"><c r="A2"/></row><row r="3"><c r="A3"/></row></sheetData><tail/></worksheet>'
|
|
with zipfile.ZipFile(raw, 'w') as z:
|
|
z.writestr('xl/worksheets/sheet1.xml', sheet)
|
|
z.writestr('xl/worksheets/sheet2.xml', '<foreign>source-options</foreign>')
|
|
z.writestr('xl/workbook.xml', '<names>source-names</names>')
|
|
output = probe.workbook(raw.getvalue(), ['<&'] + [''] * 29 + ['marker'])
|
|
with zipfile.ZipFile(io.BytesIO(output)) as z:
|
|
self.assertEqual(z.read('xl/worksheets/sheet2.xml'), b'<foreign>source-options</foreign>')
|
|
self.assertEqual(z.read('xl/workbook.xml'), b'<names>source-names</names>')
|
|
xml = z.read('xl/worksheets/sheet1.xml').decode()
|
|
self.assertIn('r="AE3"', xml)
|
|
self.assertIn('<&', xml)
|
|
self.assertEqual(xml.count('<row r="3">'), 1)
|
|
self.assertIn('<tail/>', xml)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|