ec1f89217a
Full Mattermost server source with integrated Community Enterprise features. Includes vendor directory for offline/air-gapped builds. Structure: - enterprise-impl/: Enterprise feature implementations - enterprise-community/: Init files that register implementations - enterprise/: Bridge imports (community_imports.go) - vendor/: All dependencies for offline builds Build (online): go build ./cmd/mattermost Build (offline/air-gapped): go build -mod=vendor ./cmd/mattermost 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package storetest
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
func TestCommandWebhookStore(t *testing.T, rctx request.CTX, ss store.Store) {
|
|
t.Run("", func(t *testing.T) { testCommandWebhookStore(t, rctx, ss) })
|
|
}
|
|
|
|
func testCommandWebhookStore(t *testing.T, rctx request.CTX, ss store.Store) {
|
|
cws := ss.CommandWebhook()
|
|
|
|
h1 := &model.CommandWebhook{}
|
|
h1.CommandId = model.NewId()
|
|
h1.UserId = model.NewId()
|
|
h1.ChannelId = model.NewId()
|
|
h1, err := cws.Save(h1)
|
|
require.NoError(t, err)
|
|
|
|
var r1 *model.CommandWebhook
|
|
r1, nErr := cws.Get(h1.Id)
|
|
require.NoError(t, nErr)
|
|
assert.Equal(t, *r1, *h1, "invalid returned webhook")
|
|
|
|
_, nErr = cws.Get("123")
|
|
var nfErr *store.ErrNotFound
|
|
require.True(t, errors.As(nErr, &nfErr), "Should have set the status as not found for missing id")
|
|
|
|
h2 := &model.CommandWebhook{}
|
|
h2.CreateAt = model.GetMillis() - 2*model.CommandWebhookLifetime
|
|
h2.CommandId = model.NewId()
|
|
h2.UserId = model.NewId()
|
|
h2.ChannelId = model.NewId()
|
|
h2, err = cws.Save(h2)
|
|
require.NoError(t, err)
|
|
|
|
_, nErr = cws.Get(h2.Id)
|
|
require.Error(t, nErr, "Should have set the status as not found for expired webhook")
|
|
require.True(t, errors.As(nErr, &nfErr), "Should have set the status as not found for expired webhook")
|
|
|
|
cws.Cleanup()
|
|
|
|
_, nErr = cws.Get(h1.Id)
|
|
require.NoError(t, nErr, "Should have no error getting unexpired webhook")
|
|
|
|
_, nErr = cws.Get(h2.Id)
|
|
require.True(t, errors.As(nErr, &nfErr), "Should have set the status as not found for expired webhook")
|
|
|
|
nErr = cws.TryUse(h1.Id, 1)
|
|
require.NoError(t, nErr, "Should be able to use webhook once")
|
|
|
|
nErr = cws.TryUse(h1.Id, 1)
|
|
require.Error(t, nErr, "Should be able to use webhook once")
|
|
var invErr *store.ErrInvalidInput
|
|
require.True(t, errors.As(nErr, &invErr), "Should be able to use webhook once")
|
|
}
|