package models

import (
	"encoding/json"
	"testing"

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

type dgTestNode struct {
	Name string `json:"Name"`
}

func (n *dgTestNode) Id() string {
	return n.Name
}

func dependencyGraphAddTestData(dg *DependencyGraph[*dgTestNode]) {
	dg.AddDependency(&dgTestNode{Name: "a"}, &dgTestNode{Name: "b"})
	dg.AddDependency(&dgTestNode{Name: "a"}, &dgTestNode{Name: "c"})
	dg.AddDependency(&dgTestNode{Name: "b"}, &dgTestNode{Name: "c"})
	dg.AddDependency(&dgTestNode{Name: "c"}, &dgTestNode{Name: "d"})
}

func TestDependencyGraphIsPresent(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	assert.False(t, dg.Present())

	dg.SetPresent(true)
	assert.True(t, dg.Present())
}

func TestDependencyGraphGetDependencies(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	dependencyGraphAddTestData(dg)

	assert.Equal(t, []*dgTestNode{{Name: "b"}, {Name: "c"}}, dg.GetDependencies(&dgTestNode{Name: "a"}))
	assert.Equal(t, []*dgTestNode{{Name: "c"}}, dg.GetDependencies(&dgTestNode{Name: "b"}))
	assert.Equal(t, []*dgTestNode{{Name: "d"}}, dg.GetDependencies(&dgTestNode{Name: "c"}))
	assert.Equal(t, []*dgTestNode{}, dg.GetDependencies(&dgTestNode{Name: "d"}))
}

func TestDependencyGraphGetDependents(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	dependencyGraphAddTestData(dg)

	assert.Equal(t, []*dgTestNode{}, dg.GetDependents(&dgTestNode{Name: "a"}))
	assert.Equal(t, []*dgTestNode{{Name: "a"}}, dg.GetDependents(&dgTestNode{Name: "b"}))
	assert.Equal(t, []*dgTestNode{{Name: "a"}, {Name: "b"}}, dg.GetDependents(&dgTestNode{Name: "c"}))
	assert.Equal(t, []*dgTestNode{{Name: "c"}}, dg.GetDependents(&dgTestNode{Name: "d"}))
}

func TestDependencyGraphGetNodes(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	dependencyGraphAddTestData(dg)

	nodes := dg.GetNodes()
	assert.Contains(t, nodes, &dgTestNode{Name: "a"})
	assert.Contains(t, nodes, &dgTestNode{Name: "b"})
	assert.Contains(t, nodes, &dgTestNode{Name: "c"})
	assert.Contains(t, nodes, &dgTestNode{Name: "d"})
}

func TestDependencyGraphPathToRoot(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	dependencyGraphAddTestData(dg)

	assert.Equal(t,
		[]*dgTestNode{
			{Name: "d"},
			{Name: "c"},
			{Name: "a"},
		}, dg.PathToRoot(&dgTestNode{Name: "d"}))
}

func TestDependencyGraphMarshalJSON(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	dependencyGraphAddTestData(dg)
	dg.SetPresent(true)

	json, err := json.Marshal(dg)
	assert.Nil(t, err)
	assert.Equal(t, "{\"present\":true,\"nodes\":{\"a\":{\"data\":{\"Name\":\"a\"},\"children\":[{\"Name\":\"b\"},{\"Name\":\"c\"}]},\"b\":{\"data\":{\"Name\":\"b\"},\"children\":[{\"Name\":\"c\"}]},\"c\":{\"data\":{\"Name\":\"c\"},\"children\":[{\"Name\":\"d\"}]},\"d\":{\"data\":{\"Name\":\"d\"},\"children\":[]}}}", string(json))
}

func TestDependencyGraphUnmarshalJSON(t *testing.T) {
	dg := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg)

	dependencyGraphAddTestData(dg)
	dg.SetPresent(true)

	data, err := json.Marshal(dg)
	assert.Nil(t, err)

	dg2 := NewDependencyGraph[*dgTestNode]()
	assert.NotNil(t, dg2)

	err = json.Unmarshal(data, dg2)
	assert.Nil(t, err)
	assert.Equal(t, dg, dg2)
}