83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
package db
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
whispertypes "github.com/status-im/status-protocol-go/transport/whisper/types"
|
|
statusproto "github.com/status-im/status-protocol-go/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func createInMemStore(t *testing.T) HistoryStore {
|
|
db, err := NewMemoryDB()
|
|
require.NoError(t, err)
|
|
return NewHistoryStore(LevelDBStorage{db: db})
|
|
}
|
|
|
|
func TestGetNewHistory(t *testing.T) {
|
|
topic := whispertypes.TopicType{1}
|
|
duration := time.Hour
|
|
store := createInMemStore(t)
|
|
th, err := store.GetHistory(topic, duration)
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, th.Duration)
|
|
require.Equal(t, topic, th.Topic)
|
|
}
|
|
|
|
func TestGetExistingHistory(t *testing.T) {
|
|
topic := whispertypes.TopicType{1}
|
|
duration := time.Hour
|
|
store := createInMemStore(t)
|
|
th, err := store.GetHistory(topic, duration)
|
|
require.NoError(t, err)
|
|
|
|
now := time.Now()
|
|
th.Current = now
|
|
require.NoError(t, th.Save())
|
|
|
|
th, err = store.GetHistory(topic, duration)
|
|
require.NoError(t, err)
|
|
require.Equal(t, now.Unix(), th.Current.Unix())
|
|
}
|
|
|
|
func TestNewHistoryRequest(t *testing.T) {
|
|
store := createInMemStore(t)
|
|
id := statusproto.Hash{1}
|
|
req, err := store.GetRequest(id)
|
|
require.Error(t, err)
|
|
req = store.NewRequest()
|
|
req.ID = id
|
|
|
|
th, err := store.GetHistory(whispertypes.TopicType{1}, time.Hour)
|
|
require.NoError(t, err)
|
|
req.AddHistory(th)
|
|
require.NoError(t, req.Save())
|
|
|
|
req, err = store.GetRequest(id)
|
|
require.NoError(t, err)
|
|
require.Len(t, req.Histories(), 1)
|
|
}
|
|
|
|
func TestGetAllRequests(t *testing.T) {
|
|
store := createInMemStore(t)
|
|
idOne := statusproto.Hash{1}
|
|
idTwo := statusproto.Hash{2}
|
|
|
|
req := store.NewRequest()
|
|
req.ID = idOne
|
|
require.NoError(t, req.Save())
|
|
|
|
all, err := store.GetAllRequests()
|
|
require.NoError(t, err)
|
|
require.Len(t, all, 1)
|
|
|
|
req = store.NewRequest()
|
|
req.ID = idTwo
|
|
require.NoError(t, req.Save())
|
|
|
|
all, err = store.GetAllRequests()
|
|
require.NoError(t, err)
|
|
require.Len(t, all, 2)
|
|
}
|