package com.ruoyi.order.controller;

import com.ruoyi.chargingPile.api.dto.GetSiteListDTO;
import com.ruoyi.chargingPile.api.feignClient.SiteClient;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.security.service.TokenService;
import com.ruoyi.order.service.TOrderAppealService;
import com.ruoyi.system.api.domain.SysUser;
import org.junit.jupiter.api.Test;
import org.springframework.test.util.ReflectionTestUtils;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Map;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

class TOrderAppealControllerTest {

    @Test
    void partnerAppealSitesUseAllSitesAssociatedWithCurrentAccount() {
        TOrderAppealService appealService = mock(TOrderAppealService.class);
        TokenService tokenService = mock(TokenService.class);
        SiteClient siteClient = mock(SiteClient.class);
        TOrderAppealController controller = new TOrderAppealController(appealService, tokenService);
        ReflectionTestUtils.setField(controller, "siteClient", siteClient);

        SysUser partnerUser = new SysUser();
        partnerUser.setUserId(7L);
        partnerUser.setObjectId(9);
        when(siteClient.getSiteListByUserId(7L))
                .thenReturn(R.ok(Arrays.asList(site(-1), site(11), null, site(12))));

        Map<Integer, GetSiteListDTO> result = controller.getVisibleSiteMap(partnerUser);

        assertEquals(new HashSet<>(Arrays.asList(11, 12)), result.keySet());
        verify(siteClient).getSiteListByUserId(7L);
    }

    private GetSiteListDTO site(Integer id) {
        GetSiteListDTO site = new GetSiteListDTO();
        site.setId(id);
        return site;
    }
}