container_test.go

package ioc_test

import (
	"context"
	"fmt"
	"reflect"
	"testing"
	"time"

	"github.com/mylxsw/go-ioc"
)

type GetUserInterface interface {
	GetUser() string
}

type GetRoleInterface interface {
	GetRole() string
}

type RoleService struct{}

func (r RoleService) GetRole() string {
	return "admin"
}

type UserService struct {
	repo *UserRepo
}

func (u *UserService) GetUser() string {
	return fmt.Sprintf("lookupInstance user from connection: %s", u.repo.connStr)
}

type UserRepo struct {
	connStr string
}

var expectedValue = "lookupInstance user from connection: root:root@/my_db?charset=utf8"

// TestPrototype 测试原型模式
func TestPrototype(t *testing.T) {
	c := ioc.New()

	c.MustBindValue("conn_str", "root:root@/my_db?charset=utf8")
	c.MustSingleton(func(c ioc.Container) (*UserRepo, error) {
		connStr, err := c.Get("conn_str")
		if err != nil {
			return nil, err
		}

		return &UserRepo{connStr: connStr.(string)}, nil
	})
	c.MustPrototype(func(userRepo *UserRepo) *UserService {
		return &UserService{repo: userRepo}
	})

	if err := c.Resolve(func(userService *UserService) {
		if userService.GetUser() != expectedValue {
			t.Error("test failed")
		}
	}); err != nil {
		t.Errorf("test failed: %s", err)
		return
	}
	{
		userService, err := c.Get(new(UserService))
		if err != nil {
			t.Error(err)
			return
		}

		fmt.Println(userService.(*UserService).GetUser())
	}
	// reflect.TypeOf((*UserService)(nil))
	{
		userService, err := c.Get(reflect.TypeOf((*UserService)(nil)))
		if err != nil {
			t.Error(err)
			return
		}

		if userService.(*UserService).GetUser() != expectedValue {
			t.Error("test failed")
		}
	}

	{
		userService, err := c.Get((*UserService)(nil))
		if err != nil {
			t.Error(err)
			return
		}

		if userService.(*UserService).GetUser() != expectedValue {
			t.Error("test failed")
		}
	}

	{
		c.MustResolve(func(cc ioc.Container) {
			userService, err := c.Get((*UserService)(nil))
			if err != nil {
				t.Error(err)
				return
			}

			if userService.(*UserService).GetUser() != expectedValue {
				t.Error("test failed")
			}
		})
	}
}

// TestInterfaceInjection 测试接口注入
func TestInterfaceInjection(t *testing.T) {
	c := ioc.New()
	c.MustBindValue("conn_str", "root:root@/my_db?charset=utf8")
	c.MustSingleton(func(c ioc.Container) (*UserRepo, error) {
		connStr, err := c.Get("conn_str")
		if err != nil {
			return nil, err
		}

		return &UserRepo{connStr: connStr.(string)}, nil
	})
	c.MustPrototype(func(userRepo *UserRepo) (*UserService, error) {
		return &UserService{repo: userRepo}, nil
	})

	// if err := c.Resolve(func(userService GetUserInterface) {
	// 	if userService.GetUser() != expectedValue {
	// 		t.Error("test failed")
	// 	}
	// }); err != nil {
	// 	t.Errorf("test failed: %s", err)
	// }

	c.MustPrototype(func() (RoleService, error) {
		return RoleService{}, nil
	})

	// err := c.Resolve(func(roleService GetRoleInterface) {
	// 	if roleService.GetRole() != "admin" {
	// 		t.Error("test failed")
	// 	}
	// })
	// if err != nil {
	// 	t.Error(err)
	// }

	for _, k := range c.Keys() {
		fmt.Println(k)
	}
}

// TestWithContext 测试默认添加 Context 实例
func TestWithContext(t *testing.T) {
	ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
	defer cancel()

	c := ioc.NewWithContext(ctx)
	c.MustResolve(func(ctx context.Context) {
		startTime := time.Now().UnixNano()
		<-ctx.Done()

		if (time.Now().UnixNano()-startTime)/1000/1000 < 100 {
			t.Error("test failed")
		}
	})

	time.Sleep(200 * time.Millisecond)
}

