mirror of
https://github.com/sourcegraph/sourcegraph.git
synced 2026-02-06 17:51:57 +00:00
8d3e231322
Team work makes the linter dream work. Co-authored-by: Varun Gandhi <varun.gandhi@sourcegraph.com>
139 lines
3.8 KiB
Go
139 lines
3.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/sourcegraph/sourcegraph/internal/database"
|
|
"github.com/sourcegraph/sourcegraph/internal/database/dbtest"
|
|
)
|
|
|
|
func TestCodeMonitorStoreWebhooks(t *testing.T) {
|
|
ctx := context.Background()
|
|
url1 := "https://icanhazcheezburger.com/webhook"
|
|
url2 := "https://icanthazcheezburger.com/webhook"
|
|
|
|
t.Run("CreateThenGet", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db := database.NewDB(dbtest.NewDB(t))
|
|
_, _, ctx := newTestUser(ctx, t, db)
|
|
s := CodeMonitors(db)
|
|
fixtures := s.insertTestMonitor(ctx, t)
|
|
|
|
action, err := s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url1)
|
|
require.NoError(t, err)
|
|
|
|
got, err := s.GetWebhookAction(ctx, action.ID)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, action, got)
|
|
})
|
|
|
|
t.Run("CreateUpdateGet", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db := database.NewDB(dbtest.NewDB(t))
|
|
_, _, ctx := newTestUser(ctx, t, db)
|
|
s := CodeMonitors(db)
|
|
fixtures := s.insertTestMonitor(ctx, t)
|
|
|
|
action, err := s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url1)
|
|
require.NoError(t, err)
|
|
|
|
updated, err := s.UpdateWebhookAction(ctx, action.ID, false, false, url2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, false, updated.Enabled)
|
|
require.Equal(t, url2, updated.URL)
|
|
|
|
got, err := s.GetWebhookAction(ctx, action.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, updated, got)
|
|
})
|
|
|
|
t.Run("ErrorOnUpdateNonexistent", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db := database.NewDB(dbtest.NewDB(t))
|
|
_, _, ctx := newTestUser(ctx, t, db)
|
|
s := CodeMonitors(db)
|
|
|
|
_, err := s.UpdateWebhookAction(ctx, 383838, false, false, url2)
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("CreateDeleteGet", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db := database.NewDB(dbtest.NewDB(t))
|
|
_, _, ctx := newTestUser(ctx, t, db)
|
|
s := CodeMonitors(db)
|
|
fixtures := s.insertTestMonitor(ctx, t)
|
|
|
|
action1, err := s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url1)
|
|
require.NoError(t, err)
|
|
|
|
action2, err := s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url1)
|
|
require.NoError(t, err)
|
|
|
|
err = s.DeleteWebhookActions(ctx, fixtures.monitor.ID, action1.ID)
|
|
require.NoError(t, err)
|
|
|
|
_, err = s.GetWebhookAction(ctx, action1.ID)
|
|
require.Error(t, err)
|
|
|
|
_, err = s.GetWebhookAction(ctx, action2.ID)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("CountCreateCount", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db := database.NewDB(dbtest.NewDB(t))
|
|
_, _, ctx := newTestUser(ctx, t, db)
|
|
s := CodeMonitors(db)
|
|
fixtures := s.insertTestMonitor(ctx, t)
|
|
|
|
count, err := s.CountWebhookActions(ctx, fixtures.monitor.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, count)
|
|
|
|
_, err = s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url1)
|
|
require.NoError(t, err)
|
|
|
|
count, err = s.CountWebhookActions(ctx, fixtures.monitor.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, count)
|
|
})
|
|
|
|
t.Run("ListCreateList", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
db := database.NewDB(dbtest.NewDB(t))
|
|
_, _, ctx := newTestUser(ctx, t, db)
|
|
s := CodeMonitors(db)
|
|
fixtures := s.insertTestMonitor(ctx, t)
|
|
|
|
actions, err := s.ListWebhookActions(ctx, ListActionsOpts{MonitorID: &fixtures.monitor.ID})
|
|
require.NoError(t, err)
|
|
require.Len(t, actions, 0)
|
|
|
|
_, err = s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url1)
|
|
require.NoError(t, err)
|
|
|
|
_, err = s.CreateWebhookAction(ctx, fixtures.monitor.ID, true, false, url2)
|
|
require.NoError(t, err)
|
|
|
|
actions2, err := s.ListWebhookActions(ctx, ListActionsOpts{MonitorID: &fixtures.monitor.ID})
|
|
require.NoError(t, err)
|
|
require.Len(t, actions2, 2)
|
|
|
|
first := 1
|
|
actions3, err := s.ListWebhookActions(ctx, ListActionsOpts{MonitorID: &fixtures.monitor.ID, First: &first})
|
|
require.NoError(t, err)
|
|
require.Len(t, actions3, 1)
|
|
})
|
|
}
|