package controller

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"testing"

	"ezops/model"

	"github.com/gin-gonic/gin"
	"github.com/glebarez/sqlite"
	"gorm.io/gorm"
	gormlogger "gorm.io/gorm/logger"
)

func TestReceiveReportMatchesExistingAssetByIP(t *testing.T) {
	gin.SetMode(gin.TestMode)
	db, err := gorm.Open(sqlite.Open("file:agent_report_match?mode=memory&cache=shared"), &gorm.Config{
		Logger: gormlogger.Default.LogMode(gormlogger.Silent),
	})
	if err != nil {
		t.Fatalf("打开测试数据库失败: %v", err)
	}
	if err := db.AutoMigrate(&model.Server{}, &model.AgentReport{}, &model.HealthProbe{}); err != nil {
		t.Fatalf("迁移测试表失败: %v", err)
	}

	server := model.Server{Hostname: "c-redis-cluster-3", IP: "172.19.16.59", Status: "running"}
	if err := db.Create(&server).Error; err != nil {
		t.Fatalf("创建测试资产失败: %v", err)
	}

	body := []byte(`{
		"hostname":"VM-16-59-tencentos",
		"ip":"172.19.16.59",
		"agent_version":"v2.1.0",
		"cpu":{"usage_percent":12.5,"iowait":0.5,"core_count":4},
		"memory":{"total_gb":7.8,"used_gb":3.1,"available_gb":4.7,"usage_percent":39.7},
		"disks":[{"mount":"/","total_gb":100,"used_gb":25,"usage_percent":25}],
		"system":{"os_version":"TencentOS Server 4","kernel":"6.6.0","uptime":3600}
	}`)
	req := httptest.NewRequest(http.MethodPost, "/api/v1/agent/report", bytes.NewReader(body))
	req.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()
	c, _ := gin.CreateTestContext(recorder)
	c.Request = req

	NewAgentHandler(db, nil, nil).ReceiveReport(c)
	if recorder.Code != http.StatusOK {
		t.Fatalf("上报返回码 = %d, body=%s", recorder.Code, recorder.Body.String())
	}

	var gotServer model.Server
	if err := db.Take(&gotServer, server.ID).Error; err != nil {
		t.Fatalf("读取资产失败: %v", err)
	}
	if gotServer.Hostname != server.Hostname || gotServer.AgentStatus != "online" {
		t.Fatalf("资产关联错误: hostname=%q status=%q", gotServer.Hostname, gotServer.AgentStatus)
	}
	if gotServer.OS != "TencentOS Server 4" || gotServer.Kernel != "6.6.0" || gotServer.Uptime != 3600 {
		t.Fatalf("系统字段更新错误: os=%q kernel=%q uptime=%d", gotServer.OS, gotServer.Kernel, gotServer.Uptime)
	}

	var report model.AgentReport
	if err := db.Order("id DESC").Take(&report).Error; err != nil {
		t.Fatalf("读取 Agent 报告失败: %v", err)
	}
	if report.Hostname != server.Hostname {
		t.Fatalf("报告主机名未归一: got=%q want=%q", report.Hostname, server.Hostname)
	}

	var serverCount int64
	db.Model(&model.Server{}).Count(&serverCount)
	if serverCount != 1 {
		t.Fatalf("上报后产生重复资产: count=%d", serverCount)
	}
}