@@ -0,0 +1,27 @@
|
||||
package usermod
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestRenewDramaExpire(t *testing.T) {
|
||||
now := time.Date(2026, 8, 26, 12, 0, 0, 0, time.UTC)
|
||||
tests := []struct {
|
||||
name string
|
||||
current time.Time
|
||||
days int
|
||||
want time.Time
|
||||
}{
|
||||
{name: "starts from now", days: 30, want: now.AddDate(0, 0, 30)},
|
||||
{name: "extends active entitlement", current: now.AddDate(0, 0, 10), days: 30, want: now.AddDate(0, 0, 40)},
|
||||
{name: "refunds active entitlement", current: now.AddDate(0, 0, 30), days: -7, want: now.AddDate(0, 0, 23)},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := RenewDramaExpire(tt.current, now, tt.days); !got.Equal(tt.want) {
|
||||
t.Fatalf("RenewDramaExpire() = %s, want %s", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user