fix(correctedmemstore): pass CID to ErrNotFound

ipld · Oct 3, 2023 · a9406e4 · a9406e4
1 parent e92f567
commit a9406e4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/testutil/correctedmemstore.go b/testutil/correctedmemstore.go
@@ -42,10 +42,10 @@ func (cms *CorrectedMemStore) Get(ctx context.Context, key string) ([]byte, erro
 	} else if err != nil {
 		return nil, err
 	}
-
 	data, err := cms.ParentStore.Get(ctx, key)
 	if err != nil && err.Error() == "404" {
-		err = format.ErrNotFound{}
+		cid, _ := cid.Cast([]byte(key))
+		err = format.ErrNotFound{Cid: cid}
 	}
 	return data, err
 }
@@ -58,7 +58,8 @@ func (cms *CorrectedMemStore) GetStream(ctx context.Context, key string) (io.Rea
 	}
 	rc, err := cms.ParentStore.GetStream(ctx, key)
 	if err != nil && err.Error() == "404" {
-		err = format.ErrNotFound{}
+		cid, _ := cid.Cast([]byte(key))
+		err = format.ErrNotFound{Cid: cid}
 	}
 	return rc, err
 }