mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package agentapi_test
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/mock/gomock"
|
|
|
|
agentproto "github.com/coder/coder/v2/agent/proto"
|
|
"github.com/coder/coder/v2/coderd/agentapi"
|
|
"github.com/coder/coder/v2/coderd/database/dbmock"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
)
|
|
|
|
func TestGetServiceBanner(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("OK", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cfg := codersdk.ServiceBannerConfig{
|
|
Enabled: true,
|
|
Message: "hello world",
|
|
BackgroundColor: "#000000",
|
|
}
|
|
cfgJSON, err := json.Marshal(cfg)
|
|
require.NoError(t, err)
|
|
|
|
dbM := dbmock.NewMockStore(gomock.NewController(t))
|
|
dbM.EXPECT().GetServiceBanner(gomock.Any()).Return(string(cfgJSON), nil)
|
|
|
|
api := &agentapi.ServiceBannerAPI{
|
|
Database: dbM,
|
|
}
|
|
|
|
resp, err := api.GetServiceBanner(context.Background(), &agentproto.GetServiceBannerRequest{})
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, &agentproto.ServiceBanner{
|
|
Enabled: cfg.Enabled,
|
|
Message: cfg.Message,
|
|
BackgroundColor: cfg.BackgroundColor,
|
|
}, resp)
|
|
})
|
|
|
|
t.Run("None", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
dbM := dbmock.NewMockStore(gomock.NewController(t))
|
|
dbM.EXPECT().GetServiceBanner(gomock.Any()).Return("", sql.ErrNoRows)
|
|
|
|
api := &agentapi.ServiceBannerAPI{
|
|
Database: dbM,
|
|
}
|
|
|
|
resp, err := api.GetServiceBanner(context.Background(), &agentproto.GetServiceBannerRequest{})
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, &agentproto.ServiceBanner{
|
|
Enabled: false,
|
|
Message: "",
|
|
BackgroundColor: "",
|
|
}, resp)
|
|
})
|
|
|
|
t.Run("BadJSON", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
dbM := dbmock.NewMockStore(gomock.NewController(t))
|
|
dbM.EXPECT().GetServiceBanner(gomock.Any()).Return("hi", nil)
|
|
|
|
api := &agentapi.ServiceBannerAPI{
|
|
Database: dbM,
|
|
}
|
|
|
|
resp, err := api.GetServiceBanner(context.Background(), &agentproto.GetServiceBannerRequest{})
|
|
require.Error(t, err)
|
|
require.ErrorContains(t, err, "unmarshal json")
|
|
require.Nil(t, resp)
|
|
})
|
|
}
|