feedset_test.go

// SPDX-FileCopyrightText: 2021 The Go-SSB Authors
//
// SPDX-License-Identifier: MIT

package ssb

import (
	"testing"

	refs "github.com/ssbc/go-ssb-refs"
	"github.com/stretchr/testify/require"
)

func TestFeedSetEmpty(t *testing.T) {
	r := require.New(t)

	fs := NewFeedSet(0)

	newkey, err := NewKeyPair(nil, refs.RefAlgoFeedSSB1)
	r.NoError(err)
	r.False(fs.Has(newkey.ID()))
}

func TestFeedSetCount(t *testing.T) {
	r := require.New(t)
	kps := make([]KeyPair, 50)

	fs := NewFeedSet(50)
	for i := 0; i < 50; i++ {
		var err error
		kps[i], err = NewKeyPair(nil, refs.RefAlgoFeedSSB1)
		r.NoError(err)
		err = fs.AddRef(kps[i].ID())
		r.NoError(err)
	}
	r.Equal(50, fs.Count())
	lst, err := fs.List()
	r.NoError(err)
	r.Len(lst, 50, "first len(List()) wrong")
	// twice
	for i := 0; i < 50; i++ {
		err := fs.AddRef(kps[i].ID())
		r.NoError(err)
	}
	r.Equal(50, fs.Count())
	lst, err = fs.List()
	r.NoError(err)
	r.Len(lst, 50, "twice len(List()) wrong")
	// some
	for i := 0; i < 15; i++ {
		err := fs.AddRef(kps[i].ID())
		r.NoError(err)
	}
	r.Equal(50, fs.Count())
	lst, err = fs.List()
	r.NoError(err)
	r.Len(lst, 50, "some len(List()) wrong")
}