147 lines
3.4 KiB
Go
147 lines
3.4 KiB
Go
// SPDX-FileCopyrightText: 2021 The NGI Pointer Secure-Scuttlebutt Team of 2020/2021
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package sqlite
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"crypto/rand"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
refs "github.com/ssbc/go-ssb-refs"
|
|
"github.com/ssbc/go-ssb-room/v2/internal/repo"
|
|
"github.com/ssbc/go-ssb-room/v2/roomdb"
|
|
)
|
|
|
|
func TestAliases(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
testRepo := filepath.Join("testrun", t.Name())
|
|
os.RemoveAll(testRepo)
|
|
tr := repo.New(testRepo)
|
|
|
|
// fake feed for testing, looks ok at least
|
|
newMember, err := refs.NewFeedRefFromBytes(bytes.Repeat([]byte("acab"), 8), refs.RefAlgoFeedSSB1)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
// 64 bytes of random for testing (validation is handled by the handlers)
|
|
testSig := make([]byte, 64)
|
|
rand.Read(testSig)
|
|
|
|
db, err := Open(tr)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
r := require.New(t)
|
|
|
|
lst, err := db.Aliases.List(ctx)
|
|
r.NoError(err)
|
|
r.Len(lst, 0)
|
|
|
|
_, err = db.Aliases.GetByID(ctx, 9999)
|
|
r.Error(err)
|
|
r.EqualError(err, roomdb.ErrNotFound.Error())
|
|
|
|
_, err = db.Aliases.Resolve(ctx, "unknown")
|
|
r.Error(err)
|
|
r.EqualError(err, roomdb.ErrNotFound.Error())
|
|
|
|
err = db.Aliases.Revoke(ctx, "unknown")
|
|
r.Error(err)
|
|
r.EqualError(errors.Unwrap(err), roomdb.ErrNotFound.Error())
|
|
})
|
|
|
|
t.Run("register and revoke again", func(t *testing.T) {
|
|
r := require.New(t)
|
|
|
|
testName := "flaky"
|
|
|
|
// shouldnt work while not a member
|
|
err = db.Aliases.Register(ctx, testName, newMember, testSig)
|
|
r.Error(err)
|
|
|
|
// allow the member
|
|
_, err = db.Members.Add(ctx, newMember, roomdb.RoleMember)
|
|
r.NoError(err)
|
|
|
|
err = db.Aliases.Register(ctx, testName, newMember, testSig)
|
|
r.NoError(err)
|
|
|
|
// should have one alias now
|
|
lst, err := db.Aliases.List(ctx)
|
|
r.NoError(err)
|
|
r.Len(lst, 1)
|
|
|
|
aliasByID, err := db.Aliases.GetByID(ctx, lst[0].ID)
|
|
r.NoError(err)
|
|
r.Equal(testName, aliasByID.Name)
|
|
r.Equal(testSig, aliasByID.Signature)
|
|
|
|
resolvedAlias, err := db.Aliases.Resolve(ctx, testName)
|
|
r.NoError(err)
|
|
r.Equal(aliasByID, resolvedAlias)
|
|
|
|
err = db.Aliases.Revoke(ctx, testName)
|
|
r.NoError(err)
|
|
|
|
_, err = db.Aliases.GetByID(ctx, lst[0].ID)
|
|
r.Error(err)
|
|
r.EqualError(err, roomdb.ErrNotFound.Error())
|
|
|
|
_, err = db.Aliases.Resolve(ctx, testName)
|
|
r.Error(err)
|
|
r.EqualError(err, roomdb.ErrNotFound.Error())
|
|
})
|
|
}
|
|
|
|
func TestAliasesUniqueError(t *testing.T) {
|
|
r := require.New(t)
|
|
ctx := context.Background()
|
|
|
|
testRepo := filepath.Join("testrun", t.Name())
|
|
os.RemoveAll(testRepo)
|
|
tr := repo.New(testRepo)
|
|
|
|
db, err := Open(tr)
|
|
r.NoError(err)
|
|
|
|
// fake feed for testing, looks ok at least
|
|
newMember, err := refs.NewFeedRefFromBytes(bytes.Repeat([]byte("acab"), 8), refs.RefAlgoFeedSSB1)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
// 64 bytes of random for testing (validation is handled by the handlers)
|
|
testSig := make([]byte, 64)
|
|
rand.Read(testSig)
|
|
|
|
testName := "thealias"
|
|
|
|
// allow the member
|
|
_, err = db.Members.Add(ctx, newMember, roomdb.RoleMember)
|
|
r.NoError(err)
|
|
|
|
err = db.Aliases.Register(ctx, testName, newMember, testSig)
|
|
r.NoError(err)
|
|
|
|
// should have one alias now
|
|
lst, err := db.Aliases.List(ctx)
|
|
r.NoError(err)
|
|
r.Len(lst, 1)
|
|
|
|
err = db.Aliases.Register(ctx, testName, newMember, testSig)
|
|
r.Error(err)
|
|
var takenErr roomdb.ErrAliasTaken
|
|
r.True(errors.As(err, &takenErr), "expected a special error value")
|
|
r.Equal(testName, takenErr.Name)
|
|
}
|