From af30a55ca657fe00794ba6cd6e2f9d125300b2fa Mon Sep 17 00:00:00 2001
From: Simone Gotti <simone.gotti@gmail.com>
Date: Wed, 11 Oct 2023 15:35:59 +0200
Subject: [PATCH] testutils/db: handle col type bool on export

Since we are retrieving any type from the db rows.Scan we should handle
the possible returned types for a bool col type. Currently handle
booleans, int64 0 or 1, uint8[] of chars making a string of value "true"
or "false".
---
 internal/testutil/db.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/internal/testutil/db.go b/internal/testutil/db.go
index 9b3df6a69..bca2dad5a 100644
--- a/internal/testutil/db.go
+++ b/internal/testutil/db.go
@@ -448,6 +448,24 @@ func (c *DBContext) Export(ctx context.Context, tables []string, w io.Writer) er
 						return errors.WithStack(err)
 					}
 					switch colType {
+					case ColTypeBool:
+						switch vt := v.(type) {
+						case bool:
+							data.Values[col] = vt
+						case int64:
+							if vt != 0 && vt != 1 {
+								return errors.Errorf("unknown type int64 value %d for bool column type", vt)
+							}
+							data.Values[col] = vt != 0
+						case []uint8:
+							bv, err := strconv.ParseBool(string(vt))
+							if err != nil {
+								return errors.WithStack(err)
+							}
+							data.Values[col] = bv
+						default:
+							return errors.Errorf("unknown type %T for bool column type", v)
+						}
 					case ColTypeJSON:
 						var vj any
 						if err := json.Unmarshal(v.([]byte), &vj); err != nil {