test: align to interface changes

Signed-off-by: Boris Glimcher <[email protected]>
glimchb · Oct 9, 2023 · 660c0a5 · 660c0a5
1 parent a3b9b68
commit 660c0a5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/pkg/kvm/kvm_test.go b/pkg/kvm/kvm_test.go
@@ -5,6 +5,7 @@
 package kvm
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"net"
@@ -48,6 +49,9 @@ type stubJSONRRPC struct {
 	err error
 }
 
+// build time check that struct implements interface
+var _ spdk.JSONRPC = (*stubJSONRRPC)(nil)
+
 func (s stubJSONRRPC) GetID() uint64 {
 	return 0
 }
@@ -56,11 +60,11 @@ func (s stubJSONRRPC) StartUnixListener() net.Listener {
 	return nil
 }
 
-func (s stubJSONRRPC) GetVersion() string {
+func (s stubJSONRRPC) GetVersion(_ context.Context) string {
 	return ""
 }
 
-func (s stubJSONRRPC) Call(method string, _, result interface{}) error {
+func (s stubJSONRRPC) Call(_ context.Context, method string, _, result interface{}) error {
 	if method == "vhost_create_blk_controller" {
 		if s.err == nil {
 			resultCreateVirtioBLk, ok := result.(*spdk.VhostCreateBlkControllerResult)

diff --git a/pkg/middleend/qos_test.go b/pkg/middleend/qos_test.go
@@ -5,6 +5,7 @@
 package middleend
 
 import (
+	"context"
 	"fmt"
 	"net"
 	"testing"
@@ -33,6 +34,9 @@ type stubJSONRRPC struct {
 	params []any
 }
 
+// build time check that struct implements interface
+var _ spdk.JSONRPC = (*stubJSONRRPC)(nil)
+
 func (s *stubJSONRRPC) GetID() uint64 {
 	return 0
 }
@@ -41,11 +45,11 @@ func (s *stubJSONRRPC) StartUnixListener() net.Listener {
 	return nil
 }
 
-func (s *stubJSONRRPC) GetVersion() string {
+func (s *stubJSONRRPC) GetVersion(_ context.Context) string {
 	return ""
 }
 
-func (s *stubJSONRRPC) Call(_ string, param interface{}, _ interface{}) error {
+func (s *stubJSONRRPC) Call(_ context.Context, _ string, param interface{}, _ interface{}) error {
 	s.params = append(s.params, param)
 	return nil
 }