package cn.celess.blog.controller;

import cn.celess.blog.BaseTest;
import cn.celess.blog.entity.model.VisitorModel;
import com.github.pagehelper.PageInfo;
import net.sf.json.JSONObject;
import org.junit.Test;
import org.springframework.test.web.servlet.result.MockMvcResultHandlers;

import static org.junit.Assert.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
import static cn.celess.blog.enmu.ResponseEnum.*;

public class VisitorControllerTest extends BaseTest {

    @Test
    public void getVisitorCount() throws Exception {
        mockMvc.perform(get("/visitor/count")).andDo(result -> {
            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
            assertEquals(SUCCESS.getCode(), object.getInt(Code));
            assertTrue(object.containsKey(Result));
        });
    }

    @Test
    public void page() throws Exception {
        int count = 10;
        int page = 1;
        mockMvc.perform(get("/admin/visitor/page?count=" + count + "&page=" + page).header("Authorization", adminLogin()))
                .andDo(result -> {
                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
                    JSONObject resultJson = JSONObject.fromObject(object.getJSONObject(Result));
                    PageInfo pageInfo = (PageInfo) JSONObject.toBean(resultJson, PageInfo.class);
                    assertNotEquals(0, pageInfo.getTotal());
                    assertNotEquals(0, pageInfo.getStartRow());
                    assertNotEquals(0, pageInfo.getEndRow());
                    assertEquals(1, pageInfo.getPageNum());
                    assertEquals(10, pageInfo.getPageSize());
                    for (Object ver : pageInfo.getList()) {
                        VisitorModel v = (VisitorModel) JSONObject.toBean(JSONObject.fromObject(ver), VisitorModel.class);
                        assertNotEquals(0, v.getId());
                        assertNotNull(v.getDate());
                    }
                });
    }

    @Test
    public void add() throws Exception {
        mockMvc.perform(post("/visit")).andDo(MockMvcResultHandlers.print()).andDo(result -> {
            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
            assertEquals(SUCCESS.getCode(), object.getInt(Code));
            VisitorModel visitorModel = (VisitorModel) JSONObject.toBean(object.getJSONObject(Result), VisitorModel.class);
            assertNotEquals(0, visitorModel.getId());
            assertNotNull(visitorModel.getIp());
        });
    }

    @Test
    public void dayVisitCount() throws Exception {
        mockMvc.perform(get("/dayVisitCount")).andDo(MockMvcResultHandlers.print()).andDo(result ->
                assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
        );
    }

    // 手动测试
    // @Test
    public void ipLocation() throws Exception {
        String ip = "127.0.0.1";
        mockMvc.perform(get("/ip/" + ip)).andDo(MockMvcResultHandlers.print()).andDo(result -> {
            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
            assertTrue(JSONObject.fromObject(result.getResponse().getContentAsString()).containsKey(Result));
        });
    }

    @Test
    public void getIp() throws Exception {
        mockMvc.perform(get("/ip")).andDo(MockMvcResultHandlers.print()).andDo(result -> {
            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
            assertEquals("127.0.0.1", JSONObject.fromObject(result.getResponse().getContentAsString()).getString(Result));
        });
    }
}