pkgdash/service/client_git/client_test.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()
}