status-im/status-go

package db

import (
    "testing"
    "time"

    "github.com/stretchr/testify/require"

    "github.com/status-im/status-go/eth-node/types"
)

func createInMemStore(t *testing.T) HistoryStore {
    db, err := NewMemoryDB()
    require.NoError(t, err)
    return NewHistoryStore(LevelDBStorage{db: db})
}

func TestGetNewHistory(t *testing.T) {
    topic := types.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 := types.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 := types.Hash{1}
    _, err := store.GetRequest(id)
    require.Error(t, err)
    req := store.NewRequest()
    req.ID = id

    th, err := store.GetHistory(types.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 := types.Hash{1}
    idTwo := types.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)
}