package models

import (
	"encoding/json"
	"testing"
	"time"

	"github.com/shopspring/decimal"
	"github.com/stretchr/testify/assert"
)

func TestTask_TableName(t *testing.T) {
	task := Task{}
	assert.Equal(t, "task_task", task.TableName())
}

func TestTaskCategory_TableName(t *testing.T) {
	cat := TaskCategory{}
	assert.Equal(t, "task_taskcategory", cat.TableName())
}

func TestTaskPriority_TableName(t *testing.T) {
	p := TaskPriority{}
	assert.Equal(t, "task_taskpriority", p.TableName())
}

func TestTaskStatus_TableName(t *testing.T) {
	s := TaskStatus{}
	assert.Equal(t, "task_taskstatus", s.TableName())
}

func TestTaskFrequency_TableName(t *testing.T) {
	f := TaskFrequency{}
	assert.Equal(t, "task_taskfrequency", f.TableName())
}

func TestTaskCompletion_TableName(t *testing.T) {
	c := TaskCompletion{}
	assert.Equal(t, "task_taskcompletion", c.TableName())
}

func TestContractor_TableName(t *testing.T) {
	c := Contractor{}
	assert.Equal(t, "task_contractor", c.TableName())
}

func TestContractorSpecialty_TableName(t *testing.T) {
	s := ContractorSpecialty{}
	assert.Equal(t, "task_contractorspecialty", s.TableName())
}

func TestDocument_TableName(t *testing.T) {
	d := Document{}
	assert.Equal(t, "task_document", d.TableName())
}

func TestTask_JSONSerialization(t *testing.T) {
	dueDate := time.Date(2024, 12, 31, 0, 0, 0, 0, time.UTC)
	cost := decimal.NewFromFloat(150.50)

	task := Task{
		ResidenceID:   1,
		CreatedByID:   1,
		Title:         "Fix leaky faucet",
		Description:   "Kitchen faucet is dripping",
		DueDate:       &dueDate,
		EstimatedCost: &cost,
		IsCancelled:   false,
		IsArchived:    false,
	}
	task.ID = 1

	data, err := json.Marshal(task)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(1), result["id"])
	assert.Equal(t, float64(1), result["residence_id"])
	assert.Equal(t, float64(1), result["created_by_id"])
	assert.Equal(t, "Fix leaky faucet", result["title"])
	assert.Equal(t, "Kitchen faucet is dripping", result["description"])
	assert.Equal(t, "150.5", result["estimated_cost"]) // Decimal serializes as string
	assert.Equal(t, false, result["is_cancelled"])
	assert.Equal(t, false, result["is_archived"])
}

func TestTaskCategory_JSONSerialization(t *testing.T) {
	cat := TaskCategory{
		Name:         "Plumbing",
		Description:  "Plumbing related tasks",
		Icon:         "wrench",
		Color:        "#3498db",
		DisplayOrder: 1,
	}
	cat.ID = 1

	data, err := json.Marshal(cat)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(1), result["id"])
	assert.Equal(t, "Plumbing", result["name"])
	assert.Equal(t, "Plumbing related tasks", result["description"])
	assert.Equal(t, "wrench", result["icon"])
	assert.Equal(t, "#3498db", result["color"])
	assert.Equal(t, float64(1), result["display_order"])
}

func TestTaskPriority_JSONSerialization(t *testing.T) {
	priority := TaskPriority{
		Name:         "High",
		Level:        3,
		Color:        "#e74c3c",
		DisplayOrder: 3,
	}
	priority.ID = 3

	data, err := json.Marshal(priority)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(3), result["id"])
	assert.Equal(t, "High", result["name"])
	assert.Equal(t, float64(3), result["level"])
	assert.Equal(t, "#e74c3c", result["color"])
}

func TestTaskStatus_JSONSerialization(t *testing.T) {
	status := TaskStatus{
		Name:         "In Progress",
		Description:  "Task is being worked on",
		Color:        "#3498db",
		DisplayOrder: 2,
	}
	status.ID = 2

	data, err := json.Marshal(status)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(2), result["id"])
	assert.Equal(t, "In Progress", result["name"])
	assert.Equal(t, "Task is being worked on", result["description"])
	assert.Equal(t, "#3498db", result["color"])
}

