72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package gocron_server_test
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
gocron_server "github.com/strnophix/gocron-server/pkg"
|
|
pb "github.com/strnophix/gocron-server/pkg/proto"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/test/bufconn"
|
|
)
|
|
|
|
const bufSize = 1024 * 1024
|
|
|
|
var lis *bufconn.Listener
|
|
var c *Counter
|
|
|
|
func init() {
|
|
c = &Counter{Current: 1}
|
|
|
|
lis = bufconn.Listen(bufSize)
|
|
gs := grpc.NewServer()
|
|
s := gocron_server.NewSchedulerService()
|
|
defer s.Shutdown()
|
|
|
|
incr := gocron_server.NewUnitExecFn(c.Increment)
|
|
unit := gocron_server.NewManualUnit("incr", incr)
|
|
s.AddUnit(unit)
|
|
|
|
pb.RegisterSchedulerServer(gs, s)
|
|
go func() {
|
|
if err := gs.Serve(lis); err != nil {
|
|
log.Fatalf("Failed to serve: %v", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func bufDialer(context.Context, string) (net.Conn, error) {
|
|
return lis.Dial()
|
|
}
|
|
|
|
func TestRunJob(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
conn, err := grpc.DialContext(ctx, "bufnet", grpc.WithContextDialer(bufDialer), grpc.WithInsecure())
|
|
if err != nil {
|
|
t.Fatalf("Failed to dial bufnet: %v", err)
|
|
}
|
|
|
|
defer conn.Close()
|
|
|
|
client := pb.NewSchedulerClient(conn)
|
|
|
|
_, err = client.RunJob(ctx, &pb.RunJobRequest{UnitName: "niku"})
|
|
if err == nil {
|
|
t.Fatalf("RunJob should have returned an error for unit niku")
|
|
}
|
|
|
|
_, err = client.RunJob(ctx, &pb.RunJobRequest{UnitName: "incr"})
|
|
if err != nil {
|
|
t.Fatalf("RunJob call should have passed but got: %v", err)
|
|
}
|
|
|
|
time.Sleep(1 * time.Second)
|
|
if c.Current != 2 {
|
|
t.Fatalf("RunJob call `incr` should have incremented counter but stays at: %d", c.Current)
|
|
}
|
|
}
|