dps-sdk/test/v1/sdk.go

package v1

import (
	"fmt"
	"git.sxidc.com/service-supports/dps-sdk"
	"git.sxidc.com/service-supports/dps-sdk/ports"
	"testing"
)

var clientInstance ports.Client

func initClient(t *testing.T, address string) {
	client, err := dps.NewClient(address, "v1")
	if err != nil {
		t.Fatal(err)
	}

	clientInstance = client
}

func destroyClient(t *testing.T) {
	err := dps.DestroyClient("v1")
	if err != nil {
		t.Fatal(err)
	}

	clientInstance = nil
}

type ToolKit struct {
	t *testing.T
}

func newToolKit(t *testing.T) *ToolKit {
	return &ToolKit{t: t}
}

func (toolKit *ToolKit) autoMigrate(req *ports.AutoMigrateRequest) *ToolKit {
	err := clientInstance.AutoMigrate(req)
	if err != nil {
		toolKit.t.Fatal(err)
	}

	return toolKit
}

func (toolKit *ToolKit) insert(req *ports.InsertRequest) *ToolKit {
	statement, err := clientInstance.Insert(req)
	if err != nil {
		toolKit.t.Fatal(err)
	}

	fmt.Println(statement)

	return toolKit
}

func (toolKit *ToolKit) delete(req *ports.DeleteRequest) *ToolKit {
	statement, err := clientInstance.Delete(req)
	if err != nil {
		toolKit.t.Fatal(err)
	}

	fmt.Println(statement)

	return toolKit
}

func (toolKit *ToolKit) update(req *ports.UpdateRequest) *ToolKit {
	statement, err := clientInstance.Update(req)
	if err != nil {
		toolKit.t.Fatal(err)
	}

	fmt.Println(statement)

	return toolKit
}