func TestTaskFrequency_JSONSerialization(t *testing.T) {
	days := 7
	freq := TaskFrequency{
		Name:         "Weekly",
		Days:         &days,
		DisplayOrder: 3,
	}
	freq.ID = 3

	data, err := json.Marshal(freq)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(3), result["id"])
	assert.Equal(t, "Weekly", result["name"])
	assert.Equal(t, float64(7), result["days"])
}

func TestTaskCompletion_JSONSerialization(t *testing.T) {
	completedAt := time.Date(2024, 6, 15, 14, 30, 0, 0, time.UTC)
	cost := decimal.NewFromFloat(125.00)

	completion := TaskCompletion{
		TaskID:        1,
		CompletedByID: 2,
		CompletedAt:   completedAt,
		Notes:         "Fixed the leak",
		ActualCost:    &cost,
	}
	completion.ID = 1

	data, err := json.Marshal(completion)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(1), result["id"])
	assert.Equal(t, float64(1), result["task_id"])
	assert.Equal(t, float64(2), result["completed_by_id"])
	assert.Equal(t, "Fixed the leak", result["notes"])
	assert.Equal(t, "125", result["actual_cost"]) // Decimal serializes as string
}

func TestContractor_JSONSerialization(t *testing.T) {
	contractor := Contractor{
		ResidenceID: 1,
		CreatedByID: 1,
		Name:        "Mike's Plumbing",
		Company:     "Mike's Plumbing Co.",
		Phone:       "+1-555-1234",
		Email:       "mike@plumbing.com",
		Website:     "https://mikesplumbing.com",
		Notes:       "Great service",
		IsFavorite:  true,
		IsActive:    true,
	}
	contractor.ID = 1

	data, err := json.Marshal(contractor)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(1), result["id"])
	assert.Equal(t, float64(1), result["residence_id"])
	assert.Equal(t, "Mike's Plumbing", result["name"])
	assert.Equal(t, "Mike's Plumbing Co.", result["company"])
	assert.Equal(t, "+1-555-1234", result["phone"])
	assert.Equal(t, "mike@plumbing.com", result["email"])
	assert.Equal(t, "https://mikesplumbing.com", result["website"])
	assert.Equal(t, true, result["is_favorite"])
	assert.Equal(t, true, result["is_active"])
}

func TestDocument_JSONSerialization(t *testing.T) {
	purchaseDate := time.Date(2023, 6, 15, 0, 0, 0, 0, time.UTC)
	expiryDate := time.Date(2028, 6, 15, 0, 0, 0, 0, time.UTC)
	price := decimal.NewFromFloat(5000.00)

	doc := Document{
		ResidenceID:   1,
		CreatedByID:   1,
		Title:         "HVAC Warranty",
		Description:   "Warranty for central air",
		DocumentType:  "warranty",
		FileURL:       "/uploads/hvac.pdf",
		FileName:      "hvac.pdf",
		PurchaseDate:  &purchaseDate,
		ExpiryDate:    &expiryDate,
		PurchasePrice: &price,
		Vendor:        "Cool Air HVAC",
		SerialNumber:  "HVAC-123",
	}
	doc.ID = 1

	data, err := json.Marshal(doc)
	assert.NoError(t, err)

	var result map[string]interface{}
	err = json.Unmarshal(data, &result)
	assert.NoError(t, err)

	assert.Equal(t, float64(1), result["id"])
	assert.Equal(t, "HVAC Warranty", result["title"])
	assert.Equal(t, "warranty", result["document_type"])
	assert.Equal(t, "/uploads/hvac.pdf", result["file_url"])
	assert.Equal(t, "Cool Air HVAC", result["vendor"])
	assert.Equal(t, "HVAC-123", result["serial_number"])
	assert.Equal(t, "5000", result["purchase_price"]) // Decimal serializes as string
}