You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package contextx
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestUnmarshalContext(t *testing.T) {
|
|
type Person struct {
|
|
Name string `ctx:"name"`
|
|
Age int `ctx:"age"`
|
|
}
|
|
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, "name", "kevin")
|
|
ctx = context.WithValue(ctx, "age", 20)
|
|
|
|
var person Person
|
|
err := For(ctx, &person)
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "kevin", person.Name)
|
|
assert.Equal(t, 20, person.Age)
|
|
}
|
|
|
|
func TestUnmarshalContextWithOptional(t *testing.T) {
|
|
type Person struct {
|
|
Name string `ctx:"name"`
|
|
Age int `ctx:"age,optional"`
|
|
}
|
|
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, "name", "kevin")
|
|
|
|
var person Person
|
|
err := For(ctx, &person)
|
|
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "kevin", person.Name)
|
|
assert.Equal(t, 0, person.Age)
|
|
}
|
|
|
|
func TestUnmarshalContextWithMissing(t *testing.T) {
|
|
type Person struct {
|
|
Name string `ctx:"name"`
|
|
Age int `ctx:"age"`
|
|
}
|
|
type name string
|
|
const PersonNameKey name = "name"
|
|
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, PersonNameKey, "kevin")
|
|
|
|
var person Person
|
|
err := For(ctx, &person)
|
|
|
|
assert.NotNil(t, err)
|
|
}
|