fix scanning into enums (#145)

* fix scanning into enums

Signed-off-by: Joshua Rubin <[email protected]>

* test scanning into enum array

Signed-off-by: Joshua Rubin <[email protected]>

Author: joshuarubin

internal/dinosql/gen.go

@@ -851,6 +851,11 @@ const (
 	{{.Name}} {{.Type}} = "{{.Value}}"
 	{{- end}}
 )
+
+func (e *{{.Name}}) Scan(src interface{}) error {
+	*e = {{.Name}}(src.([]byte))
+	return nil
+}
 {{end}}
 
 {{range .Structs}}

internal/dinosql/testdata/ondeck/db_test.go

@@ -19,7 +19,7 @@ import (
 func id() string {
 	bytes := make([]byte, 10)
 	for i := 0; i < 10; i++ {
-		bytes[i] = byte(65 + rand.Intn(25)) //A=65 and Z = 65+25
+		bytes[i] = byte(65 + rand.Intn(25)) // A=65 and Z = 65+25
 	}
 	return string(bytes)
 }
@@ -115,6 +115,7 @@ func TestQueries(t *testing.T) {
 		City:            city.Slug,
 		SpotifyPlaylist: "spotify:uri",
 		Status:          StatusOpen,
+		Statuses:        []Status{StatusOpen, StatusClosed},
 		Tags:            []string{"rock", "punk"},
 	})
 	if err != nil {

internal/dinosql/testdata/ondeck/models.go

@@ -21,6 +21,7 @@ INSERT INTO venue (
     created_at,
     spotify_playlist,
     status,
+    statuses,
     tags
 ) VALUES (
     $1,
@@ -29,7 +30,8 @@ INSERT INTO venue (
     NOW(),
     $4,
     $5,
-    $6
+    $6,
+    $7
 ) RETURNING id;
 
 -- name: UpdateVenueName :one

internal/dinosql/testdata/ondeck/prepared/models.go

@@ -4,6 +4,7 @@ CREATE TABLE venues (
     id               SERIAL primary key,
     dropped          text,
     status           status       not null,
+    statuses         status[],
     slug             text         not null,
     name             varchar(255) not null,
     city             text         not null references city(slug),