71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package accounts
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func TestNewAccountant(t *testing.T) {
|
|
// Very basic verify here for now
|
|
accountant := NewAccountant()
|
|
if accountant == nil {
|
|
t.Error("Failed to create accountant")
|
|
}
|
|
}
|
|
|
|
func TestAccountant_RegisterAccount(t *testing.T) {
|
|
|
|
accountant := NewAccountant()
|
|
|
|
// Start by making two accounts
|
|
|
|
namea := uuid.New().String()
|
|
acca, err := accountant.RegisterAccount(namea)
|
|
if err != nil {
|
|
t.Error(err)
|
|
} else if acca.Name != namea {
|
|
t.Errorf("Missmatched account name after register, expected: %s, actual: %s", namea, acca.Name)
|
|
}
|
|
|
|
nameb := uuid.New().String()
|
|
accb, err := accountant.RegisterAccount(nameb)
|
|
if err != nil {
|
|
t.Error(err)
|
|
} else if accb.Name != nameb {
|
|
t.Errorf("Missmatched account name after register, expected: %s, actual: %s", nameb, acca.Name)
|
|
}
|
|
|
|
// Verify our accounts have differing IDs
|
|
if acca.Id == accb.Id {
|
|
t.Error("Duplicate account IDs fo separate accounts")
|
|
}
|
|
|
|
// Verify another request gets rejected
|
|
_, err = accountant.RegisterAccount(namea)
|
|
if err == nil {
|
|
t.Error("Duplicate account name did not produce error")
|
|
}
|
|
}
|
|
|
|
func TestAccountant_AssignPrimary(t *testing.T) {
|
|
accountant := NewAccountant()
|
|
if len(accountant.Accounts) != 0 {
|
|
t.Error("New accountant created with non-zero account number")
|
|
}
|
|
|
|
name := uuid.New().String()
|
|
a, err := accountant.RegisterAccount(name)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
inst := uuid.New()
|
|
|
|
err = accountant.AssignPrimary(a.Id, inst)
|
|
if err != nil {
|
|
t.Error("Failed to set primary for created account")
|
|
} else if accountant.Accounts[a.Id].Primary != inst {
|
|
t.Error("Primary for assigned account is incorrect")
|
|
}
|
|
}
|