account_spark_shadow_test.go 583 B

12345678910111213141516171819
  1. package service
  2. import (
  3. "testing"
  4. "github.com/stretchr/testify/require"
  5. )
  6. func TestAccountSparkShadowHelpers(t *testing.T) {
  7. pid := int64(100)
  8. normal := &Account{ID: 100}
  9. require.False(t, normal.IsShadow())
  10. require.False(t, normal.IsCredentialShadow())
  11. require.Equal(t, QuotaDimensionGlobal, normal.QuotaDimensionOrDefault())
  12. shadow := &Account{ID: 200, ParentAccountID: &pid, QuotaDimension: QuotaDimensionSpark}
  13. require.True(t, shadow.IsShadow())
  14. require.True(t, shadow.IsCredentialShadow())
  15. require.Equal(t, QuotaDimensionSpark, shadow.QuotaDimensionOrDefault())
  16. }