mirror of
https://github.com/coder/coder.git
synced 2025-07-08 11:39:50 +00:00
113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package cli_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/cli/clitest"
|
|
"github.com/coder/coder/v2/coderd/coderdtest"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
func createOpts(t *testing.T) *coderdtest.Options {
|
|
t.Helper()
|
|
|
|
dt := coderdtest.DeploymentValues(t)
|
|
dt.Experiments = []string{string(codersdk.ExperimentNotifications)}
|
|
return &coderdtest.Options{
|
|
DeploymentValues: dt,
|
|
}
|
|
}
|
|
|
|
func TestNotifications(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
|
name string
|
|
command string
|
|
expectPaused bool
|
|
}{
|
|
{
|
|
name: "PauseNotifications",
|
|
command: "pause",
|
|
expectPaused: true,
|
|
},
|
|
{
|
|
name: "ResumeNotifications",
|
|
command: "resume",
|
|
expectPaused: false,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tt := tt
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// given
|
|
ownerClient, db := coderdtest.NewWithDatabase(t, createOpts(t))
|
|
_ = coderdtest.CreateFirstUser(t, ownerClient)
|
|
|
|
// when
|
|
inv, root := clitest.New(t, "notifications", tt.command)
|
|
clitest.SetupConfig(t, ownerClient, root)
|
|
|
|
var buf bytes.Buffer
|
|
inv.Stdout = &buf
|
|
err := inv.Run()
|
|
require.NoError(t, err)
|
|
|
|
// then
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
|
|
t.Cleanup(cancel)
|
|
settingsJSON, err := db.GetNotificationsSettings(ctx)
|
|
require.NoError(t, err)
|
|
|
|
var settings codersdk.NotificationsSettings
|
|
err = json.Unmarshal([]byte(settingsJSON), &settings)
|
|
require.NoError(t, err)
|
|
require.Equal(t, tt.expectPaused, settings.NotifierPaused)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPauseNotifications_RegularUser(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// given
|
|
ownerClient, db := coderdtest.NewWithDatabase(t, createOpts(t))
|
|
owner := coderdtest.CreateFirstUser(t, ownerClient)
|
|
anotherClient, _ := coderdtest.CreateAnotherUser(t, ownerClient, owner.OrganizationID)
|
|
|
|
// when
|
|
inv, root := clitest.New(t, "notifications", "pause")
|
|
clitest.SetupConfig(t, anotherClient, root)
|
|
|
|
var buf bytes.Buffer
|
|
inv.Stdout = &buf
|
|
err := inv.Run()
|
|
var sdkError *codersdk.Error
|
|
require.Error(t, err)
|
|
require.ErrorAsf(t, err, &sdkError, "error should be of type *codersdk.Error")
|
|
assert.Equal(t, http.StatusForbidden, sdkError.StatusCode())
|
|
assert.Contains(t, sdkError.Message, "Forbidden.")
|
|
|
|
// then
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
|
|
t.Cleanup(cancel)
|
|
settingsJSON, err := db.GetNotificationsSettings(ctx)
|
|
require.NoError(t, err)
|
|
|
|
var settings codersdk.NotificationsSettings
|
|
err = json.Unmarshal([]byte(settingsJSON), &settings)
|
|
require.NoError(t, err)
|
|
require.False(t, settings.NotifierPaused) // still running
|
|
}
|