mirror of
https://github.com/jackc/pgx.git
synced 2025-05-02 05:30:04 +00:00
bcc139a365
commit fc020c24ac9590f6547f8ad1d291fc75b4873a84
Author: Nicholas Wilson <nicholas.wilson@realvnc.com>
Date: Wed Jul 24 12:32:18 2019 +0100
Add support for pgtype.UUID to write into any [16]byte type
126 lines
3.2 KiB
Go
126 lines
3.2 KiB
Go
package pgtype_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgtype"
|
|
"github.com/jackc/pgtype/testutil"
|
|
)
|
|
|
|
func TestUUIDTranscode(t *testing.T) {
|
|
testutil.TestSuccessfulTranscode(t, "uuid", []interface{}{
|
|
&pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present},
|
|
&pgtype.UUID{Status: pgtype.Null},
|
|
})
|
|
}
|
|
|
|
type SomeUUIDType [16]byte
|
|
|
|
func TestUUIDSet(t *testing.T) {
|
|
successfulTests := []struct {
|
|
source interface{}
|
|
result pgtype.UUID
|
|
}{
|
|
{
|
|
source: nil,
|
|
result: pgtype.UUID{Status: pgtype.Null},
|
|
},
|
|
{
|
|
source: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
|
|
result: pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present},
|
|
},
|
|
{
|
|
source: []byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
|
|
result: pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present},
|
|
},
|
|
{
|
|
source: SomeUUIDType{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
|
|
result: pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present},
|
|
},
|
|
{
|
|
source: ([]byte)(nil),
|
|
result: pgtype.UUID{Status: pgtype.Null},
|
|
},
|
|
{
|
|
source: "00010203-0405-0607-0809-0a0b0c0d0e0f",
|
|
result: pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present},
|
|
},
|
|
}
|
|
|
|
for i, tt := range successfulTests {
|
|
var r pgtype.UUID
|
|
err := r.Set(tt.source)
|
|
if err != nil {
|
|
t.Errorf("%d: %v", i, err)
|
|
}
|
|
|
|
if r != tt.result {
|
|
t.Errorf("%d: expected %v to convert to %v, but it was %v", i, tt.source, tt.result, r)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestUUIDAssignTo(t *testing.T) {
|
|
{
|
|
src := pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present}
|
|
var dst [16]byte
|
|
expected := [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}
|
|
|
|
err := src.AssignTo(&dst)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if dst != expected {
|
|
t.Errorf("expected %v to assign %v, but result was %v", src, expected, dst)
|
|
}
|
|
}
|
|
|
|
{
|
|
src := pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present}
|
|
var dst []byte
|
|
expected := []byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}
|
|
|
|
err := src.AssignTo(&dst)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if bytes.Compare(dst, expected) != 0 {
|
|
t.Errorf("expected %v to assign %v, but result was %v", src, expected, dst)
|
|
}
|
|
}
|
|
|
|
{
|
|
src := pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present}
|
|
var dst SomeUUIDType
|
|
expected := [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}
|
|
|
|
err := src.AssignTo(&dst)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if dst != expected {
|
|
t.Errorf("expected %v to assign %v, but result was %v", src, expected, dst)
|
|
}
|
|
}
|
|
|
|
{
|
|
src := pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Status: pgtype.Present}
|
|
var dst string
|
|
expected := "00010203-0405-0607-0809-0a0b0c0d0e0f"
|
|
|
|
err := src.AssignTo(&dst)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if dst != expected {
|
|
t.Errorf("expected %v to assign %v, but result was %v", src, expected, dst)
|
|
}
|
|
}
|
|
|
|
}
|