update tests

viamrobotics · May 7, 2024 · 3f6c2f8 · 3f6c2f8
1 parent 32f0ad6
commit 3f6c2f8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/components/board/client_test.go b/components/board/client_test.go
@@ -152,11 +152,13 @@ func TestWorkingClient(t *testing.T) {
 		actualExtra = nil
 
 		// write analog
-		injectBoard.WriteAnalogFunc = func(ctx context.Context, pin string, value int32, extra map[string]interface{}) error {
+		injectAnalog.WriteFunc = func(ctx context.Context, value int, extra map[string]interface{}) error {
 			actualExtra = extra
 			return nil
 		}
-		err = injectBoard.WriteAnalog(context.Background(), "pin1", 6, expectedExtra)
+		analog2, err := injectBoard.AnalogByName("pin1")
+		test.That(t, err, test.ShouldBeNil)
+		err = analog2.Write(context.Background(), 6, expectedExtra)
 		test.That(t, err, test.ShouldBeNil)
 		test.That(t, actualExtra, test.ShouldResemble, expectedExtra)
 		actualExtra = nil

diff --git a/components/board/server_test.go b/components/board/server_test.go
@@ -597,10 +597,15 @@ func TestServerWriteAnalog(t *testing.T) {
 			test.That(t, err, test.ShouldBeNil)
 			var actualExtra map[string]interface{}
 
-			injectBoard.WriteAnalogFunc = func(ctx context.Context, pin string, value int32, extra map[string]interface{}) error {
+			injectAnalog := inject.Analog{}
+			injectAnalog.WriteFunc = func(ctx context.Context, value int, extra map[string]interface{}) error {
 				actualExtra = extra
 				return tc.injectErr
 			}
+			injectBoard.AnalogByNameFunc = func(pin string) (board.Analog, error) {
+				return &injectAnalog, nil
+			}
+
 			resp, err := server.WriteAnalog(ctx, tc.req)
 			if tc.expRespErr == "" {
 				test.That(t, err, test.ShouldBeNil)