package service

import (
	"testing"
	"time"

	"github.com/DATA-DOG/go-sqlmock"
	"github.com/wmms/wmms/pkg/database"
)

func organizationAccessMock(t *testing.T) sqlmock.Sqlmock {
	t.Helper()
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatal(err)
	}
	previous := database.DB
	database.DB = db
	t.Cleanup(func() { database.DB = previous; db.Close() })
	return mock
}

// 组织初始化列表要为每家公司附带与切换接口一致的仓库摘要。这里覆盖
// 无仓库（null）、仓库停用（如实返回停用状态）、启用唯一仓库加启用库位三类。
func TestOrganizationAccessListIncludesWarehouseSummary(t *testing.T) {
	mock := organizationAccessMock(t)
	now := time.Now()
	organizationRows := sqlmock.NewRows([]string{"id", "code", "name", "status", "default_timezone", "created_at", "updated_at", "is_default"}).
		AddRow(2, "TWO", "无仓库公司", "active", "Asia/Shanghai", now, now, false).
		AddRow(3, "THREE", "停用仓库公司", "active", "Asia/Shanghai", now, now, false).
		AddRow(4, "FOUR", "启用仓库公司", "active", "Asia/Shanghai", now, now, true)
	mock.ExpectQuery("SELECT o.id,o.code,o.name,o.status").WithArgs(int64(7), int64(7)).WillReturnRows(organizationRows)

	warehouseColumns := []string{"id", "organization_id", "code", "name", "status", "location_id", "location_code", "location_name", "location_status"}
	// 无仓库：查询返回空集，摘要为 null。
	mock.ExpectQuery("SELECT w.id, w.organization_id").WithArgs(int64(2)).WillReturnRows(sqlmock.NewRows(warehouseColumns))
	// 仓库停用：仍返回该公司自己的仓库与库位，状态为停用。
	mock.ExpectQuery("SELECT w.id, w.organization_id").WithArgs(int64(3)).
		WillReturnRows(sqlmock.NewRows(warehouseColumns).AddRow(30, 3, "D3", "停用仓库", "inactive", 31, "L3", "停用仓库的库位", "active"))
	// 启用唯一仓库加启用库位。
	mock.ExpectQuery("SELECT w.id, w.organization_id").WithArgs(int64(4)).
		WillReturnRows(sqlmock.NewRows(warehouseColumns).AddRow(40, 4, "D4", "启用仓库", "active", 41, "L4", "启用仓库的库位", "active"))

	access, err := NewOrganizationAccessService().List(7)
	if err != nil {
		t.Fatal(err)
	}
	if len(access.Organizations) != 3 || access.DefaultOrganizationID == nil || *access.DefaultOrganizationID != 4 {
		t.Fatalf("公司列表错误 %+v", access)
	}
	byID := make(map[int64]int, len(access.Organizations))
	for i, item := range access.Organizations {
		byID[item.ID] = i
	}
	if warehouse := access.Organizations[byID[2]].Warehouse; warehouse != nil {
		t.Fatalf("无仓库公司应返回空摘要，实际 %+v", warehouse)
	}
	disabled := access.Organizations[byID[3]].Warehouse
	if disabled == nil || disabled.ID != 30 || disabled.Status != "inactive" ||
		disabled.Location == nil || disabled.Location.ID != 31 || disabled.Location.Status != "active" {
		t.Fatalf("停用仓库摘要错误 %+v", disabled)
	}
	enabled := access.Organizations[byID[4]].Warehouse
	if enabled == nil || enabled.ID != 40 || enabled.Status != "active" ||
		enabled.Location == nil || enabled.Location.ID != 41 || enabled.Location.Status != "active" {
		t.Fatalf("启用仓库摘要错误 %+v", enabled)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatal(err)
	}
}