diff --git a/generator/templates.go b/generator/templates.go index cd32a70..224c295 100644 --- a/generator/templates.go +++ b/generator/templates.go @@ -96,4 +96,8 @@ func (e *{{camelize $.Name}}) Scan(value interface{}) error { } } +func (e {{camelize $.Name}}) String() string { + return string(e) +} + ` diff --git a/tests/select_test.go b/tests/select_test.go index fd3ae37..6ea0443 100644 --- a/tests/select_test.go +++ b/tests/select_test.go @@ -286,10 +286,10 @@ LIMIT 15; query := Film. INNER_JOIN(Language, Film.LanguageID.Eq(Language.LanguageID)). SELECT(Language.AllColumns, Film.AllColumns). - WHERE(Film.Rating.EqString(string(model.MpaaRating_NC17))). + WHERE(Film.Rating.EqString(model.MpaaRating_NC17.String())). LIMIT(15) - assertQuery(t, query, expectedSql, string(model.MpaaRating_NC17), int64(15)) + assertQuery(t, query, expectedSql, model.MpaaRating_NC17.String(), int64(15)) err := query.Query(db, &filmsPerLanguage)