working commit

pkg/client/client_test.go

@@ -10,7 +10,6 @@ import (
 	"testing"
 	"time"
 
-	//"mstore/pkg/client"
 	"mstore/app/server"
 
 	"github.com/stretchr/testify/require"
@@ -67,12 +66,13 @@ func TestService(t *testing.T) {
 		require.NoError(t, err)
 		require.True(t, helloRes)
 	}
+	filesize := 32
 	{
 		// PutFile
 		tmpdir := t.TempDir()
 		tmpfile := filepath.Join(tmpdir, "foo.bin")
 
-		filedata := make([]byte, 32)
+		filedata := make([]byte, filesize)
 		_, err = rand.Read(filedata)
 		require.NoError(t, err)
 
@@ -86,11 +86,21 @@ func TestService(t *testing.T) {
 
 		err = cli.PutFile(ctx, tmpfile, srvaddr+"/foo.bin")
 		require.NoError(t, err)
+	}
+	{
+		// FileExists
+		fmt.Printf("=== FileExists ===\n")
+		cli := NewClient()
+		ctx := context.Background()
+		ctx, _ = context.WithTimeout(ctx, 1*time.Second)
 
+		exists, err := cli.FileExists(ctx, srvaddr+"/foo.bin")
+		require.NoError(t, err)
+		require.True(t, exists)
 	}
 	{
 		// GetFile
-		fmt.Printf("=== GetFil ===\n")
+		fmt.Printf("=== GetFile ===\n")
 		cli := NewClient()
 		ctx := context.Background()
 		ctx, _ = context.WithTimeout(ctx, 1*time.Second)
@@ -98,9 +108,33 @@ func TestService(t *testing.T) {
 		tmpdir := t.TempDir()
 		tmpfile := filepath.Join(tmpdir, "foo.bin")
 
-		_, err = cli.GetFile(ctx, srvaddr+"/foo.bin", tmpfile)
+		recsize, err := cli.GetFile(ctx, srvaddr+"/foo.bin", tmpfile)
 		require.NoError(t, err)
+		require.Equal(t, recsize, int64(filesize))
+	}
+	{
+		// DeleteFile
+		fmt.Printf("=== DeleteFile ===\n")
+		cli := NewClient()
+		ctx := context.Background()
+		ctx, _ = context.WithTimeout(ctx, 1*time.Second)
 
+		tmpdir := t.TempDir()
+		tmpfile := filepath.Join(tmpdir, "foo.bin")
+
+		err = cli.DeleteFile(ctx, srvaddr+"/foo.bin", tmpfile)
+		require.NoError(t, err)
+	}
+	{
+		// !FileExists
+		fmt.Printf("=== FileExists ===\n")
+		cli := NewClient()
+		ctx := context.Background()
+		ctx, _ = context.WithTimeout(ctx, 1*time.Second)
+
+		exists, err := cli.FileExists(ctx, srvaddr+"/foo.bin")
+		require.NoError(t, err)
+		require.False(t, exists)
 	}
 
 }