fix(pkg/test): cast mime-bundle content to appropriate type

bevzzz · Jan 25, 2024 · bc5d78c · bc5d78c
1 parent d3f8a53
commit bc5d78c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pkg/test/cell.go b/pkg/test/cell.go
@@ -166,7 +166,15 @@ func (mb mimebundle) MimeType() string {
 }
 
 func (mb mimebundle) Text() []byte {
-	return mb[mb.MimeType()].([]byte)
+	if txt, ok := mb[mb.MimeType()]; ok {
+		switch v := txt.(type) {
+		case []byte:
+			return v
+		case string:
+			return []byte(v)
+		}
+	}
+	return nil
 }
 
 func (mb mimebundle) PlainText() []byte {