chore: migrate part of shortcut store

This commit is contained in:
Steven
2023-08-02 21:57:32 +08:00
parent 98d4bb40b2
commit b365355610
6 changed files with 83 additions and 244 deletions

View File

@ -14,7 +14,7 @@ func TestShortcutStore(t *testing.T) {
ts := NewTestingStore(ctx, t)
user, err := createTestingAdminUser(ctx, ts)
require.NoError(t, err)
shortcut, err := ts.CreateShortcutV1(ctx, &storepb.Shortcut{
shortcut, err := ts.CreateShortcut(ctx, &storepb.Shortcut{
CreatorId: user.ID,
Name: "test",
Link: "https://test.link",
@ -24,7 +24,7 @@ func TestShortcutStore(t *testing.T) {
OgMetadata: &storepb.OpenGraphMetadata{},
})
require.NoError(t, err)
shortcuts, err := ts.ListShortcutsV1(ctx, &store.FindShortcut{
shortcuts, err := ts.ListShortcuts(ctx, &store.FindShortcut{
CreatorID: &user.ID,
})
require.NoError(t, err)
@ -38,7 +38,7 @@ func TestShortcutStore(t *testing.T) {
require.NoError(t, err)
require.Equal(t, newLink, updatedShortcut.Link)
tag := "test"
shortcut, err = ts.GetShortcutV1(ctx, &store.FindShortcut{
shortcut, err = ts.GetShortcut(ctx, &store.FindShortcut{
Tag: &tag,
})
require.NoError(t, err)
@ -46,7 +46,7 @@ func TestShortcutStore(t *testing.T) {
ID: shortcut.Id,
})
require.NoError(t, err)
shortcuts, err = ts.ListShortcutsV1(ctx, &store.FindShortcut{
shortcuts, err = ts.ListShortcuts(ctx, &store.FindShortcut{
CreatorID: &user.ID,
})
require.NoError(t, err)

View File

@ -4,8 +4,8 @@ import (
"context"
"testing"
storepb "github.com/boojack/slash/proto/gen/store"
"github.com/boojack/slash/store"
"github.com/stretchr/testify/require"
"golang.org/x/crypto/bcrypt"
)
@ -26,11 +26,11 @@ func TestUserStore(t *testing.T) {
Nickname: &userPatchNickname,
})
require.NoError(t, err)
_, err = ts.CreateShortcut(ctx, &store.Shortcut{
CreatorID: user.ID,
_, err = ts.CreateShortcut(ctx, &storepb.Shortcut{
CreatorId: user.ID,
Name: "test_shortcut",
Link: "https://www.google.com",
Visibility: store.VisibilityPublic,
Visibility: storepb.Visibility_PUBLIC,
})
require.NoError(t, err)
require.Equal(t, userPatchNickname, user.Nickname)
@ -41,7 +41,7 @@ func TestUserStore(t *testing.T) {
users, err = ts.ListUsers(ctx, &store.FindUser{})
require.NoError(t, err)
require.Equal(t, 0, len(users))
shortcuts, err := ts.ListShortcutsV1(ctx, &store.FindShortcut{})
shortcuts, err := ts.ListShortcuts(ctx, &store.FindShortcut{})
require.NoError(t, err)
require.Equal(t, 0, len(shortcuts))
}