mirror of
https://github.com/AvengeMedia/DankMaterialShell.git
synced 2025-12-13 00:42:49 -05:00
rename backend to core
This commit is contained in:
196
core/internal/server/plugins/handlers_test.go
Normal file
196
core/internal/server/plugins/handlers_test.go
Normal file
@@ -0,0 +1,196 @@
|
||||
package plugins
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"testing"
|
||||
|
||||
"github.com/AvengeMedia/DankMaterialShell/core/internal/mocks/net"
|
||||
"github.com/AvengeMedia/DankMaterialShell/core/internal/server/models"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/mock"
|
||||
)
|
||||
|
||||
func TestHandleList(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
conn.EXPECT().Write(mock.Anything).Return(0, nil).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.list",
|
||||
Params: map[string]interface{}{},
|
||||
}
|
||||
|
||||
HandleList(conn, req)
|
||||
}
|
||||
|
||||
func TestHandleListInstalled(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
conn.EXPECT().Write(mock.Anything).Return(0, nil).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.listInstalled",
|
||||
Params: map[string]interface{}{},
|
||||
}
|
||||
|
||||
HandleListInstalled(conn, req)
|
||||
}
|
||||
|
||||
func TestHandleInstallMissingName(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
var written []byte
|
||||
conn.EXPECT().Write(mock.Anything).RunAndReturn(func(b []byte) (int, error) {
|
||||
written = b
|
||||
return len(b), nil
|
||||
}).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.install",
|
||||
Params: map[string]interface{}{},
|
||||
}
|
||||
|
||||
HandleInstall(conn, req)
|
||||
|
||||
var resp models.Response[SuccessResult]
|
||||
err := json.Unmarshal(written, &resp)
|
||||
assert.NoError(t, err)
|
||||
assert.NotEmpty(t, resp.Error)
|
||||
assert.Contains(t, resp.Error, "missing or invalid 'name' parameter")
|
||||
}
|
||||
|
||||
func TestHandleInstallInvalidName(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
var written []byte
|
||||
conn.EXPECT().Write(mock.Anything).RunAndReturn(func(b []byte) (int, error) {
|
||||
written = b
|
||||
return len(b), nil
|
||||
}).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.install",
|
||||
Params: map[string]interface{}{
|
||||
"name": 123,
|
||||
},
|
||||
}
|
||||
|
||||
HandleInstall(conn, req)
|
||||
|
||||
var resp models.Response[SuccessResult]
|
||||
err := json.Unmarshal(written, &resp)
|
||||
assert.NoError(t, err)
|
||||
assert.NotEmpty(t, resp.Error)
|
||||
}
|
||||
|
||||
func TestHandleUninstallMissingName(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
var written []byte
|
||||
conn.EXPECT().Write(mock.Anything).RunAndReturn(func(b []byte) (int, error) {
|
||||
written = b
|
||||
return len(b), nil
|
||||
}).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.uninstall",
|
||||
Params: map[string]interface{}{},
|
||||
}
|
||||
|
||||
HandleUninstall(conn, req)
|
||||
|
||||
var resp models.Response[SuccessResult]
|
||||
err := json.Unmarshal(written, &resp)
|
||||
assert.NoError(t, err)
|
||||
assert.NotEmpty(t, resp.Error)
|
||||
}
|
||||
|
||||
func TestHandleUpdateMissingName(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
var written []byte
|
||||
conn.EXPECT().Write(mock.Anything).RunAndReturn(func(b []byte) (int, error) {
|
||||
written = b
|
||||
return len(b), nil
|
||||
}).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.update",
|
||||
Params: map[string]interface{}{},
|
||||
}
|
||||
|
||||
HandleUpdate(conn, req)
|
||||
|
||||
var resp models.Response[SuccessResult]
|
||||
err := json.Unmarshal(written, &resp)
|
||||
assert.NoError(t, err)
|
||||
assert.NotEmpty(t, resp.Error)
|
||||
}
|
||||
|
||||
func TestHandleSearchMissingQuery(t *testing.T) {
|
||||
conn := net.NewMockConn(t)
|
||||
var written []byte
|
||||
conn.EXPECT().Write(mock.Anything).RunAndReturn(func(b []byte) (int, error) {
|
||||
written = b
|
||||
return len(b), nil
|
||||
}).Maybe()
|
||||
|
||||
req := models.Request{
|
||||
ID: 123,
|
||||
Method: "plugins.search",
|
||||
Params: map[string]interface{}{},
|
||||
}
|
||||
|
||||
HandleSearch(conn, req)
|
||||
|
||||
var resp models.Response[[]PluginInfo]
|
||||
err := json.Unmarshal(written, &resp)
|
||||
assert.NoError(t, err)
|
||||
assert.NotEmpty(t, resp.Error)
|
||||
}
|
||||
|
||||
func TestSortPluginInfoByFirstParty(t *testing.T) {
|
||||
plugins := []PluginInfo{
|
||||
{Name: "third-party", Repo: "https://github.com/other/test"},
|
||||
{Name: "first-party", Repo: "https://github.com/AvengeMedia/test"},
|
||||
}
|
||||
|
||||
SortPluginInfoByFirstParty(plugins)
|
||||
|
||||
assert.Equal(t, "first-party", plugins[0].Name)
|
||||
assert.Equal(t, "third-party", plugins[1].Name)
|
||||
}
|
||||
|
||||
func TestPluginInfoJSON(t *testing.T) {
|
||||
info := PluginInfo{
|
||||
Name: "test",
|
||||
Description: "test description",
|
||||
Installed: true,
|
||||
FirstParty: true,
|
||||
}
|
||||
|
||||
data, err := json.Marshal(info)
|
||||
assert.NoError(t, err)
|
||||
|
||||
var unmarshaled PluginInfo
|
||||
err = json.Unmarshal(data, &unmarshaled)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, info.Name, unmarshaled.Name)
|
||||
assert.Equal(t, info.Installed, unmarshaled.Installed)
|
||||
}
|
||||
|
||||
func TestSuccessResult(t *testing.T) {
|
||||
result := SuccessResult{
|
||||
Success: true,
|
||||
Message: "test message",
|
||||
}
|
||||
|
||||
data, err := json.Marshal(result)
|
||||
assert.NoError(t, err)
|
||||
|
||||
var unmarshaled SuccessResult
|
||||
err = json.Unmarshal(data, &unmarshaled)
|
||||
assert.NoError(t, err)
|
||||
assert.True(t, unmarshaled.Success)
|
||||
assert.Equal(t, "test message", unmarshaled.Message)
|
||||
}
|
||||
Reference in New Issue
Block a user