43 lines
757 B
Go
43 lines
757 B
Go
|
package storage
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"database/sql"
|
||
|
"fmt"
|
||
|
pb "go.unistack.org/unistack-org/pkgdash/proto/go_generate"
|
||
|
"go.unistack.org/unistack-org/pkgdash/storage/sqlite"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
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, err := sqlite.NewStorage(conn)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
s, ok := st.(Storage)
|
||
|
if !ok {
|
||
|
t.Fatal("typecast error")
|
||
|
}
|
||
|
|
||
|
req := &pb.GetModuleReq{
|
||
|
Id: []uint64{1, 2, 5, 40},
|
||
|
}
|
||
|
|
||
|
module, err := s.GetModule(context.Background(), req)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
fmt.Println(module)
|
||
|
}
|