Function TestPostgresArray

internal/dbtest/pg_test.go:23–70

(t *testing.T)

)

func TestPostgresArray(t *testing.T) {
	type Model struct {
		ID     int64     `bun:",pk,autoincrement"`
		Array1 []string  `bun:",array"`
		Array2 *[]string `bun:",array"`
		Array3 *[]string `bun:",array"`
		Array4 []*string `bun:",array"`
	}

	db := pg(t)
	t.Cleanup(func() { db.Close() })
	mustResetModel(t, ctx, db, (*Model)(nil))

	str1 := "hello"
	str2 := "world"
	model1 := &Model{
		ID:     123,
		Array1: []string{"one", "two", "three"},
		Array2: &[]string{"hello", "world"},
		Array4: []*string{&str1, &str2},
	}
	_, err := db.NewInsert().Model(model1).Exec(ctx)
	require.NoError(t, err)

	model2 := new(Model)
	err = db.NewSelect().Model(model2).Scan(ctx)
	require.NoError(t, err)
	require.Equal(t, model1, model2)

	var strs []string
	err = db.NewSelect().Model((*Model)(nil)).
		Column("array1").
		Scan(ctx, pgdialect.Array(&strs))
	require.NoError(t, err)
	require.Equal(t, []string{"one", "two", "three"}, strs)

	err = db.NewSelect().Model((*Model)(nil)).
		Column("array3").
		Scan(ctx, pgdialect.Array(&strs))
	require.NoError(t, err)
	require.Nil(t, strs)

	err = db.NewSelect().Model((*Model)(nil)).
		Column("array4").
		Scan(ctx, pgdialect.Array(&strs))
	require.NoError(t, err)
	require.Equal(t, []string{"hello", "world"}, strs)
}

func TestPostgresArrayQuote(t *testing.T) {
	db := pg(t)