type TestObject struct {
	Name string
}

// TestWithProvider 测试使用 Provider 提供额外的实例配置
func TestWithProvider(t *testing.T) {
	c := ioc.New()
	c.MustBindValue("conn_str", "root:root@/my_db?charset=utf8")
	c.MustSingleton(func(c ioc.Container) (*UserRepo, error) {
		connStr, err := c.Get("conn_str")
		if err != nil {
			return nil, err
		}

		return &UserRepo{connStr: connStr.(string)}, nil
	})
	c.MustPrototype(func(userRepo *UserRepo) *UserService {
		return &UserService{repo: userRepo}
	})

	if _, err := c.CallWithProvider(func(userService *UserService, testObject *TestObject) {
		if userService.GetUser() != expectedValue {
			t.Error("test failed")
		}

		if testObject.Name != "mylxsw" {
			t.Error("test failed")
		}
	}, c.Provider(func() *TestObject {
		return &TestObject{Name: "mylxsw"}
	})); err != nil {
		t.Errorf("test failed: %s", err)
		return
	}
}

// TestBindValue 测试直接绑定实体对象
func TestBindValue(t *testing.T) {
	c := ioc.New()
	userRepoStruct := UserRepo{connStr: "user struct"}
	userRepoPointer := &UserRepo{connStr: "user pointer"}

	c.MustSingleton(userRepoStruct)
	c.MustSingleton(userRepoPointer)

	c.MustResolve(func(r UserRepo) {
		if r.connStr != "user struct" {
			t.Error("test failed")
		}
	})

	c.MustResolve(func(r *UserRepo) {
		if r.connStr != "user pointer" {
			t.Error("test failed")
		}
	})
}

func TestSearchAdvanced(t *testing.T) {
	c := ioc.New()
	c.MustSingleton(func() *UserRepo {
		return &UserRepo{connStr: "this is user repo"}
	})
	c.MustSingleton(func(userRepo *UserRepo) UserService {
		return UserService{repo: userRepo}
	})

	for _, k := range c.Keys() {
		fmt.Printf("%-50v: type=%v, val=%v\n", k, reflect.ValueOf(k).Type(), c.MustGet(k))
	}

	c.MustResolve(func(userRepo *UserRepo) {
		fmt.Println(userRepo.connStr)
	})
	err := c.Resolve(func(userService *UserService) { fmt.Println(userService.GetUser()) })
	if err == nil || err.Error() != "args not instanced: not found in container: key=*container_test.UserService not found, may be you want container_test.UserService" {
		t.Errorf("test failed")
	}
	err = c.Resolve(func(userRepo UserRepo) { fmt.Println(userRepo.connStr) })
	if err == nil || err.Error() != "args not instanced: not found in container: key=container_test.UserRepo not found" {
		t.Errorf("test failed")
	}
}

// TestExtend 测试容器扩展
func TestExtend(t *testing.T) {
	c := ioc.New()
	c.MustBindValue("conn_str", "root:root@/my_db?charset=utf8")
	c.MustSingleton(func(c ioc.Container) (*UserRepo, error) {
		connStr, err := c.Get("conn_str")
		if err != nil {
			return nil, err
		}

		return &UserRepo{connStr: connStr.(string)}, nil
	})
	c.MustPrototype(func(userRepo *UserRepo) *UserService {
		return &UserService{repo: userRepo}
	})

	if err := c.Resolve(func(userRepo *UserRepo) {
		if userRepo.connStr == "" {
			t.Error("test failed")
		}
	}); err != nil {
		t.Error("test failed")
	}

	c2 := ioc.Extend(c)
	c2.MustBindValue("name", "c2")
	if err := c2.Resolve(func(userRepo *UserRepo) {
		if userRepo.connStr == "" {
			t.Error("test failed")
		}
	}); err != nil {
		t.Error("test failed")
	}

	val, err := c2.Get("name")
	if err != nil {
		t.Error("test failed")
	}

	if val.(string) != "c2" {
		t.Error("test failed")
	}

	if _, err := c.Get("name"); err == nil {
		t.Error("test failed")
	}
}

