From 6d3846ac6285904bca74deb2a0a7d7ef65cbc91c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michael=20Mur=C3=A9?= <batolettre@gmail.com>
Date: Thu, 27 Feb 2025 14:27:02 +0100
Subject: [PATCH] container: add a helper for test only

---
 pkg/container/writer.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/pkg/container/writer.go b/pkg/container/writer.go
index 6bd9dcc..a8ca0ff 100644
--- a/pkg/container/writer.go
+++ b/pkg/container/writer.go
@@ -120,3 +120,20 @@ func (ctn Writer) toWriter(header header, w io.Writer) (err error) {
 
 	return ipld.EncodeStreaming(encoder, node, cbor.Encode)
 }
+
+// ToReader convert a container Writer into a Reader.
+// Most likely, you only want to use this in tests for convenience.
+// This is not optimized and can panic.
+func (ctn Writer) ToReader() Reader {
+	data, err := ctn.ToBytes()
+	if err != nil {
+		panic(err)
+	}
+
+	reader, err := FromBytes(data)
+	if err != nil {
+		panic(err)
+	}
+
+	return reader
+}