# -*- coding: utf-8 -*-
"""票 07 客户测试构造点补 mock（DictReferenceService）"""
import io, sys
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')

MOCK = 'Mockito.mock(com.crm.dict.service.DictReferenceService.class)'

def patch(path, old, new):
    txt = open(path, encoding='utf-8').read()
    assert old in txt, path + ' anchor missing'
    open(path, 'w', encoding='utf-8').write(txt.replace(old, new, 1))
    print('patched', path)

base = 'crm-customer/src/test/java/com/crm/customer/'

patch(base + 'port/impl/CustomerCatalogPortImplIntegrationTest.java',
      '        CustomerServiceImpl custImpl = new CustomerServiceImpl(\n'
      '                new CustomerOplogWriter(oplogMapper, userService, deptService),\n'
      '                dedup, dictQueryService, name -> List.of(),\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper));',
      '        CustomerServiceImpl custImpl = new CustomerServiceImpl(\n'
      '                new CustomerOplogWriter(oplogMapper, userService, deptService),\n'
      '                dedup, dictQueryService, name -> List.of(),\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper),\n'
      '                ' + MOCK + ');')

patch(base + 'service/impl/CustomerServiceImplIntegrationTest.java',
      '        service = new CustomerServiceImpl(new CustomerOplogWriter(oplogMapper, userService, deptService),\n'
      '                dedup, dictQueryService, companyLookupPort,\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper));',
      '        service = new CustomerServiceImpl(new CustomerOplogWriter(oplogMapper, userService, deptService),\n'
      '                dedup, dictQueryService, companyLookupPort,\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper),\n'
      '                ' + MOCK + ');')

patch(base + 'service/impl/CustomerContactServiceIntegrationTest.java',
      '        CustomerServiceImpl custImpl = new CustomerServiceImpl(\n'
      '                new CustomerOplogWriter(oplogMapper, userService, deptService),\n'
      '                dedup, dictQueryService, name -> List.of(),\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper));',
      '        CustomerServiceImpl custImpl = new CustomerServiceImpl(\n'
      '                new CustomerOplogWriter(oplogMapper, userService, deptService),\n'
      '                dedup, dictQueryService, name -> List.of(),\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper),\n'
      '                ' + MOCK + ');')

patch(base + 'service/impl/CustomerImportIntegrationTest.java',
      '        executor = new CustomerImportExecutor(\n'
      '                taskMapper, failMapper, customerMapper, contactMapper, analyzer, writer,\n'
      '                userService, fileApi,\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper));',
      '        executor = new CustomerImportExecutor(\n'
      '                taskMapper, failMapper, customerMapper, contactMapper, analyzer, writer,\n'
      '                userService, fileApi,\n'
      '                new CustomerOwnershipWriter(customerMapper, userService, deptService),\n'
      '                new CustomerNoGenerator(customerMapper),\n'
      '                ' + MOCK + ');')