package com.epmet;

import com.epmet.common.token.constant.LoginConstant;
import com.epmet.commons.tools.security.dto.GovTokenDto;
import com.epmet.commons.tools.utils.CpUserDetailRedis;
import com.epmet.jwt.JwtTokenProperties;
import com.epmet.jwt.JwtTokenUtils;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

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

@RunWith(SpringRunner.class)
@SpringBootTest
public class TokenGenTest {

    @Autowired
    private JwtTokenProperties jwtTokenProperties;

    @Autowired
    private JwtTokenUtils jwtTokenUtils;

    @Autowired
    private CpUserDetailRedis cpUserDetailRedis;

    @Test
    public void genToken() {
        String staffId = "wxz";
        String tokenStr = generateGovWxmpToken(staffId);
        int expire = jwtTokenProperties.getExpire();
        GovTokenDto govTokenDto = new GovTokenDto();
        govTokenDto.setApp(LoginConstant.APP_GOV);
        govTokenDto.setClient(LoginConstant.CLIENT_WXMP);
        govTokenDto.setUserId(staffId);
        govTokenDto.setOpenId("");
        govTokenDto.setSessionKey("");
        govTokenDto.setUnionId("");
        govTokenDto.setToken(tokenStr);
        govTokenDto.setUpdateTime(System.currentTimeMillis());
        govTokenDto.setExpireTime(jwtTokenUtils.getExpiration(tokenStr).getTime());
        govTokenDto.setAgencyId("1");
        govTokenDto.setDeptIdList(new HashSet<>(Arrays.asList("1","2","3")));
        govTokenDto.setCustomerId("f76def116c9c2dc0269cc17867af122c");
        cpUserDetailRedis.set(govTokenDto, expire);
    }

    /**
     * @Description 生成token
     * @Date 2020/4/18 23:04
     **/
    private String generateGovWxmpToken(String staffId) {
        Map<String, Object> map = new HashMap<>();
        map.put("app", LoginConstant.APP_GOV);
        map.put("client", LoginConstant.CLIENT_WXMP);
        map.put("userId", staffId);
        String token = jwtTokenUtils.createToken(map);
        return token;
    }

}