mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-04 23:45:07 +00:00
add duration type
This commit is contained in:
parent
5cd9ca506a
commit
6bf96ecbd7
15
internal/config/types/duration.go
Normal file
15
internal/config/types/duration.go
Normal file
@ -0,0 +1,15 @@
|
||||
package types
|
||||
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
type Duration struct {
|
||||
time.Duration
|
||||
}
|
||||
|
||||
func (d *Duration) UnmarshalText(data []byte) error {
|
||||
var err error
|
||||
d.Duration, err = time.ParseDuration(string(data))
|
||||
return err
|
||||
}
|
46
internal/config/types/duration_test.go
Normal file
46
internal/config/types/duration_test.go
Normal file
@ -0,0 +1,46 @@
|
||||
package types
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestDuration_UnmarshalText(t *testing.T) {
|
||||
type args struct {
|
||||
data []byte
|
||||
}
|
||||
tests := []struct {
|
||||
name string
|
||||
args args
|
||||
wantErr bool
|
||||
want time.Duration
|
||||
}{
|
||||
{
|
||||
"ok",
|
||||
args{
|
||||
data: []byte("10s"),
|
||||
},
|
||||
false,
|
||||
time.Duration(10 * time.Second),
|
||||
},
|
||||
{
|
||||
"error",
|
||||
args{
|
||||
data: []byte("10"),
|
||||
},
|
||||
true,
|
||||
time.Duration(0),
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
d := &Duration{}
|
||||
if err := d.UnmarshalText(tt.args.data); (err != nil) != tt.wantErr {
|
||||
t.Errorf("UnmarshalText() error = %v, wantErr %v", err, tt.wantErr)
|
||||
}
|
||||
if d.Duration != tt.want {
|
||||
t.Errorf("UnmarshalText() got = %v, want %v", d.Duration, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user