// --------------- 测试实例覆盖  ------------------

type InterfaceDemo interface {
	String() string
}

type demo1 struct{}

func (d demo1) String() string { return "demo1" }

type demo2 struct{}

func (d demo2) String() string { return "demo2" }

func TestContainerImpl_Override(t *testing.T) {
	c := ioc.New()

	c.MustSingletonOverride(func() InterfaceDemo {
		return demo1{}
	})

	c.MustSingleton(func() InterfaceDemo {
		return demo2{}
	})

	res := c.MustGet(new(InterfaceDemo))
	if "demo2" != res.(InterfaceDemo).String() {
		t.Error("test failed")
	}

	c.MustResolve(func(demo InterfaceDemo) {
		if "demo2" != demo.String() {
			t.Error("test failed")
		}
	})

}

// ----------- 测试自动注入 --------------

type UserManager struct {
	UserRepo *UserRepo `autowire:"@" json:"-"`
	field1   string    `autowire:"version"`
	Field2   string    `json:"field2"`
}

func TestContainerImpl_AutoWire(t *testing.T) {
	c := ioc.New()

	userRepoStruct := UserRepo{connStr: "user struct"}
	userRepoPointer := &UserRepo{connStr: "user pointer"}

	c.MustSingleton(userRepoStruct)
	c.MustSingleton(userRepoPointer)
	c.MustBindValue("version", "1.0.1")

	manager := UserManager{}
	if err := c.AutoWire(&manager); err != nil {
		t.Error("test failed")
	}

	if manager.UserRepo.connStr != "user pointer" {
		t.Error("test failed")
	}

	if manager.field1 != "1.0.1" {
		t.Error("test failed")
	}

	if manager.Field2 != "" {
		t.Error("test failed")
	}
}

// ------------- 测试 Keys -------------

func TestContainerImpl_Keys(t *testing.T) {
	c := ioc.New()
	c.MustSingleton(func() InterfaceDemo { return demo1{} })
	c.MustSingleton(demo2{})
	c.MustBindValue("key1", "value1")
	c.MustBindValue("key2", 1233)
	c.MustBindValue("container.Container", "与接口同名的value")

	for _, k := range c.Keys() {
		fmt.Printf("%-50v: type=%v, val=%v\n", k, reflect.ValueOf(k).Type(), c.MustGet(k))
	}

	if c.MustGet("container.Container") != "与接口同名的value" {
		t.Error("test failed")
	}
}

func TestConditional(t *testing.T) {
	{
		c := ioc.New()
		c.MustSingleton(ioc.WithCondition(func() InterfaceDemo {
			return demo1{}
		}, func() bool { return true }))

		c.MustSingleton(ioc.WithCondition(func() InterfaceDemo {
			return demo2{}
		}, func() bool { return false }))

		c.MustResolve(func(demo InterfaceDemo) {
			if demo.String() != "demo1" {
				t.Error("test failed")
			}
		})
	}

	{
		c := ioc.New()
		c.MustPrototype(ioc.WithCondition(func() InterfaceDemo {
			return demo2{}
		}, func() bool { return true }))

		c.MustPrototype(ioc.WithCondition(func() InterfaceDemo {
			return demo1{}
		}, func() bool { return false }))

		c.MustResolve(func(demo InterfaceDemo) {
			if demo.String() != "demo2" {
				t.Error("test failed")
			}
		})
	}

}

func TestResolveReflectValue(t *testing.T) {
	cc := ioc.New()
	cc.MustSingleton(func() InterfaceDemo { return demo1{} })

	callback := func(demo InterfaceDemo) {
		fmt.Println(demo.String())
	}

	cc.MustResolve(reflect.ValueOf(callback))
}