Add DefaultUnsafe global

Setting it to true will enable Iter's `Unsafe()` behavior for all queries.
This commit is contained in:
Daniel Lohse
2020-01-15 15:11:54 +01:00
committed by Michal Jan Matczuk
parent 0f06ab790c
commit a471f98eed
2 changed files with 21 additions and 1 deletions

View File

@@ -276,6 +276,22 @@ func TestUnsafe(t *testing.T) {
t.Fatal("select failed")
}
})
t.Run("DefaultUnsafe select", func(t *testing.T) {
gocqlx.DefaultUnsafe = true
defer func() { gocqlx.DefaultUnsafe = true }()
var v []UnsafeTable
i := gocqlx.Iter(session.Query(`SELECT * FROM unsafe_table`))
if err := i.Select(&v); err != nil {
t.Fatal(err)
}
if len(v) != 1 {
t.Fatal("select failed")
}
if v[0].Testtext != "test" {
t.Fatal("select failed")
}
})
}
func TestNotFound(t *testing.T) {