sourcegraph/internal/database/code_monitor_webhook_test.go

package database

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/sourcegraph/log/logtest"

	"github.com/sourcegraph/sourcegraph/internal/actor"
	"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"

	logger := logtest.Scoped(t)

	t.Run("CreateThenGet", func(t *testing.T) {
		t.Parallel()

		db := NewDB(logger, dbtest.NewDB(t))
		_, _, ctx := newTestUser(ctx, t, db)
		s := CodeMonitorsWith(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 := NewDB(logger, dbtest.NewDB(t))
		_, _, ctx := newTestUser(ctx, t, db)
		s := CodeMonitorsWith(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 := NewDB(logger, dbtest.NewDB(t))
		_, _, ctx := newTestUser(ctx, t, db)
		s := CodeMonitorsWith(db)

		_, err := s.UpdateWebhookAction(ctx, 383838, false, false, url2)
		require.Error(t, err)
	})

	t.Run("CreateDeleteGet", func(t *testing.T) {
		t.Parallel()

		db := NewDB(logger, dbtest.NewDB(t))
		_, _, ctx := newTestUser(ctx, t, db)
		s := CodeMonitorsWith(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 := NewDB(logger, dbtest.NewDB(t))
		_, _, ctx := newTestUser(ctx, t, db)
		s := CodeMonitorsWith(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 := NewDB(logger, dbtest.NewDB(t))
		_, _, ctx := newTestUser(ctx, t, db)
		s := CodeMonitorsWith(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)
	})

	t.Run("Update permissions", func(t *testing.T) {
		ctx, db, s := newTestStore(t)
		uid1 := insertTestUser(ctx, t, db, "u1", false)
		ctx1 := actor.WithActor(ctx, actor.FromUser(uid1))
		uid2 := insertTestUser(ctx, t, db, "u2", false)
		ctx2 := actor.WithActor(ctx, actor.FromUser(uid2))
		uid3 := insertTestUser(ctx, t, db, "u3", true)
		ctx3 := actor.WithActor(ctx, actor.FromUser(uid3))
		fixtures := s.insertTestMonitor(ctx1, t)
		_ = s.insertTestMonitor(ctx2, t)

		wa, err := s.CreateWebhookAction(ctx1, fixtures.monitor.ID, true, true, "https://true.com")
		require.NoError(t, err)

		// User1 can update it
		_, err = s.UpdateWebhookAction(ctx1, wa.ID, true, true, "https://false.com")
		require.NoError(t, err)

		// User2 cannot update it
		_, err = s.UpdateWebhookAction(ctx2, wa.ID, true, true, "https://truer.com")
		require.Error(t, err)

		// User3 can update it
		_, err = s.UpdateWebhookAction(ctx3, wa.ID, true, true, "https://false.com")
		require.NoError(t, err)

		wa, err = s.GetWebhookAction(ctx1, wa.ID)
		require.NoError(t, err)
		require.Equal(t, wa.URL, "https://false.com")
	})
}