| | |
|---|
| | | package com.ruoyi.chargingPile.service.impl; |
|---|
| | | |
|---|
| | | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|---|
| | | import com.ruoyi.chargingPile.api.dto.GetSiteListDTO; |
|---|
| | | import com.ruoyi.chargingPile.api.model.Site; |
|---|
| | | import com.ruoyi.chargingPile.api.model.TAccountingStrategyDetail; |
|---|
| | | import com.ruoyi.chargingPile.api.model.TChargingGun; |
|---|
| | | import com.ruoyi.chargingPile.api.model.TChargingPile; |
|---|
| | | import com.ruoyi.chargingPile.mapper.SiteMapper; |
|---|
| | | import com.ruoyi.chargingPile.service.IPartnerService; |
|---|
| | | import com.ruoyi.chargingPile.service.ITAccountingStrategyDetailService; |
|---|
| | | import com.ruoyi.chargingPile.service.ITChargingGunService; |
|---|
| | | import com.ruoyi.chargingPile.service.ITChargingPileService; |
|---|
| | | import com.ruoyi.common.core.domain.R; |
|---|
| | | import com.ruoyi.integration.api.feignClient.ChengDuJianGuanClient; |
|---|
| | | import com.ruoyi.integration.api.feignClient.IntegrationClient; |
|---|
| | | import com.ruoyi.integration.api.feignClient.TcecClient; |
|---|
| | | import com.ruoyi.system.api.domain.SysUser; |
|---|
| | | import com.ruoyi.system.api.feignClient.SysUserClient; |
|---|
| | | import org.junit.jupiter.api.BeforeEach; |
|---|
| | | import org.junit.jupiter.api.Test; |
|---|
| | | import org.springframework.test.util.ReflectionTestUtils; |
|---|
| | | |
|---|
| | | import java.util.ArrayList; |
|---|
| | | import java.util.Collections; |
|---|
| | | import java.util.List; |
|---|
| | | |
|---|
| | | import static org.junit.jupiter.api.Assertions.assertDoesNotThrow; |
|---|
| | | import static org.mockito.ArgumentMatchers.any; |
|---|
| | | import static org.mockito.ArgumentMatchers.eq; |
|---|
| | | import static org.mockito.ArgumentMatchers.isNull; |
|---|
| | | import static org.mockito.Mockito.atLeastOnce; |
|---|
| | | import static org.mockito.Mockito.mock; |
|---|
| | | import static org.mockito.Mockito.never; |
|---|
| | | import static org.mockito.Mockito.timeout; |
|---|
| | | import static org.mockito.Mockito.verify; |
|---|
| | | import static org.mockito.Mockito.when; |
|---|
| | | |
|---|
| | | class SiteServiceImplTest { |
|---|
| | | |
|---|
| | | private SiteServiceImpl service; |
|---|
| | | private SiteMapper siteMapper; |
|---|
| | | private ITChargingGunService chargingGunService; |
|---|
| | | private ITAccountingStrategyDetailService accountingStrategyDetailService; |
|---|
| | | private ITChargingPileService chargingPileService; |
|---|
| | | private IntegrationClient integrationClient; |
|---|
| | | private ChengDuJianGuanClient chengDuJianGuanClient; |
|---|
| | | private TcecClient tcecClient; |
|---|
| | | |
|---|
| | | @BeforeEach |
|---|
| | | void setUp() { |
|---|
| | | service = new SiteServiceImpl(); |
|---|
| | | siteMapper = mock(SiteMapper.class); |
|---|
| | | chargingGunService = mock(ITChargingGunService.class); |
|---|
| | | accountingStrategyDetailService = mock(ITAccountingStrategyDetailService.class); |
|---|
| | | chargingPileService = mock(ITChargingPileService.class); |
|---|
| | | integrationClient = mock(IntegrationClient.class); |
|---|
| | | chengDuJianGuanClient = mock(ChengDuJianGuanClient.class); |
|---|
| | | tcecClient = mock(TcecClient.class); |
|---|
| | | |
|---|
| | | ReflectionTestUtils.setField(service, "baseMapper", siteMapper); |
|---|
| | | ReflectionTestUtils.setField(service, "chargingGunService", chargingGunService); |
|---|
| | | ReflectionTestUtils.setField(service, "accountingStrategyDetailService", accountingStrategyDetailService); |
|---|
| | | ReflectionTestUtils.setField(service, "chargingPileService", chargingPileService); |
|---|
| | | ReflectionTestUtils.setField(service, "integrationClient", integrationClient); |
|---|
| | | ReflectionTestUtils.setField(service, "chengDuJianGuanClient", chengDuJianGuanClient); |
|---|
| | | ReflectionTestUtils.setField(service, "tcecClient", tcecClient); |
|---|
| | | } |
|---|
| | | |
|---|
| | | @Test |
|---|
| | | void partnerSiteListIncludesAllSitesRelatedToCurrentPartner() { |
|---|
| | | IPartnerService partnerService = mock(IPartnerService.class); |
|---|
| | | SysUserClient sysUserClient = mock(SysUserClient.class); |
|---|
| | | ReflectionTestUtils.setField(service, "partnerService", partnerService); |
|---|
| | | ReflectionTestUtils.setField(service, "sysUserClient", sysUserClient); |
|---|
| | | |
|---|
| | | SysUser sysUser = new SysUser(); |
|---|
| | | sysUser.setUserId(100L); |
|---|
| | | sysUser.setRoleType(2); |
|---|
| | | sysUser.setObjectId(715); |
|---|
| | | |
|---|
| | | when(sysUserClient.getSysUser(100L)).thenReturn(R.ok(sysUser)); |
|---|
| | | when(siteMapper.getSiteOptions(isNull(), eq(715))) |
|---|
| | | .thenReturn(Collections.<GetSiteListDTO>emptyList()); |
|---|
| | | |
|---|
| | | service.getSiteList1(100L); |
|---|
| | | |
|---|
| | | verify(siteMapper).getSiteOptions(isNull(), eq(715)); |
|---|
| | | verify(partnerService, never()).authSite(any(), any()); |
|---|
| | | verify(sysUserClient, never()).isAdmin(any()); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 验证 setAccountingStrategy 使用 lambdaUpdate 只更新 accountingStrategyId 字段， |
|---|
| | | * 并且远程调用在异步线程中执行，避免事务长时间占用数据库连接 |
|---|
| | | */ |
|---|
| | | @Test |
|---|
| | | void setAccountingStrategyUpdatesOnlyTargetFieldAndAsyncRemoteCalls() throws Exception { |
|---|
| | | Integer siteId = 1; |
|---|
| | | Integer accountingStrategyId = 100; |
|---|
| | | |
|---|
| | | // mock 充电枪列表 |
|---|
| | | List<TChargingGun> guns = new ArrayList<>(); |
|---|
| | | TChargingGun gun = new TChargingGun(); |
|---|
| | | gun.setId(10); |
|---|
| | | gun.setSiteId(siteId); |
|---|
| | | guns.add(gun); |
|---|
| | | |
|---|
| | | // mock 充电桩列表 |
|---|
| | | List<TChargingPile> piles = new ArrayList<>(); |
|---|
| | | TChargingPile pile = new TChargingPile(); |
|---|
| | | pile.setCode("PILE001"); |
|---|
| | | pile.setId(200); |
|---|
| | | pile.setSiteId(siteId); |
|---|
| | | piles.add(pile); |
|---|
| | | |
|---|
| | | // mock 计费策略明细 |
|---|
| | | List<TAccountingStrategyDetail> details = new ArrayList<>(); |
|---|
| | | |
|---|
| | | // mock Site 查询（用于 isOpenSynchronization） |
|---|
| | | Site mockSite = new Site(); |
|---|
| | | mockSite.setId(siteId); |
|---|
| | | mockSite.setIsOpenSynchronization(1); |
|---|
| | | |
|---|
| | | when(siteMapper.selectList(any(LambdaQueryWrapper.class))).thenReturn(Collections.singletonList(mockSite)); |
|---|
| | | when(chargingGunService.list(any(LambdaQueryWrapper.class))).thenReturn(guns); |
|---|
| | | when(accountingStrategyDetailService.list(any(LambdaQueryWrapper.class))).thenReturn(details); |
|---|
| | | |
|---|
| | | assertDoesNotThrow(() -> service.setAccountingStrategy(siteId, accountingStrategyId)); |
|---|
| | | |
|---|
| | | // 验证数据库操作：站点和充电枪都通过 lambdaUpdate 更新 |
|---|
| | | verify(siteMapper, atLeastOnce()).update(any(), any()); |
|---|
| | | verify(siteMapper).selectList(any(LambdaQueryWrapper.class)); |
|---|
| | | |
|---|
| | | // 异步线程执行需要等待一小段时间 |
|---|
| | | Thread.sleep(500); |
|---|
| | | |
|---|
| | | // 验证远程调用在异步线程中执行 |
|---|
| | | verify(integrationClient, timeout(1000)).setupBillingModel1(eq("PILE001"), any()); |
|---|
| | | verify(chengDuJianGuanClient, timeout(1000)).notificationStationFee(eq(200)); |
|---|
| | | verify(tcecClient, timeout(1000)).notificationStationChange(any()); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 验证 setAccountingStrategy 在站点无需同步时不会调用三方平台推送 |
|---|
| | | */ |
|---|
| | | @Test |
|---|
| | | void setAccountingStrategySkipsSyncWhenNotOpen() throws Exception { |
|---|
| | | Integer siteId = 2; |
|---|
| | | Integer accountingStrategyId = 200; |
|---|
| | | |
|---|
| | | // mock 充电枪列表 |
|---|
| | | TChargingGun gun = new TChargingGun(); |
|---|
| | | gun.setId(20); |
|---|
| | | gun.setSiteId(siteId); |
|---|
| | | List<TChargingGun> guns = Collections.singletonList(gun); |
|---|
| | | |
|---|
| | | // mock 充电桩列表 |
|---|
| | | TChargingPile pile = new TChargingPile(); |
|---|
| | | pile.setCode("PILE002"); |
|---|
| | | pile.setId(201); |
|---|
| | | pile.setSiteId(siteId); |
|---|
| | | List<TChargingPile> piles = Collections.singletonList(pile); |
|---|
| | | |
|---|
| | | List<TAccountingStrategyDetail> details = new ArrayList<>(); |
|---|
| | | |
|---|
| | | // isOpenSynchronization = 0，不需要同步 |
|---|
| | | Site mockSite = new Site(); |
|---|
| | | mockSite.setId(siteId); |
|---|
| | | mockSite.setIsOpenSynchronization(0); |
|---|
| | | |
|---|
| | | when(siteMapper.selectList(any(LambdaQueryWrapper.class))).thenReturn(Collections.singletonList(mockSite)); |
|---|
| | | when(chargingGunService.list(any(LambdaQueryWrapper.class))).thenReturn(guns); |
|---|
| | | when(accountingStrategyDetailService.list(any(LambdaQueryWrapper.class))).thenReturn(details); |
|---|
| | | |
|---|
| | | assertDoesNotThrow(() -> service.setAccountingStrategy(siteId, accountingStrategyId)); |
|---|
| | | |
|---|
| | | Thread.sleep(500); |
|---|
| | | |
|---|
| | | verify(integrationClient, timeout(1000)).setupBillingModel1(eq("PILE002"), any()); |
|---|
| | | verify(chengDuJianGuanClient, timeout(1000)).notificationStationFee(eq(201)); |
|---|
| | | // tcecClient.notificationStationChange 不应被调用 |
|---|
| | | verify(tcecClient, never()).notificationStationChange(any()); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | //package com.ruoyi.chargingPile.service.impl; |
|---|
| | | // |
|---|
| | | //import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|---|
| | | //import com.ruoyi.chargingPile.api.dto.GetSiteListDTO; |
|---|
| | | //import com.ruoyi.chargingPile.api.model.Site; |
|---|
| | | //import com.ruoyi.chargingPile.api.model.TAccountingStrategyDetail; |
|---|
| | | //import com.ruoyi.chargingPile.api.model.TChargingGun; |
|---|
| | | //import com.ruoyi.chargingPile.api.model.TChargingPile; |
|---|
| | | //import com.ruoyi.chargingPile.mapper.SiteMapper; |
|---|
| | | //import com.ruoyi.chargingPile.service.IPartnerService; |
|---|
| | | //import com.ruoyi.common.core.domain.R; |
|---|
| | | //import com.ruoyi.integration.api.feignClient.ChengDuJianGuanClient; |
|---|
| | | //import com.ruoyi.integration.api.feignClient.IntegrationClient; |
|---|
| | | //import com.ruoyi.system.api.domain.SysUser; |
|---|
| | | //import com.ruoyi.system.api.feignClient.SysUserClient; |
|---|
| | | //import org.junit.jupiter.api.BeforeEach; |
|---|
| | | //import org.junit.jupiter.api.Test; |
|---|
| | | //import org.springframework.test.util.ReflectionTestUtils; |
|---|
| | | // |
|---|
| | | //import java.util.ArrayList; |
|---|
| | | //import java.util.Collections; |
|---|
| | | //import java.util.List; |
|---|
| | | // |
|---|
| | | //import static org.junit.jupiter.api.Assertions.assertDoesNotThrow; |
|---|
| | | //import static org.mockito.ArgumentMatchers.any; |
|---|
| | | //import static org.mockito.ArgumentMatchers.eq; |
|---|
| | | //import static org.mockito.ArgumentMatchers.isNull; |
|---|
| | | //import static org.mockito.Mockito.atLeastOnce; |
|---|
| | | //import static org.mockito.Mockito.mock; |
|---|
| | | //import static org.mockito.Mockito.never; |
|---|
| | | //import static org.mockito.Mockito.timeout; |
|---|
| | | //import static org.mockito.Mockito.verify; |
|---|
| | | //import static org.mockito.Mockito.when; |
|---|
| | | // |
|---|
| | | //class SiteServiceImplTest { |
|---|
| | | // |
|---|
| | | // private SiteServiceImpl service; |
|---|
| | | // private SiteMapper siteMapper; |
|---|
| | | // private ITChargingGunService chargingGunService; |
|---|
| | | // private ITAccountingStrategyDetailService accountingStrategyDetailService; |
|---|
| | | // private ITChargingPileService chargingPileService; |
|---|
| | | // private IntegrationClient integrationClient; |
|---|
| | | // private ChengDuJianGuanClient chengDuJianGuanClient; |
|---|
| | | // private TcecClient tcecClient; |
|---|
| | | // |
|---|
| | | // @BeforeEach |
|---|
| | | // void setUp() { |
|---|
| | | // service = new SiteServiceImpl(); |
|---|
| | | // siteMapper = mock(SiteMapper.class); |
|---|
| | | // chargingGunService = mock(ITChargingGunService.class); |
|---|
| | | // accountingStrategyDetailService = mock(ITAccountingStrategyDetailService.class); |
|---|
| | | // chargingPileService = mock(ITChargingPileService.class); |
|---|
| | | // integrationClient = mock(IntegrationClient.class); |
|---|
| | | // chengDuJianGuanClient = mock(ChengDuJianGuanClient.class); |
|---|
| | | // tcecClient = mock(TcecClient.class); |
|---|
| | | // |
|---|
| | | // ReflectionTestUtils.setField(service, "baseMapper", siteMapper); |
|---|
| | | // ReflectionTestUtils.setField(service, "chargingGunService", chargingGunService); |
|---|
| | | // ReflectionTestUtils.setField(service, "accountingStrategyDetailService", accountingStrategyDetailService); |
|---|
| | | // ReflectionTestUtils.setField(service, "chargingPileService", chargingPileService); |
|---|
| | | // ReflectionTestUtils.setField(service, "integrationClient", integrationClient); |
|---|
| | | // ReflectionTestUtils.setField(service, "chengDuJianGuanClient", chengDuJianGuanClient); |
|---|
| | | // ReflectionTestUtils.setField(service, "tcecClient", tcecClient); |
|---|
| | | // } |
|---|
| | | // |
|---|
| | | // @Test |
|---|
| | | // void partnerSiteListIncludesAllSitesRelatedToCurrentPartner() { |
|---|
| | | // IPartnerService partnerService = mock(IPartnerService.class); |
|---|
| | | // SysUserClient sysUserClient = mock(SysUserClient.class); |
|---|
| | | // ReflectionTestUtils.setField(service, "partnerService", partnerService); |
|---|
| | | // ReflectionTestUtils.setField(service, "sysUserClient", sysUserClient); |
|---|
| | | // |
|---|
| | | // SysUser sysUser = new SysUser(); |
|---|
| | | // sysUser.setUserId(100L); |
|---|
| | | // sysUser.setRoleType(2); |
|---|
| | | // sysUser.setObjectId(715); |
|---|
| | | // |
|---|
| | | // when(sysUserClient.getSysUser(100L)).thenReturn(R.ok(sysUser)); |
|---|
| | | // when(siteMapper.getSiteOptions(isNull(), eq(715))) |
|---|
| | | // .thenReturn(Collections.<GetSiteListDTO>emptyList()); |
|---|
| | | // |
|---|
| | | // service.getSiteList1(100L); |
|---|
| | | // |
|---|
| | | // verify(siteMapper).getSiteOptions(isNull(), eq(715)); |
|---|
| | | // verify(partnerService, never()).authSite(any(), any()); |
|---|
| | | // verify(sysUserClient, never()).isAdmin(any()); |
|---|
| | | // } |
|---|
| | | // |
|---|
| | | // /** |
|---|
| | | // * 验证 setAccountingStrategy 使用 lambdaUpdate 只更新 accountingStrategyId 字段， |
|---|
| | | // * 并且远程调用在异步线程中执行，避免事务长时间占用数据库连接 |
|---|
| | | // */ |
|---|
| | | // @Test |
|---|
| | | // void setAccountingStrategyUpdatesOnlyTargetFieldAndAsyncRemoteCalls() throws Exception { |
|---|
| | | // Integer siteId = 1; |
|---|
| | | // Integer accountingStrategyId = 100; |
|---|
| | | // |
|---|
| | | // // mock 充电枪列表 |
|---|
| | | // List<TChargingGun> guns = new ArrayList<>(); |
|---|
| | | // TChargingGun gun = new TChargingGun(); |
|---|
| | | // gun.setId(10); |
|---|
| | | // gun.setSiteId(siteId); |
|---|
| | | // guns.add(gun); |
|---|
| | | // |
|---|
| | | // // mock 充电桩列表 |
|---|
| | | // List<TChargingPile> piles = new ArrayList<>(); |
|---|
| | | // TChargingPile pile = new TChargingPile(); |
|---|
| | | // pile.setCode("PILE001"); |
|---|
| | | // pile.setId(200); |
|---|
| | | // pile.setSiteId(siteId); |
|---|
| | | // piles.add(pile); |
|---|
| | | // |
|---|
| | | // // mock 计费策略明细 |
|---|
| | | // List<TAccountingStrategyDetail> details = new ArrayList<>(); |
|---|
| | | // |
|---|
| | | // // mock Site 查询（用于 isOpenSynchronization） |
|---|
| | | // Site mockSite = new Site(); |
|---|
| | | // mockSite.setId(siteId); |
|---|
| | | // mockSite.setIsOpenSynchronization(1); |
|---|
| | | // |
|---|
| | | // when(siteMapper.selectList(any(LambdaQueryWrapper.class))).thenReturn(Collections.singletonList(mockSite)); |
|---|
| | | // when(chargingGunService.list(any(LambdaQueryWrapper.class))).thenReturn(guns); |
|---|
| | | // when(accountingStrategyDetailService.list(any(LambdaQueryWrapper.class))).thenReturn(details); |
|---|
| | | // |
|---|
| | | // assertDoesNotThrow(() -> service.setAccountingStrategy(siteId, accountingStrategyId)); |
|---|
| | | // |
|---|
| | | // // 验证数据库操作：站点和充电枪都通过 lambdaUpdate 更新 |
|---|
| | | // verify(siteMapper, atLeastOnce()).update(any(), any()); |
|---|
| | | // verify(siteMapper).selectList(any(LambdaQueryWrapper.class)); |
|---|
| | | // |
|---|
| | | // // 异步线程执行需要等待一小段时间 |
|---|
| | | // Thread.sleep(500); |
|---|
| | | // |
|---|
| | | // // 验证远程调用在异步线程中执行 |
|---|
| | | // verify(integrationClient, timeout(1000)).setupBillingModel1(eq("PILE001"), any()); |
|---|
| | | // verify(chengDuJianGuanClient, timeout(1000)).notificationStationFee(eq(200)); |
|---|
| | | // verify(tcecClient, timeout(1000)).notificationStationChange(any()); |
|---|
| | | // } |
|---|
| | | // |
|---|
| | | // /** |
|---|
| | | // * 验证 setAccountingStrategy 在站点无需同步时不会调用三方平台推送 |
|---|
| | | // */ |
|---|
| | | // @Test |
|---|
| | | // void setAccountingStrategySkipsSyncWhenNotOpen() throws Exception { |
|---|
| | | // Integer siteId = 2; |
|---|
| | | // Integer accountingStrategyId = 200; |
|---|
| | | // |
|---|
| | | // // mock 充电枪列表 |
|---|
| | | // TChargingGun gun = new TChargingGun(); |
|---|
| | | // gun.setId(20); |
|---|
| | | // gun.setSiteId(siteId); |
|---|
| | | // List<TChargingGun> guns = Collections.singletonList(gun); |
|---|
| | | // |
|---|
| | | // // mock 充电桩列表 |
|---|
| | | // TChargingPile pile = new TChargingPile(); |
|---|
| | | // pile.setCode("PILE002"); |
|---|
| | | // pile.setId(201); |
|---|
| | | // pile.setSiteId(siteId); |
|---|
| | | // List<TChargingPile> piles = Collections.singletonList(pile); |
|---|
| | | // |
|---|
| | | // List<TAccountingStrategyDetail> details = new ArrayList<>(); |
|---|
| | | // |
|---|
| | | // // isOpenSynchronization = 0，不需要同步 |
|---|
| | | // Site mockSite = new Site(); |
|---|
| | | // mockSite.setId(siteId); |
|---|
| | | // mockSite.setIsOpenSynchronization(0); |
|---|
| | | // |
|---|
| | | // when(siteMapper.selectList(any(LambdaQueryWrapper.class))).thenReturn(Collections.singletonList(mockSite)); |
|---|
| | | // when(chargingGunService.list(any(LambdaQueryWrapper.class))).thenReturn(guns); |
|---|
| | | // when(accountingStrategyDetailService.list(any(LambdaQueryWrapper.class))).thenReturn(details); |
|---|
| | | // |
|---|
| | | // assertDoesNotThrow(() -> service.setAccountingStrategy(siteId, accountingStrategyId)); |
|---|
| | | // |
|---|
| | | // Thread.sleep(500); |
|---|
| | | // |
|---|
| | | // verify(integrationClient, timeout(1000)).setupBillingModel1(eq("PILE002"), any()); |
|---|
| | | // verify(chengDuJianGuanClient, timeout(1000)).notificationStationFee(eq(201)); |
|---|
| | | // // tcecClient.notificationStationChange 不应被调用 |
|---|
| | | // verify(tcecClient, never()).notificationStationChange(any()); |
|---|
| | | // } |
|---|
| | | //} |
|---|
