You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.4 KiB
69 lines
1.4 KiB
package arg_test
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"gitoa.ru/go-4devs/config/provider/arg"
|
|
"gitoa.ru/go-4devs/config/test"
|
|
)
|
|
|
|
func TestProvider(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
args := os.Args
|
|
|
|
defer func() {
|
|
os.Args = args
|
|
}()
|
|
|
|
os.Args = []string{
|
|
"main.go",
|
|
"--listen=8080",
|
|
"--config=config.hcl",
|
|
"--url=http://4devs.io",
|
|
"--url=https://4devs.io",
|
|
"--timeout=1m",
|
|
"--timeout=1h",
|
|
"--start-at=2010-01-02T15:04:05Z",
|
|
"--end-after=2009-01-02T15:04:05Z",
|
|
"--end-after=2008-01-02T15:04:05+03:00",
|
|
}
|
|
read := []test.Read{
|
|
test.NewRead(8080, "listen"),
|
|
test.NewRead("config.hcl", "config"),
|
|
test.NewRead(test.Time("2010-01-02T15:04:05Z"), "start-at"),
|
|
test.NewReadUnmarshal(&[]string{"http://4devs.io", "https://4devs.io"}, &[]string{}, "url"),
|
|
test.NewReadUnmarshal(&[]Duration{{time.Minute}, {time.Hour}}, &[]Duration{}, "timeout"),
|
|
test.NewReadUnmarshal(&[]time.Time{
|
|
test.Time("2009-01-02T15:04:05Z"),
|
|
test.Time("2008-01-02T15:04:05+03:00"),
|
|
}, &[]time.Time{}, "end-after"),
|
|
}
|
|
|
|
prov := arg.New()
|
|
|
|
test.Run(t, prov, read)
|
|
}
|
|
|
|
type Duration struct {
|
|
time.Duration
|
|
}
|
|
|
|
func (d *Duration) UnmarshalJSON(in []byte) error {
|
|
o, err := time.ParseDuration(strings.Trim(string(in), `"`))
|
|
if err != nil {
|
|
return fmt.Errorf("parse:%w", err)
|
|
}
|
|
|
|
d.Duration = o
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Duration) MarshalJSON() ([]byte, error) {
|
|
return []byte(fmt.Sprintf("%q", d)), nil
|
|
}
|
|
|