RSDK-4074 - quick fix for logging (#210)

viam-modules · Jul 20, 2023 · b8fcea6 · b8fcea6
1 parent fd875ba
commit b8fcea6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sensorprocess/sensorprocess.go b/sensorprocess/sensorprocess.go
@@ -5,6 +5,7 @@ import (
 	"context"
 	"errors"
 	"math"
+	"strings"
 	"time"
 
 	"github.com/edaniels/golog"
@@ -51,9 +52,8 @@ func addSensorReading(
 	tsr, err := config.Lidar.TimedSensorReading(ctx)
 	if err != nil {
 		config.Logger.Warn(err)
-		return errors.Is(err, replaypcd.ErrEndOfDataset)
+		return strings.Contains(err.Error(), replaypcd.ErrEndOfDataset.Error())
 	}
-
 	if tsr.Replay {
 		addSensorReadingFromReplaySensor(ctx, tsr.Reading, tsr.ReadingTime, config)
 	} else {

diff --git a/testhelper/testhelper.go b/testhelper/testhelper.go
@@ -216,7 +216,7 @@ func getIntegrationLidar() *inject.Camera {
 func getFinishedReplaySensor() *inject.Camera {
 	cam := &inject.Camera{}
 	cam.NextPointCloudFunc = func(ctx context.Context) (pointcloud.PointCloud, error) {
-		return nil, replaypcd.ErrEndOfDataset
+		return nil, errors.Wrap(errors.New("wrapped error"), replaypcd.ErrEndOfDataset.Error())
 	}
 	cam.StreamFunc = func(ctx context.Context, errHandlers ...gostream.ErrorHandler) (gostream.VideoStream, error) {
 		return nil, errors.New("lidar not camera")