package cn.huge.module;

import java.util.*;
import java.math.*;
import cn.huge.base.common.utils.ReturnFailUtils;
import cn.huge.base.common.utils.ReturnSucUtils;
import cn.huge.module.knowledge.controller.web.LawProvisionWebController;
import cn.huge.module.knowledge.domain.po.LawProvision;
import cn.huge.module.knowledge.service.LawProvisionService;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
import org.springframework.test.web.servlet.setup.SharedHttpSessionConfigurer;

import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class LawProvisionWebControllerTest {

    private MockMvc mockMvc;

    @Mock
    private LawProvisionService service;

    @InjectMocks
    private LawProvisionWebController controller;

    @Mock
    private HttpServletRequest request;

    private AutoCloseable closeable;

    @Before
    public void setUp() {
        closeable = MockitoAnnotations.openMocks(this);
        mockMvc = MockMvcBuilders.standaloneSetup(controller).build();
    }

    @After
    public void tearDown() throws Exception {
        closeable.close();
    }

    @Test
    public void pageQuery_ValidParameters_ReturnsSuccess() throws Exception {
        // Arrange
        int page = 1;
        int size = 10;
        Map<String, Object> terms = new HashMap<>();
        Page<LawProvision> provisionPage = new PageImpl<>(new ArrayList<>());

        // Mocking service response
        PageRequest pageRequest = PageRequest.of(page - 1, size, Sort.by(Sort.Direction.DESC, "create_time"));
        when(service.pageQuery(pageRequest, terms)).thenReturn(provisionPage);

        // Mocking request parameters
        // when(request.getParameter("lawInfoId")).thenReturn("123");
        // when(request.getParameter("lawTitle")).thenReturn("Test Law");
        // ... Mock other parameters as needed

        // Act & Assert
        mockMvc.perform(MockMvcRequestBuilders.get("/api/web/pageQuery")
                        .param("page", String.valueOf(page))
                        .param("size", String.valueOf(size))
                        .accept(MediaType.APPLICATION_JSON))
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.code").value(0))
                .andExpect(MockMvcResultMatchers.jsonPath("$.msg").value("处理成功"));
    }

    @Test
    public void pageQuery_ExceptionThrown_ReturnsFailure() throws Exception {
        // Arrange
        int page = 1;
        int size = 10;
        Map<String, Object> terms = new HashMap<>();

        // Mocking service to throw exception
        doThrow(new RuntimeException("Test Exception")).when(service).pageQuery(any(PageRequest.class), any(Map.class));

        // Mocking request parameters
        when(request.getParameter("lawInfoId")).thenReturn("123");
        when(request.getParameter("lawTitle")).thenReturn("Test Law");
        // ... Mock other parameters as needed

        // Act & Assert
        mockMvc.perform(MockMvcRequestBuilders.get("/api/web/pageQuery")
                        .param("page", String.valueOf(page))
                        .param("size", String.valueOf(size))
                        .accept(MediaType.APPLICATION_JSON))
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.code").value(-1))
                .andExpect(MockMvcResultMatchers.jsonPath("$.msg").value("系统异常，请稍后再试！"));
    }
}