63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func TestNewAccountant(t *testing.T) {
|
|
// Very basic verify here for now
|
|
accountant := NewSimpleAccountant()
|
|
if accountant == nil {
|
|
t.Error("Failed to create accountant")
|
|
}
|
|
}
|
|
|
|
func TestAccountant_RegisterAccount(t *testing.T) {
|
|
|
|
accountant := NewSimpleAccountant()
|
|
|
|
// 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 another request gets rejected
|
|
_, err = accountant.RegisterAccount(namea)
|
|
if err == nil {
|
|
t.Error("Duplicate account name did not produce error")
|
|
}
|
|
}
|
|
|
|
func TestAccountant_AssignGetData(t *testing.T) {
|
|
accountant := NewSimpleAccountant()
|
|
|
|
name := uuid.New().String()
|
|
a, err := accountant.RegisterAccount(name)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = accountant.AssignData(a.Name, "key", "value")
|
|
if err != nil {
|
|
t.Error("Failed to set data for created account")
|
|
} else if id, err := accountant.GetValue(a.Name, "key"); err != nil {
|
|
t.Error("Failed to get data for account")
|
|
} else if id != "value" {
|
|
t.Error("Fetched data is incorrect for account")
|
|
}
|
|
}
|