Begin building a common compliance test suite.

.github/workflows/go.yml

@@ -35,7 +35,10 @@ jobs:
       run: go test -v -coverprofile=profile.cov ./...
 
     - name: Race Test
-      run: go test -race ./...
+      run: |
+          for i in `find . -type d -mindepth 1 -maxdepth 1 | egrep -v ".git|.github|fluent"`; do
+            go test -race ./$i
+          done
 
     - name: Coveralls
       if: ${{ matrix.go == '1.14' }}

chk/chk.go

@@ -0,0 +1,51 @@
+// Package chk implements checks against the gRIBI client return values, it can be
+// used to determine whether there are expected results within a specific set of return
+// values.
+package chk
+
+import (
+	"github.com/openconfig/gribigo/client"
+	"google.golang.org/grpc/status"
+	"google.golang.org/protobuf/proto"
+)
+
+// HasElectionID returns true if the result queue supplied contains an update with the election ID
+// with the value of the uint128{low,high} value specified.
+func HasElectionID(res []*client.OpResult, low, high uint64) bool {
+	for _, r := range res {
+		if v := r.CurrentServerElectionID; v != nil {
+			if v.High == high && v.Low == low {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+// HasSuccessfulSessionParams checks whether the res results array contains a session parameters
+// result that is successful.
+func HasSuccessfulSessionParams(res []*client.OpResult) bool {
+	for _, r := range res {
+		if v := r.SessionParameters; v != nil {
+			return true
+		}
+	}
+	return false
+}
+
+// HasRecvClientErrorWithStatus checks whether the supplied ClientErr ce containa  status with
+// the code and details set to the values supplied in want.
+func HasRecvClientErrorWithStatus(ce *client.ClientErr, want *status.Status) bool {
+	for _, e := range ce.Recv {
+		s, ok := status.FromError(e)
+		if !ok {
+			continue
+		}
+		ns := s.Proto()
+		ns.Message = "" // blank out message so that we don't compare it.
+		if proto.Equal(ns, want.Proto()) {
+			return true
+		}
+	}
+	return false
+}

chk/chk_test.go

@@ -0,0 +1,62 @@
+package chk
+
+import (
+	"testing"
+
+	"github.com/openconfig/gribigo/client"
+
+	spb "github.com/openconfig/gribi/v1/proto/service"
+)
+
+func TestHasElectionID(t *testing.T) {
+	tests := []struct {
+		desc     string
+		inResult []*client.OpResult
+		inLow    uint64
+		inHigh   uint64
+		want     bool
+	}{{
+		desc: "is present",
+		inResult: []*client.OpResult{{
+			CurrentServerElectionID: &spb.Uint128{
+				High: 0,
+				Low:  42,
+			},
+		}},
+		inLow:  42,
+		inHigh: 0,
+		want:   true,
+	}, {
+		desc:     "is not present",
+		inResult: []*client.OpResult{},
+	}}
+
+	for _, tt := range tests {
+		if got, want := HasElectionID(tt.inResult, tt.inLow, tt.inHigh), tt.want; got != want {
+			t.Errorf("%s: HasElectionDI(%s, %d, %d): did not get expected result, got: %v, want: %v", tt.desc, tt.inResult, tt.inLow, tt.inHigh, got, want)
+		}
+	}
+}
+
+func TestHasSuccessfulSessionParams(t *testing.T) {
+	tests := []struct {
+		desc     string
+		inResult []*client.OpResult
+		want     bool
+	}{{
+		desc: "is present",
+		inResult: []*client.OpResult{{
+			SessionParameters: &spb.SessionParametersResult{},
+		}},
+		want: true,
+	}, {
+		desc:     "is not present",
+		inResult: []*client.OpResult{},
+	}}
+
+	for _, tt := range tests {
+		if got, want := HasSuccessfulSessionParams(tt.inResult), tt.want; got != want {
+			t.Errorf("%s: HasSuccessfulSessionParams(%s): did not get expected result, got: %v, want: %v", tt.desc, tt.inResult, got, want)
+		}
+	}
+}