add more unit tests

Author: LinuxSuRen

pkg/server/remote_server.go

@@ -122,6 +122,7 @@ func NewRemoteServer(loader testing.Writer, storeWriterFactory testing.StoreWrit
 	if secretServer == nil {
 		secretServer = &fakeSecretServer{}
 	}
+	loader.WithUserConfigDir(configDir)
 	GrpcMaxRecvMsgSize = grpcMaxRecvMsgSize
 	return &server{
 		loader:             loader,
@@ -193,7 +194,7 @@ func resetEnv(oldEnv map[string]string) {
 	}
 }
 
-func (s *server) getLoaders(ctx context.Context) (loader []testing.Writer, err error) {
+func (s *server) getLoaders() (loader []testing.Writer, err error) {
 	var stores []testing.Store
 	if stores, err = testing.NewStoreFactory(s.configDir).GetStores(); err != nil {
 		return
@@ -1449,7 +1450,7 @@ func (s *server) GetMenus(ctx context.Context, _ *Empty) (result *MenuList, err
 
 	result = &MenuList{}
 	var loaders []testing.Writer
-	if loaders, err = s.getLoaders(ctx); err != nil {
+	if loaders, err = s.getLoaders(); err != nil {
 		return
 	}
 

pkg/server/remote_server_test.go

@@ -1,5 +1,5 @@
 /*
-Copyright 2024 API Testing Authors.
+Copyright 2024-2025 API Testing Authors.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -21,6 +21,7 @@ import (
 	"io"
 	"net/http"
 	"os"
+	"path/filepath"
 	"reflect"
 	"sync"
 	"sync/atomic"
@@ -950,10 +951,21 @@ func TestRemoteRunnerAdapter(t *testing.T) {
 	assert.Error(t, err)
 }
 
+//go:embed testdata/extension.yaml
+var extensionConfig []byte
+
 func getRemoteServerInTempDir() (server RunnerServer, call func()) {
 	dir, _ := os.MkdirTemp(os.TempDir(), "remote-server-test")
 	call = func() { os.RemoveAll(dir) }
 
+	corePath := filepath.Join(dir, "data", "core")
+	os.MkdirAll(corePath, 0755)
+	os.WriteFile(filepath.Join(corePath, "extension.yaml"), extensionConfig, 0755)
+
+	themePath := filepath.Join(dir, "data", "theme")
+	os.MkdirAll(themePath, 0755)
+	os.WriteFile(filepath.Join(themePath, "simple.json"), []byte(simplePostman), 0755)
+
 	writer := atest.NewFileWriter(dir)
 	server = NewRemoteServer(writer, newLocalloaderFromStore(), nil, nil, dir, 1024*1024*4)
 	return
@@ -1039,3 +1051,32 @@ func (s *fakeServerStream) RecvMsg(m interface{}) error {
 
 	return nil
 }
+
+func TestGetStoreKinds(t *testing.T) {
+	server, clean := getRemoteServerInTempDir()
+	defer clean()
+
+	reply, err := server.GetStoreKinds(context.Background(), &Empty{})
+	assert.NoError(t, err)
+
+	assert.Equal(t, 1, len(reply.Data))
+}
+
+func TestGetThemes(t *testing.T) {
+	server, clean := getRemoteServerInTempDir()
+	defer clean()
+
+	themeServer, ok := server.(ThemeExtensionServer)
+	assert.True(t, ok)
+
+	reply, err := themeServer.GetThemes(context.Background(), &Empty{})
+	assert.NoError(t, err)
+	assert.Equal(t, 1, len(reply.Data))
+
+	var theme *CommonResult
+	theme, err = themeServer.GetTheme(context.Background(), &SimpleName{
+		Name: "simple",
+	})
+	assert.NoError(t, err)
+	assert.NotNil(t, theme)
+}

pkg/server/testdata/extension.yaml

@@ -0,0 +1,8 @@
+items:
+  - name: atest-store-database
+    dependencies:
+      - name: atest-store-database
+    link: https://github.com/LinuxSuRen/atest-ext-store-database
+    params:
+      - key: database
+        defaultValue: 'atest'

pkg/testing/loader.go

@@ -23,6 +23,7 @@ type Loader interface {
 	GetContext() string
 	GetCount() int
 	Reset()
+	WithUserConfigDir(string)
 
 	Verify() (readOnly bool, version string, err error)
 	PProf(name string) []byte

pkg/testing/loader_file.go

@@ -41,20 +41,34 @@ var (
 )
 
 type fileLoader struct {
-	paths  []string
-	index  int
-	parent string
+	paths         []string
+	index         int
+	parent        string
+	userConfigDir string
 
 	lock *sync.RWMutex
 }
 
 // NewFileLoader creates the instance of file loader
 func NewFileLoader() Loader {
-	return &fileLoader{index: -1, lock: &sync.RWMutex{}}
+	return &fileLoader{
+		index:         -1,
+		lock:          &sync.RWMutex{},
+		userConfigDir: home.GetUserConfigDir(),
+	}
 }
 
 func NewFileWriter(parent string) Writer {
-	return &fileLoader{index: -1, parent: parent, lock: &sync.RWMutex{}}
+	return &fileLoader{
+		index:         -1,
+		parent:        parent,
+		lock:          &sync.RWMutex{},
+		userConfigDir: home.GetUserConfigDir(),
+	}
+}
+
+func (l *fileLoader) WithUserConfigDir(userConfigDir string) {
+	l.userConfigDir = userConfigDir
 }
 
 // HasMore returns if there are more test cases
@@ -529,7 +543,7 @@ func (l *fileLoader) GetRoundTripper(name string) (roundTripper http.RoundTrippe
 }
 
 func (l *fileLoader) GetThemes() (result []string, err error) {
-	dataDir := home.GetThemeDir()
+	dataDir := filepath.Join(l.userConfigDir, "data/theme")
 	_ = filepath.WalkDir(dataDir, func(path string, d os.DirEntry, err error) error {
 		if err != nil {
 			return err
@@ -544,7 +558,7 @@ func (l *fileLoader) GetThemes() (result []string, err error) {
 }
 
 func (l *fileLoader) GetTheme(name string) (result string, err error) {
-	dataDir := home.GetThemeDir()
+	dataDir := filepath.Join(l.userConfigDir, "data/theme")
 	themeFile := filepath.Join(dataDir, name+".json")
 	var data []byte
 	if data, err = os.ReadFile(themeFile); err == nil {

pkg/testing/loader_non.go

@@ -27,6 +27,10 @@ func NewNonWriter() Writer {
 	return &nonLoader{}
 }
 
+func (l *nonLoader) WithUserConfigDir(userConfigDir string) {
+	// non-implement
+}
+
 // HasMore returns if there are more test cases
 func (l *nonLoader) HasMore() bool {
 	return false

pkg/util/home/common.go

@@ -37,10 +37,6 @@ func GetUserDataDir() string {
 	return filepath.Join(GetUserConfigDir(), "data")
 }
 
-func GetThemeDir() string {
-	return filepath.Join(GetUserConfigDir(), "data/theme")
-}
-
 func GetBindingDir() string {
 	return filepath.Join(GetUserConfigDir(), "data/key-binding")
 }