user_test.go

// Copyright © Martin Tournoij – This file is part of GoatCounter and published
// under the terms of a slightly modified EUPL v1.2 license, which can be found
// in the LICENSE file or at https://license.goatcounter.com

package goatcounter_test

import (
	"testing"
	"time"

	"zgo.at/goatcounter/v2"
	"zgo.at/tz"
	"zgo.at/zstd/zint"
	"zgo.at/zstd/ztime"
)

func TestUserEmailReportRange(t *testing.T) {
	now := time.Date(2019, 6, 18, 14, 42, 0, 0, time.UTC)
	ztime.Now = func() time.Time { return now }
	t.Cleanup(func() { ztime.Now = func() time.Time { return time.Now().UTC() } })
	wita := tz.MustNew("", "Asia/Makassar")

	tests := []struct {
		user               goatcounter.User
		wantStart, wantEnd time.Time
	}{
		{goatcounter.User{
			LastReportAt: now,
			Settings: goatcounter.UserSettings{
				EmailReports: zint.Int(goatcounter.EmailReportDaily),
				Timezone:     tz.UTC,
			},
		}, ztime.FromString("2019-06-18 00:00:00"), ztime.FromString("2019-06-18 23:59:59")},
		{goatcounter.User{
			LastReportAt: now,
			Settings: goatcounter.UserSettings{
				SundayStartsWeek: false,
				EmailReports:     zint.Int(goatcounter.EmailReportWeekly),
				Timezone:         tz.UTC,
			},
		}, ztime.FromString("2019-06-17 00:00:00"), ztime.FromString("2019-06-23 23:59:59")},
		{goatcounter.User{
			LastReportAt: now,
			Settings: goatcounter.UserSettings{
				SundayStartsWeek: true,
				EmailReports:     zint.Int(goatcounter.EmailReportWeekly),
				Timezone:         tz.UTC,
			},
		}, ztime.FromString("2019-06-16 00:00:00"), ztime.FromString("2019-06-22 23:59:59")},
		{goatcounter.User{
			LastReportAt: now,
			Settings: goatcounter.UserSettings{
				SundayStartsWeek: false,
				EmailReports:     zint.Int(goatcounter.EmailReportBiWeekly),
				Timezone:         tz.UTC,
			},
		}, ztime.FromString("2019-06-17 00:00:00"), ztime.FromString("2019-06-30 23:59:59")},
		{goatcounter.User{
			LastReportAt: now,
			Settings: goatcounter.UserSettings{
				SundayStartsWeek: true,
				EmailReports:     zint.Int(goatcounter.EmailReportBiWeekly),
				Timezone:         tz.UTC,
			},
		}, ztime.FromString("2019-06-16 00:00:00"), ztime.FromString("2019-06-29 23:59:59")},

		{goatcounter.User{
			LastReportAt: now,
			Settings: goatcounter.UserSettings{
				EmailReports: zint.Int(goatcounter.EmailReportDaily),
				Timezone:     wita,
			},
		}, ztime.FromString("2019-06-17 16:00:00"), ztime.FromString("2019-06-18 15:59:59")},
	}

	for _, tt := range tests {
		t.Run("", func(t *testing.T) {
			rng := tt.user.EmailReportRange()
			if !rng.Start.Equal(tt.wantStart) {
				t.Errorf("start wrong\nwant: %s\nhave: %s\n", tt.wantStart, rng.Start)
			}
			if !rng.End.Equal(tt.wantEnd) {
				t.Errorf("end wrong\nwant: %s\nhave: %s\n", tt.wantEnd, rng.End)
			}
		})
	}
}