package client import ( "context" "fmt" "sync" "testing" "time" "mstore/app/server" "github.com/stretchr/testify/require" ) func TestImageLife(t *testing.T) { var srvport int64 = 10250 srvdir := t.TempDir() srvaddr := fmt.Sprintf("127.0.0.1:%d", srvport) srv, err := server.NewServer() require.NoError(t, err) { err = srv.Configure() require.NoError(t, err) if false { srv.SetDatadir(srvdir) srv.SetLogdir(srvdir) srv.SetRundir(srvdir) } srv.SetPort(srvport) err = srv.Build() require.NoError(t, err) var svcWG sync.WaitGroup errPipe := make(chan error, 5) startFunc := func() { err := srv.Service().Run() errPipe <- err svcWG.Done() } stopFunc := func() { time.Sleep(5 * time.Second) srv.Service().Stop() svcWG.Wait() err = <-errPipe require.NoError(t, err) } defer stopFunc() svcWG.Add(1) go startFunc() time.Sleep(1 * time.Second) } { // ServiceHello fmt.Printf("=== ServiceHello ===\n") cli := NewClient() ctx := context.Background() helloRes, err := cli.ServiceHello(ctx, srvaddr+"/hello", 1*time.Second) require.NoError(t, err) require.True(t, helloRes) } { // PishImage fmt.Printf("=== PushImage ===\n") cli := NewClient() ctx := context.Background() err := cli.PushImage(ctx, "test-oci.img", srvaddr+"/foo/test:123", 1*time.Second) require.NoError(t, err) } }