87 lines
1.7 KiB
Go
87 lines
1.7 KiB
Go
package client_git
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
pb "go.unistack.org/unistack-org/pkgdash/proto"
|
|
"go.unistack.org/unistack-org/pkgdash/storage"
|
|
"go.unistack.org/unistack-org/pkgdash/storage/postgres"
|
|
"go.unistack.org/unistack-org/pkgdash/storage/sqlite"
|
|
"google.golang.org/protobuf/types/known/wrapperspb"
|
|
"testing"
|
|
)
|
|
|
|
func TestClientPG(t *testing.T) {
|
|
dsn := fmt.Sprintf("user=%s password=%s host=%s port=%s dbname=%s sslmode=disable", "test", "123", "localhost", "5432", "postgres")
|
|
conn, err := sql.Open("postgres", dsn)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer conn.Close()
|
|
if err = conn.Ping(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
st, err := postgres.NewStorage(conn)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s, ok := st.(storage.Storage)
|
|
if !ok {
|
|
t.Fatal("typecast error")
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
_ = cancel
|
|
cli := NewClient(1)
|
|
|
|
ch := cli.Run(ctx, s)
|
|
|
|
data := &pb.AddPackageReq{
|
|
Name: wrapperspb.String("test"),
|
|
Url: wrapperspb.String("https://github.com/dantedenis/service_history.git"),
|
|
}
|
|
|
|
ch <- data
|
|
|
|
<-cli.Done()
|
|
}
|
|
|
|
func TestClientLite(t *testing.T) {
|
|
conn, err := sql.Open("sqlite3", "../../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.Storage)
|
|
if !ok {
|
|
t.Fatal("typecast error")
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
_ = cancel
|
|
cli := NewClient(1)
|
|
|
|
ch := cli.Run(ctx, s)
|
|
|
|
data := &pb.AddPackageReq{
|
|
Name: wrapperspb.String("test"),
|
|
Url: wrapperspb.String("https://github.com/dantedenis/service_history.git"),
|
|
}
|
|
|
|
ch <- data
|
|
|
|
<-cli.Done()
|
|
}
|