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()