mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-11 19:07:30 +00:00
add duration type
This commit is contained in:
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)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user