mstore/test/file_test.go

package test

import (
	"github.com/stretchr/testify/require"

	"bytes"
	"encoding/hex"
	"fmt"
	"io"
	"math/rand"
	"net/http"
	"net/http/httptest"
	"testing"

	"mstore/app/router"
	"mstore/app/server"
)

func TestFileOperations(t *testing.T) {
	var err error
	fmt.Printf("=== MakeServer ===\n")
	srv, err := server.NewServer()
	require.NoError(t, err)

	{
		err = srv.Configure()
		require.NoError(t, err)

		tmpdir := t.TempDir()
		srv.SetDatadir(tmpdir)
		srv.SetLogdir(tmpdir)
		srv.SetRundir(tmpdir)

		err = srv.Build()
		require.NoError(t, err)
	}

	{
		fmt.Printf("=== PutFile ===\n")
		reqPath := `/v3/api/file/foo/bare`
		routePath := `/v3/api/file/{filepath}`

		rout := router.NewRouter()
		hand := srv.Handler()
		require.NotNil(t, hand)

		rout.Put(routePath, hand.PutFile)

		datasize := 16
		filedata := make([]byte, datasize)
		_, err = rand.Read(filedata)
		require.NoError(t, err)

		filedata = []byte(hex.EncodeToString(filedata))

		source := bytes.NewReader(filedata)

		request, err := http.NewRequest("PUT", reqPath, source)
		require.NoError(t, err)

		request.Header.Set("Content-Length", fmt.Sprintf("%d", datasize))
		request.Header.Set("Content-Type", "application/octet-stream")

		recorder := httptest.NewRecorder()
		rout.ServeHTTP(recorder, request)
		require.Equal(t, http.StatusOK, recorder.Code)

		fmt.Printf("Response code: %d\n", recorder.Code)

		bodyReader := recorder.Body
		bodyBytes, err := io.ReadAll(bodyReader)

		fmt.Printf("Response body: %s\n", string(bodyBytes))
	}
	{
		fmt.Printf("=== FileExists ===\n")

		reqPath := `/v3/api/file/foo/bare`
		routePath := `/v3/api/file/{filepath}`

		rout := router.NewRouter()
		hand := srv.Handler()
		require.NotNil(t, hand)

		rout.Head(routePath, hand.FileExists)

		request, err := http.NewRequest("HEAD", reqPath, nil)
		require.NoError(t, err)

		recorder := httptest.NewRecorder()
		rout.ServeHTTP(recorder, request)
		require.Equal(t, http.StatusOK, recorder.Code)

		fmt.Printf("Response code: %d\n", recorder.Code)

		bodyReader := recorder.Body
		bodyBytes, err := io.ReadAll(bodyReader)

		fmt.Printf("Response body: %s\n", string(bodyBytes))
	}
	{
		fmt.Printf("=== GetFile ===\n")
		reqPath := `/v3/api/file/foo/bare`
		routePath := `/v3/api/file/{filepath}`

		rout := router.NewRouter()
		hand := srv.Handler()
		require.NotNil(t, hand)

		rout.Get(routePath, hand.GetFile)

		request, err := http.NewRequest("GET", reqPath, nil)
		require.NoError(t, err)

		recorder := httptest.NewRecorder()
		rout.ServeHTTP(recorder, request)
		require.Equal(t, http.StatusOK, recorder.Code)

		fmt.Printf("Response code: %d\n", recorder.Code)

		bodyReader := recorder.Body
		bodyBytes, err := io.ReadAll(bodyReader)

		fmt.Printf("Response body: %s\n", string(bodyBytes))
	}
	{
		fmt.Printf("=== DeleteFile ===\n")
		reqPath := `/v3/api/file/foo/bare`
		routePath := `/v3/api/file/{filepath}`

		rout := router.NewRouter()
		hand := srv.Handler()
		require.NotNil(t, hand)

		rout.Delete(routePath, hand.DeleteFile)

		request, err := http.NewRequest("DELETE", reqPath, nil)
		require.NoError(t, err)

		recorder := httptest.NewRecorder()
		rout.ServeHTTP(recorder, request)
		require.Equal(t, http.StatusOK, recorder.Code)

		fmt.Printf("Response code: %d\n", recorder.Code)

		bodyReader := recorder.Body
		bodyBytes, err := io.ReadAll(bodyReader)

		fmt.Printf("Response body: %s\n", string(bodyBytes))
	}

	{
		fmt.Printf("=== FileNotExists ===\n")

		reqPath := `/v3/api/file/foo/bare`
		routePath := `/v3/api/file/{filepath}`

		rout := router.NewRouter()
		hand := srv.Handler()
		require.NotNil(t, hand)

		rout.Head(routePath, hand.FileExists)

		request, err := http.NewRequest("HEAD", reqPath, nil)
		require.NoError(t, err)

		recorder := httptest.NewRecorder()
		rout.ServeHTTP(recorder, request)
		require.Equal(t, http.StatusNotFound, recorder.Code)

		fmt.Printf("Response code: %d\n", recorder.Code)

		bodyReader := recorder.Body
		bodyBytes, err := io.ReadAll(bodyReader)

		fmt.Printf("Response body: %s\n", string(bodyBytes))
	}
	{
		fmt.Printf("=== ServiceHello ===\n")
		reqPath := "/service/hello"
		routePath := "/service/hello"

		rout := router.NewRouter()
		hand := srv.Handler()
		rout.Get(routePath, hand.SendHello)

		request, err := http.NewRequest("GET", reqPath, nil)
		require.NoError(t, err)

		recorder := httptest.NewRecorder()
		rout.ServeHTTP(recorder, request)
		require.Equal(t, http.StatusOK, recorder.Code)

		fmt.Printf("Response code: %d\n", recorder.Code)

		bodyReader := recorder.Body
		bodyBytes, err := io.ReadAll(bodyReader)

		fmt.Printf("Response body: %s\n", string(bodyBytes))
	}
}