func TestMethods(t *testing.T) {
bus := setup(t)
errc := make(chan error, 1)
instA := flux.InstanceID("steamy-windows-89")
mockA := &platform.MockPlatform{
AllServicesAnswer: []platform.Service{platform.Service{}},
RegradeError: platform.RegradeError{flux.ServiceID("foo/bar"): errors.New("foo barred")},
}
subscribe(t, bus, errc, instA, mockA)
plat, err := bus.Connect(instA)
if err != nil {
t.Fatal(err)
}
ss, err := plat.AllServices("", nil)
if err != nil {
t.Fatal(err)
}
if len(mockA.AllServicesAnswer) != len(ss) {
t.Fatalf("Expected %d result, got %d", len(mockA.AllServicesAnswer), len(ss))
}
err = plat.Regrade([]platform.RegradeSpec{})
if _, ok := err.(platform.RegradeError); !ok {
t.Fatalf("expected RegradeError, got %+v", err)
}
mockB := &platform.MockPlatform{
AllServicesError: errors.New("just didn't feel like it"),
SomeServicesAnswer: []platform.Service{platform.Service{}, platform.Service{}},
}
instB := flux.InstanceID("smokey-water-72")
subscribe(t, bus, errc, instB, mockB)
platB, err := bus.Connect(instB)
if err != nil {
t.Fatal(err)
}
ss, err = platB.SomeServices([]flux.ServiceID{})
if err != nil {
t.Fatal(err)
}
if len(mockB.SomeServicesAnswer) != len(ss) {
t.Fatalf("Expected %d result, got %d", len(mockB.SomeServicesAnswer), len(ss))
}
ss, err = platB.AllServices("", nil)
if err == nil {
t.Fatal("expected error but didn't get one")
}
close(errc)
err = <-errc
if err != nil {
t.Fatalf("expected nil from subscription channel, but got err %v", err)
}
}
func TestRPC(t *testing.T) {
namespace := "space-of-names"
serviceID := flux.ServiceID(namespace + "/service")
serviceList := []flux.ServiceID{serviceID}
services := flux.ServiceIDSet{}
services.Add(serviceList)
regrades := []platform.RegradeSpec{
platform.RegradeSpec{
ServiceID: serviceID,
NewDefinition: []byte("imagine a definition here"),
},
}
serviceAnswer := []platform.Service{
platform.Service{
ID: flux.ServiceID("foobar/hello"),
IP: "10.32.1.45",
Metadata: map[string]string{},
Status: "ok",
Containers: platform.ContainersOrExcuse{
Containers: []platform.Container{
platform.Container{
Name: "frobnicator",
Image: "quay.io/example.com/frob:v0.4.5",
},
},
},
},
platform.Service{},
}
mock := &platform.MockPlatform{
AllServicesArgTest: func(ns string, ss flux.ServiceIDSet) error {
if !(ns == namespace &&
ss.Contains(serviceID)) {
return fmt.Errorf("did not get expected args, got %q, %+v", ns, ss)
}
return nil
},
AllServicesAnswer: serviceAnswer,
SomeServicesArgTest: func(ss []flux.ServiceID) error {
if !reflect.DeepEqual(ss, serviceList) {
return fmt.Errorf("did not get expected args, got %+v", ss)
}
return nil
},
SomeServicesAnswer: serviceAnswer,
RegradeArgTest: func(specs []platform.RegradeSpec) error {
if !reflect.DeepEqual(regrades, specs) {
return fmt.Errorf("did not get expected args, got %+v", specs)
}
return nil
},
RegradeError: nil,
}
clientConn, serverConn := pipes()
server, err := NewServer(mock)
if err != nil {
t.Fatal(err)
}
go server.ServeConn(serverConn)
client := NewClient(clientConn)
if err := client.Ping(); err != nil {
t.Fatal(err)
}
ss, err := client.AllServices(namespace, services)
if err != nil {
t.Error(err)
}
if !reflect.DeepEqual(ss, mock.AllServicesAnswer) {
t.Error(fmt.Errorf("expected %d result(s), got %+v", len(mock.AllServicesAnswer), ss))
}
mock.AllServicesError = fmt.Errorf("all services query failure")
ss, err = client.AllServices(namespace, services)
if err == nil {
t.Error("expected error, got nil")
}
ss, err = client.SomeServices(serviceList)
if err != nil {
t.Error(err)
}
if !reflect.DeepEqual(ss, mock.SomeServicesAnswer) {
t.Error(fmt.Errorf("expected %d result(s), got %+v", len(mock.SomeServicesAnswer), ss))
}
mock.SomeServicesError = fmt.Errorf("fail for some reason")
ss, err = client.SomeServices(serviceList)
if err == nil {
t.Error("expected error, got nil")
}
err = client.Regrade(regrades)
if err != nil {
t.Error(err)
}
regradeErrors := platform.RegradeError{
serviceID: fmt.Errorf("it just failed"),
}
mock.RegradeError = regradeErrors
err = client.Regrade(regrades)
if !reflect.DeepEqual(err, regradeErrors) {
t.Errorf("expected RegradeError, got %#v", err)
}
}