mirror of https://github.com/status-im/go-waku.git
53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
|
package persistence
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"log"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
_ "github.com/mattn/go-sqlite3" // Blank import to register the sqlite3 driver
|
||
|
|
"github.com/status-im/go-waku/waku/v2/protocol/pb"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewMock() *sql.DB {
|
||
|
|
db, err := sql.Open("sqlite3", ":memory:")
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return db
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDbStore(t *testing.T) {
|
||
|
|
db := NewMock()
|
||
|
|
option := WithDB(db)
|
||
|
|
store, err := NewDBStore(option)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
res, err := store.GetAll()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Empty(t, res)
|
||
|
|
|
||
|
|
err = store.Put(
|
||
|
|
&pb.Index{
|
||
|
|
Digest: []byte("digest"),
|
||
|
|
ReceiverTime: 1.0,
|
||
|
|
SenderTime: 1.0,
|
||
|
|
},
|
||
|
|
"test",
|
||
|
|
&pb.WakuMessage{
|
||
|
|
Payload: []byte("payload"),
|
||
|
|
ContentTopic: "contenttopic",
|
||
|
|
Version: 1,
|
||
|
|
Timestamp: 1.0,
|
||
|
|
Proof: []byte("proof"),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
res, err = store.GetAll()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NotEmpty(t, res)
|
||
|
|
}
|