70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"go.unistack.org/pkgdash/internal/storage/sqlite"
|
||
|
|
pb "go.unistack.org/pkgdash/proto"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetModule(t *testing.T) {
|
||
|
|
conn, err := sql.Open("sqlite3", "/Users/devstigneev_local/GolandProjects/unistack/pkgdash/identifier.sqlite")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer conn.Close()
|
||
|
|
if err = conn.Ping(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
st := sqlite.NewStorage()
|
||
|
|
store := st(conn, fs)
|
||
|
|
|
||
|
|
s, ok := store.(Storage)
|
||
|
|
if !ok {
|
||
|
|
t.Fatal("dont implements interface Storage")
|
||
|
|
}
|
||
|
|
|
||
|
|
req := &pb.GetModuleReq{
|
||
|
|
Id: []uint64{1, 2, 3},
|
||
|
|
}
|
||
|
|
|
||
|
|
module, err := s.GetModule(context.Background(), req)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println(module)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetComment(t *testing.T) {
|
||
|
|
conn, err := sql.Open("sqlite3", "/Users/devstigneev_local/GolandProjects/unistack/pkgdash/identifier.sqlite")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer conn.Close()
|
||
|
|
if err = conn.Ping(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
st := sqlite.NewStorage()
|
||
|
|
store := st(conn, fs)
|
||
|
|
|
||
|
|
s, ok := store.(Storage)
|
||
|
|
if !ok {
|
||
|
|
t.Fatal("dont implements interface Storage")
|
||
|
|
}
|
||
|
|
|
||
|
|
req := &pb.GetCommentsReq{
|
||
|
|
Id: []uint64{1, 2, 3, 15},
|
||
|
|
}
|
||
|
|
comments, err := s.GetComment(context.Background(), req)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
fmt.Println(comments.Decode())
|
||
|
|
}
|