package mergo_test

import (
	"reflect"
	"testing"

	"github.com/GGabriele/mergo"
)

type asd struct {
	A *bool
}

type boolTransformer struct{}

func Bool(v bool) *bool { return &v }

func (t boolTransformer) Transformer(typ reflect.Type) func(dst, src reflect.Value) error {
	if typ == reflect.TypeOf(Bool(false)) {
		return func(dst, src reflect.Value) error {
			if dst.CanSet() {
				if src.IsNil() {
					src.Set(dst)
				}
			}
			return nil
		}
	}
	return nil
}

func TestBoolPointers(t *testing.T) {
	// base false, augment true
	base := asd{
		A: Bool(false),
	}
	augment := asd{
		A: Bool(true),
	}
	if err := mergo.Merge(&base, augment, mergo.WithTransformers(boolTransformer{})); err != nil {
		t.Error(err)
	}
	if *base.A != false {
		t.Errorf("1. base.A should be false")
	}

	// base true, augment false
	base = asd{
		A: Bool(true),
	}
	augment = asd{
		A: Bool(false),
	}
	if err := mergo.Merge(&base, augment, mergo.WithTransformers(boolTransformer{})); err != nil {
		t.Error(err)
	}
	if *base.A != true {
		t.Errorf("2. base.A should be true")
	}

	// base empty, augment false
	base = asd{}
	augment = asd{
		A: Bool(false),
	}
	if err := mergo.Merge(&base, augment, mergo.WithTransformers(boolTransformer{})); err != nil {
		t.Error(err)
	}
	if *base.A != false {
		t.Errorf("3. base.A should be false")
	}

	// base empty, augment true
	base = asd{}
	augment = asd{
		A: Bool(true),
	}
	if err := mergo.Merge(&base, augment, mergo.WithTransformers(boolTransformer{})); err != nil {
		t.Error(err)
	}
	if *base.A != true {
		t.Errorf("4. base.A should be true")
	}
}