MaysWind
197 lines
10 KiB
Go
197 lines
10 KiB
Go
package models
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/mayswind/ezbookkeeping/pkg/core"
|
|
"github.com/mayswind/ezbookkeeping/pkg/utils"
|
|
)
|
|
|
|
func TestUserFeatureRestrictionsAdd(t *testing.T) {
|
|
var featureRestrictions UserFeatureRestrictions
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PASSWORD)
|
|
expectedValue := UserFeatureRestrictions(1)
|
|
assert.Equal(t, expectedValue, featureRestrictions)
|
|
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_EMAIL)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PROFILE_BASIC_INFO)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_AVATAR)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_REVOKE_OTHER_SESSION)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_ENABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_DISABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_FORGET_PASSWORD)
|
|
expectedValue = UserFeatureRestrictions(255)
|
|
assert.Equal(t, expectedValue, featureRestrictions)
|
|
}
|
|
|
|
func TestUserFeatureRestrictionsRemove(t *testing.T) {
|
|
var featureRestrictions UserFeatureRestrictions
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PASSWORD)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_EMAIL)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PROFILE_BASIC_INFO)
|
|
featureRestrictions = featureRestrictions.Remove(USER_FEATURE_RESTRICTION_TYPE_UPDATE_EMAIL)
|
|
featureRestrictions = featureRestrictions.Remove(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PROFILE_BASIC_INFO)
|
|
featureRestrictions = featureRestrictions.Remove(USER_FEATURE_RESTRICTION_TYPE_UPDATE_AVATAR)
|
|
featureRestrictions = featureRestrictions.Remove(USER_FEATURE_RESTRICTION_TYPE_REVOKE_OTHER_SESSION)
|
|
expectedValue := UserFeatureRestrictions(1)
|
|
assert.Equal(t, expectedValue, featureRestrictions)
|
|
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_AVATAR)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_REVOKE_OTHER_SESSION)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_ENABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_DISABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_FORGET_PASSWORD)
|
|
featureRestrictions = featureRestrictions.Remove(USER_FEATURE_RESTRICTION_TYPE_ENABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Remove(USER_FEATURE_RESTRICTION_TYPE_DISABLE_2FA)
|
|
expectedValue = UserFeatureRestrictions(153)
|
|
assert.Equal(t, expectedValue, featureRestrictions)
|
|
}
|
|
|
|
func TestUserFeatureRestrictionsContains(t *testing.T) {
|
|
var featureRestrictions UserFeatureRestrictions
|
|
assert.False(t, featureRestrictions.Contains(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PASSWORD))
|
|
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PASSWORD)
|
|
assert.True(t, featureRestrictions.Contains(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PASSWORD))
|
|
assert.False(t, featureRestrictions.Contains(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PROFILE_BASIC_INFO))
|
|
}
|
|
|
|
func TestUserFeatureRestrictionsString(t *testing.T) {
|
|
var featureRestrictions UserFeatureRestrictions
|
|
expectedValue := ""
|
|
actualValue := featureRestrictions.String()
|
|
assert.Equal(t, expectedValue, actualValue)
|
|
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PASSWORD)
|
|
expectedValue = "Update Password"
|
|
actualValue = featureRestrictions.String()
|
|
assert.Equal(t, expectedValue, actualValue)
|
|
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_FORGET_PASSWORD)
|
|
expectedValue = "Update Password,Forget Password"
|
|
actualValue = featureRestrictions.String()
|
|
assert.Equal(t, expectedValue, actualValue)
|
|
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_EMAIL)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_PROFILE_BASIC_INFO)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_UPDATE_AVATAR)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_REVOKE_OTHER_SESSION)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_ENABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_DISABLE_2FA)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_IMPORT_TRANSACTION)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_EXPORT_TRANSACTION)
|
|
featureRestrictions = featureRestrictions.Add(USER_FEATURE_RESTRICTION_TYPE_CLEAR_ALL_DATA)
|
|
expectedValue = "Update Password," +
|
|
"Update Email," +
|
|
"Update Profile Basic Info," +
|
|
"Update Avatar," +
|
|
"Logout Other Session," +
|
|
"Enable Two-Factor Authentication," +
|
|
"Disable Enable Two-Factor Authentication," +
|
|
"Forget Password," +
|
|
"Import Transactions," +
|
|
"Export Transactions," +
|
|
"Clear All Data"
|
|
actualValue = featureRestrictions.String()
|
|
assert.Equal(t, expectedValue, actualValue)
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsNone(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_NONE,
|
|
}
|
|
|
|
assert.Equal(t, false, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(time.Now().Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsAll(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_ALL,
|
|
}
|
|
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(time.Now().Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsTodayOrLater(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_TODAY_OR_LATER,
|
|
}
|
|
|
|
now := time.Now()
|
|
todayFirstDatetime := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, time.Local)
|
|
yesterdayLastDatetime := todayFirstDatetime.Add(-1 * time.Second)
|
|
todayLastDatetime := yesterdayLastDatetime.Add(24 * time.Hour)
|
|
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(todayFirstDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(todayLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, false, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(yesterdayLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsLast24HourOrLater(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_LAST_24H_OR_LATER,
|
|
}
|
|
|
|
now := time.Now()
|
|
twentyfourHourBeforeDatetime := now.Add(-24 * time.Hour).Add(-1 * time.Second)
|
|
|
|
assert.Equal(t, false, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(twentyfourHourBeforeDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(twentyfourHourBeforeDatetime.Add(1*time.Second).Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsThisWeekOrLater(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_THIS_WEEK_OR_LATER,
|
|
FirstDayOfWeek: core.WEEKDAY_MONDAY,
|
|
}
|
|
|
|
now := time.Now()
|
|
thisWeekFirstDatetime := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, time.Local)
|
|
|
|
if thisWeekFirstDatetime.Weekday() == time.Sunday {
|
|
thisWeekFirstDatetime = thisWeekFirstDatetime.Add(-6 * 24 * time.Hour)
|
|
} else if thisWeekFirstDatetime.Weekday() != time.Monday {
|
|
thisWeekFirstDatetime = thisWeekFirstDatetime.Add(time.Duration(1-thisWeekFirstDatetime.Weekday()) * 24 * time.Hour)
|
|
}
|
|
|
|
lastWeekLastDatetime := thisWeekFirstDatetime.Add(-1 * time.Second)
|
|
thisWeekLastDatetime := lastWeekLastDatetime.Add(24 * time.Hour)
|
|
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(thisWeekFirstDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(thisWeekLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, false, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(lastWeekLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsThisMonthOrLater(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_THIS_MONTH_OR_LATER,
|
|
}
|
|
|
|
now := time.Now()
|
|
thisMonthFirstDatetime := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, time.Local)
|
|
lastMonthLastDatetime := thisMonthFirstDatetime.Add(-1 * time.Second)
|
|
thisMonthLastDatetime := lastMonthLastDatetime.Add(24 * time.Hour)
|
|
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(thisMonthFirstDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(thisMonthLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, false, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(lastMonthLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|
|
|
|
func TestUserCanEditTransactionByTransactionTime_ScopeIsThisYearOrLater(t *testing.T) {
|
|
user := &User{
|
|
TransactionEditScope: TRANSACTION_EDIT_SCOPE_THIS_YEAR_OR_LATER,
|
|
}
|
|
|
|
now := time.Now()
|
|
thisYearFirstDatetime := time.Date(now.Year(), 1, 1, 0, 0, 0, 0, time.Local)
|
|
lastYearLastDatetime := thisYearFirstDatetime.Add(-1 * time.Second)
|
|
thisYearLastDatetime := lastYearLastDatetime.Add(24 * time.Hour)
|
|
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(thisYearFirstDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, true, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(thisYearLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
assert.Equal(t, false, user.CanEditTransactionByTransactionTime(utils.GetMinTransactionTimeFromUnixTime(lastYearLastDatetime.Unix()), utils.GetServerTimezoneOffsetMinutes()))
|
|
}
|