mirror of
https://github.com/1f349/tulip.git
synced 2024-12-23 00:34:07 +00:00
53 lines
2.1 KiB
Go
53 lines
2.1 KiB
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/MrMelon54/pronouns"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"golang.org/x/text/language"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func encode(data any) string {
|
||
|
|
j, err := json.Marshal(map[string]any{"value": data})
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return string(j)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStringScanner_MarshalJSON(t *testing.T) {
|
||
|
|
assert.Equal(t, "{\"value\":\"Hello world\"}", encode(NullStringScanner{sql.NullString{String: "Hello world", Valid: true}}))
|
||
|
|
assert.Equal(t, "{\"value\":null}", encode(NullStringScanner{sql.NullString{String: "Hello world", Valid: false}}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDateScanner_MarshalJSON(t *testing.T) {
|
||
|
|
location, err := time.LoadLocation("Europe/London")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "{\"value\":\"2006-01-02\"}", encode(NullDateScanner{sql.NullTime{Time: time.Date(2006, time.January, 2, 0, 0, 0, 0, time.UTC), Valid: true}}))
|
||
|
|
assert.Equal(t, "{\"value\":\"2006-08-01\"}", encode(NullDateScanner{sql.NullTime{Time: time.Date(2006, time.August, 2, 0, 0, 0, 0, location), Valid: true}}))
|
||
|
|
assert.Equal(t, "{\"value\":null}", encode(NullDateScanner{}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocationScanner_MarshalJSON(t *testing.T) {
|
||
|
|
location, err := time.LoadLocation("Europe/London")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "{\"value\":\"Europe/London\"}", encode(LocationScanner{location}))
|
||
|
|
assert.Equal(t, "{\"value\":\"UTC\"}", encode(LocationScanner{time.UTC}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLocaleScanner_MarshalJSON(t *testing.T) {
|
||
|
|
assert.Equal(t, "{\"value\":\"en-US\"}", encode(LocaleScanner{language.AmericanEnglish}))
|
||
|
|
assert.Equal(t, "{\"value\":\"en-GB\"}", encode(LocaleScanner{language.BritishEnglish}))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPronounScanner_MarshalJSON(t *testing.T) {
|
||
|
|
assert.Equal(t, "{\"value\":\"they/them\"}", encode(PronounScanner{pronouns.TheyThem}))
|
||
|
|
assert.Equal(t, "{\"value\":\"he/him\"}", encode(PronounScanner{pronouns.HeHim}))
|
||
|
|
assert.Equal(t, "{\"value\":\"she/her\"}", encode(PronounScanner{pronouns.SheHer}))
|
||
|
|
assert.Equal(t, "{\"value\":\"it/its\"}", encode(PronounScanner{pronouns.ItIts}))
|
||
|
|
assert.Equal(t, "{\"value\":\"one/one's\"}", encode(PronounScanner{pronouns.OneOnes}))
|
||
|
|
}
|