test(order): verify foreign customer addresses rejected in XLSX imports

2026-10-03 17:08:15 +08:00
parent e49ad96cc4
commit 14696fb1a1
5 changed files with 291 additions and 0 deletions
@@ -0,0 +1,57 @@
+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('&lt;&amp;', xml)
+            self.assertEqual(xml.count('<row r="3">'), 1)
+            self.assertIn('<tail/>', xml)
+
+
+if __name__ == '__main__':
+    unittest.main()