package xorm

import (
	"testing"

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

func TestBelongsTo_Get(t *testing.T) {
	assert.NoError(t, prepareEngine())

	type Face struct {
		Id   int64
		Name string
	}

	type Nose struct {
		Id   int64
		Face Face `xorm:"belongs_to"`
	}

	err := testEngine.Sync2(new(Nose), new(Face))
	assert.NoError(t, err)

	var face = Face{
		Name: "face1",
	}
	_, err = testEngine.Insert(&face)
	assert.NoError(t, err)

	var cfgFace Face
	has, err := testEngine.Get(&cfgFace)
	assert.NoError(t, err)
	assert.Equal(t, true, has)
	assert.Equal(t, cfgFace, face)

	var nose = Nose{Face: face}
	_, err = testEngine.Insert(&nose)
	assert.NoError(t, err)

	var cfgNose Nose
	has, err = testEngine.Get(&cfgNose)
	assert.NoError(t, err)
	assert.Equal(t, true, has)
	assert.Equal(t, cfgNose.Id, nose.Id)
	// FIXME: the id should be set back to the field
	//assert.Equal(t, cfgNose.Face.Id, nose.Face.Id)
	assert.Equal(t, "", cfgNose.Face.Name)

	var cfgNose2 Nose
	has, err = testEngine.Cascade().Get(&cfgNose2)
	assert.NoError(t, err)
	assert.Equal(t, true, has)
	assert.Equal(t, cfgNose2.Id, nose.Id)
	assert.Equal(t, cfgNose2.Face.Id, nose.Face.Id)
	assert.Equal(t, "face1", cfgNose2.Face.Name)
}

func TestBelongsTo_GetPtr(t *testing.T) {
	assert.NoError(t, prepareEngine())

	type Face struct {
		Id   int64
		Name string
	}

	type Nose struct {
		Id   int64
		Face *Face `xorm:"belongs_to"`
	}

	err := testEngine.Sync2(new(Nose), new(Face))
	assert.NoError(t, err)

	var face = Face{
		Name: "face1",
	}
	_, err = testEngine.Insert(&face)
	assert.NoError(t, err)

	var cfgFace Face
	has, err := testEngine.Get(&cfgFace)
	assert.NoError(t, err)
	assert.Equal(t, true, has)
	assert.Equal(t, cfgFace, face)

	var nose = Nose{Face: &face}
	_, err = testEngine.Insert(&nose)
	assert.NoError(t, err)

	var cfgNose Nose
	has, err = testEngine.Get(&cfgNose)
	assert.NoError(t, err)
	assert.Equal(t, true, has)
	assert.Equal(t, cfgNose.Id, nose.Id)
	// FIXME: the id should be set back to the field
	//assert.Equal(t, cfgNose.Face.Id, nose.Face.Id)

	var cfgNose2 Nose
	has, err = testEngine.Cascade().Get(&cfgNose2)
	assert.NoError(t, err)
	assert.Equal(t, true, has)
	assert.Equal(t, cfgNose2.Id, nose.Id)
	assert.Equal(t, cfgNose2.Face.Id, nose.Face.Id)
	assert.Equal(t, "face1", cfgNose2.Face.Name)
}

func TestBelongsTo_Find(t *testing.T) {
	assert.NoError(t, prepareEngine())

	type Face struct {
		Id   int64
		Name string
	}

	type Nose struct {
		Id   int64
		Face Face `xorm:"belongs_to"`
	}

	err := testEngine.Sync2(new(Nose), new(Face))
	assert.NoError(t, err)

	var face1 = Face{
		Name: "face1",
	}
	var face2 = Face{
		Name: "face2",
	}
	_, err = testEngine.Insert(&face1, &face2)
	assert.NoError(t, err)

	var noses = []Nose{
		{Face: face1},
		{Face: face2},
	}
	_, err = testEngine.Insert(&noses)
	assert.NoError(t, err)

	var noses1 []Nose
	err = testEngine.Find(&noses1)
	assert.NoError(t, err)
	assert.Equal(t, 2, len(noses1))
	// FIXME:
	//assert.Equal(t, face1.Id, noses1[0].Face.Id)
	//assert.Equal(t, face2.Id, noses1[1].Face.Id)
	assert.Equal(t, "", noses1[0].Face.Name)
	assert.Equal(t, "", noses1[1].Face.Name)
}