pkgdash/storage/storage_test.go

package storage

import (
	"context"
	"database/sql"
	"fmt"
	pb "go.unistack.org/unistack-org/pkgdash/proto"
	"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)
}