Mustafa Sencer Özcan
4c5d9cb95f
* fix: add type * feat: register step * feat: add playlist support * test: add test case * fix: gen mock * fix: go gen * fix: lint * fix: lint * fix: tests * fix: add resource * fix: readd * fix: address comments * fix: independent playlist query for migrations * fix: remove lock logic for sqlite * fix: handle creation and update datetimes * fix: query templating * fix: simply resources and address comments
120 lines
3.6 KiB
Go
120 lines
3.6 KiB
Go
package migrations_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
authlib "github.com/grafana/authlib/types"
|
|
"github.com/grafana/grafana/pkg/infra/tracing"
|
|
"github.com/grafana/grafana/pkg/services/playlist"
|
|
"github.com/grafana/grafana/pkg/services/playlist/playlistimpl"
|
|
"github.com/grafana/grafana/pkg/tests/apis"
|
|
"github.com/stretchr/testify/require"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
)
|
|
|
|
// playlistsTestCase tests the "playlists" ResourceMigration
|
|
type playlistsTestCase struct {
|
|
playlistUIDs []string
|
|
}
|
|
|
|
// newPlaylistsTestCase creates a test case for the playlists migrator
|
|
func newPlaylistsTestCase() resourceMigratorTestCase {
|
|
return &playlistsTestCase{
|
|
playlistUIDs: []string{},
|
|
}
|
|
}
|
|
|
|
func (tc *playlistsTestCase) name() string {
|
|
return "playlists"
|
|
}
|
|
|
|
func (tc *playlistsTestCase) resources() []schema.GroupVersionResource {
|
|
return []schema.GroupVersionResource{
|
|
{
|
|
Group: "playlist.grafana.app",
|
|
Version: "v0alpha1",
|
|
Resource: "playlists",
|
|
},
|
|
}
|
|
}
|
|
|
|
func (tc *playlistsTestCase) setup(t *testing.T, helper *apis.K8sTestHelper) {
|
|
t.Helper()
|
|
|
|
// Get playlist service from the test environment
|
|
// The service writes directly to SQL storage, which works in Mode0
|
|
env := helper.GetEnv()
|
|
playlistSvc := playlistimpl.ProvideService(env.SQLStore, tracing.InitializeTracerForTest())
|
|
|
|
// Use a non-existent dashboard UID for testing
|
|
// This avoids interfering with other test cases
|
|
nonExistentDashboardUID := "non-existent-dashboard-uid"
|
|
|
|
// Create playlist with dashboard UID items (pointing to non-existent dashboard)
|
|
playlist1UID := createTestPlaylist(t, playlistSvc, helper.Org1.OrgID, "Playlist with Dashboard UIDs", "5m", []playlist.PlaylistItem{
|
|
{Type: "dashboard_by_uid", Value: nonExistentDashboardUID, Order: 1},
|
|
})
|
|
tc.playlistUIDs = append(tc.playlistUIDs, playlist1UID)
|
|
|
|
// Create playlist with tag items
|
|
playlist2UID := createTestPlaylist(t, playlistSvc, helper.Org1.OrgID, "Playlist with Tags", "10m", []playlist.PlaylistItem{
|
|
{Type: "dashboard_by_tag", Value: "test-tag", Order: 1},
|
|
{Type: "dashboard_by_tag", Value: "another-tag", Order: 2},
|
|
})
|
|
tc.playlistUIDs = append(tc.playlistUIDs, playlist2UID)
|
|
|
|
// Create playlist with mixed items
|
|
playlist3UID := createTestPlaylist(t, playlistSvc, helper.Org1.OrgID, "Playlist with Mixed Items", "15m", []playlist.PlaylistItem{
|
|
{Type: "dashboard_by_uid", Value: nonExistentDashboardUID, Order: 1},
|
|
{Type: "dashboard_by_tag", Value: "mixed-tag", Order: 2},
|
|
})
|
|
tc.playlistUIDs = append(tc.playlistUIDs, playlist3UID)
|
|
}
|
|
|
|
func (tc *playlistsTestCase) verify(t *testing.T, helper *apis.K8sTestHelper, shouldExist bool) {
|
|
t.Helper()
|
|
|
|
expectedPlaylistCount := 0
|
|
if shouldExist {
|
|
expectedPlaylistCount = len(tc.playlistUIDs)
|
|
}
|
|
|
|
orgID := helper.Org1.OrgID
|
|
namespace := authlib.OrgNamespaceFormatter(orgID)
|
|
|
|
// Verify playlists
|
|
playlistCli := helper.GetResourceClient(apis.ResourceClientArgs{
|
|
User: helper.Org1.Admin,
|
|
Namespace: namespace,
|
|
GVR: schema.GroupVersionResource{
|
|
Group: "playlist.grafana.app",
|
|
Version: "v0alpha1",
|
|
Resource: "playlists",
|
|
},
|
|
})
|
|
|
|
verifyResourceCount(t, playlistCli, expectedPlaylistCount)
|
|
for _, uid := range tc.playlistUIDs {
|
|
verifyResource(t, playlistCli, uid, shouldExist)
|
|
}
|
|
}
|
|
|
|
func createTestPlaylist(t *testing.T, playlistSvc playlist.Service, orgID int64, name, interval string, items []playlist.PlaylistItem) string {
|
|
t.Helper()
|
|
|
|
cmd := &playlist.CreatePlaylistCommand{
|
|
Name: name,
|
|
Interval: interval,
|
|
Items: items,
|
|
OrgId: orgID,
|
|
}
|
|
|
|
result, err := playlistSvc.Create(context.Background(), cmd)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
require.NotEmpty(t, result.UID)
|
|
|
|
return result.UID
|
|
}
|