165 lines
4.9 KiB
Go
165 lines
4.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type subscriptionExpiryRepoStub struct {
|
|
listCalls int
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) Create(context.Context, *UserSubscription) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) GetByID(context.Context, int64) (*UserSubscription, error) {
|
|
return nil, ErrSubscriptionNotFound
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) GetByUserIDAndGroupID(context.Context, int64, int64) (*UserSubscription, error) {
|
|
return nil, ErrSubscriptionNotFound
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) GetActiveByUserIDAndGroupID(context.Context, int64, int64) (*UserSubscription, error) {
|
|
return nil, ErrSubscriptionNotFound
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) Update(context.Context, *UserSubscription) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) Delete(context.Context, int64) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ListByUserID(context.Context, int64) ([]UserSubscription, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ListActiveByUserID(context.Context, int64) ([]UserSubscription, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ListByGroupID(context.Context, int64, pagination.PaginationParams) ([]UserSubscription, *pagination.PaginationResult, error) {
|
|
return nil, nil, nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) List(context.Context, pagination.PaginationParams, *int64, *int64, string, string, string, string) ([]UserSubscription, *pagination.PaginationResult, error) {
|
|
r.listCalls++
|
|
return nil, &pagination.PaginationResult{Page: 1, Pages: 1}, nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ExistsByUserIDAndGroupID(context.Context, int64, int64) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ExtendExpiry(context.Context, int64, time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) UpdateStatus(context.Context, int64, string) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) UpdateNotes(context.Context, int64, string) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ActivateWindows(context.Context, int64, time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ResetDailyUsage(context.Context, int64, time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ResetWeeklyUsage(context.Context, int64, time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) ResetMonthlyUsage(context.Context, int64, time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) IncrementUsage(context.Context, int64, float64) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpiryRepoStub) BatchUpdateExpiredStatus(context.Context) (int64, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
type subscriptionExpirySettingRepoStub struct {
|
|
values map[string]string
|
|
err error
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) Get(context.Context, string) (*Setting, error) {
|
|
return nil, ErrSettingNotFound
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) GetValue(_ context.Context, key string) (string, error) {
|
|
if r.err != nil {
|
|
return "", r.err
|
|
}
|
|
value, ok := r.values[key]
|
|
if !ok {
|
|
return "", ErrSettingNotFound
|
|
}
|
|
return value, nil
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) Set(context.Context, string, string) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) GetMultiple(context.Context, []string) (map[string]string, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) SetMultiple(context.Context, map[string]string) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) GetAll(context.Context) (map[string]string, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *subscriptionExpirySettingRepoStub) Delete(context.Context, string) error {
|
|
return nil
|
|
}
|
|
|
|
func TestSubscriptionExpiryService_ExpiryReminderEnabledDefaultsToTrue(t *testing.T) {
|
|
svc := NewSubscriptionExpiryService(nil, time.Minute)
|
|
svc.SetSettingRepository(&subscriptionExpirySettingRepoStub{values: map[string]string{}})
|
|
|
|
require.True(t, svc.expiryReminderEnabled(context.Background()))
|
|
}
|
|
|
|
func TestSubscriptionExpiryService_ExpiryReminderDisabledSkipsSubscriptionScan(t *testing.T) {
|
|
repo := &subscriptionExpiryRepoStub{}
|
|
settingRepo := &subscriptionExpirySettingRepoStub{
|
|
values: map[string]string{SettingKeySubscriptionExpiryNotifyEnabled: "false"},
|
|
}
|
|
svc := NewSubscriptionExpiryService(repo, time.Minute)
|
|
svc.SetSettingRepository(settingRepo)
|
|
svc.SetNotificationEmailService(NewNotificationEmailService(settingRepo, nil))
|
|
|
|
svc.sendExpiryReminders(context.Background())
|
|
|
|
require.Zero(t, repo.listCalls)
|
|
}
|
|
|
|
func TestSubscriptionExpiryService_ExpiryReminderSettingReadErrorFailsClosed(t *testing.T) {
|
|
svc := NewSubscriptionExpiryService(nil, time.Minute)
|
|
svc.SetSettingRepository(&subscriptionExpirySettingRepoStub{err: errors.New("db down")})
|
|
|
|
require.False(t, svc.expiryReminderEnabled(context.Background()))
|
|
}
|