mdiluz/rove
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor accounts to store a data map rather than just a rover ID

Browse source
This commit is contained in:
Marc Di Luzio 2020-06-10 22:48:45 +01:00
parent 7749854eb7
commit 8f25f55658
4 changed files with 33 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

20
pkg/accounts/accounts.go
View file

@ -2,8 +2,6 @@ package accounts
import (
"fmt"
"github.com/google/uuid"
)
const kAccountsFileName = "rove-accounts.json"
@ -13,8 +11,8 @@ type Account struct {
// Name simply describes the account and must be unique
Name string `json:"name"`
// Rover represents the rover that this account owns
Rover uuid.UUID `json:"rover"`
// Data represents internal account data
Data map[string]string `json:"data"`
}
// Represents the accountant data to store
@ -52,12 +50,12 @@ func (a *Accountant) RegisterAccount(name string) (acc Account, err error) {
return
}
// AssignRover assigns rover ownership of an rover to an account
func (a *Accountant) AssignRover(account string, rover uuid.UUID) error {
// AssignRover assigns data to an account
func (a *Accountant) AssignData(account string, key string, value string) error {
// Find the account matching the ID
if this, ok := a.Accounts[account]; ok {
this.Rover = rover
this.Data[key] = value
a.Accounts[account] = this
} else {
return fmt.Errorf("no account found for id: %s", account)
@ -67,13 +65,11 @@ func (a *Accountant) AssignRover(account string, rover uuid.UUID) error {
}
// GetRover gets the rover rover for the account
func (a *Accountant) GetRover(account string) (uuid.UUID, error) {
func (a *Accountant) GetData(account string, key string) (string, error) {
// Find the account matching the ID
if this, ok := a.Accounts[account]; !ok {
return uuid.UUID{}, fmt.Errorf("no account found for id: %s", account)
} else if this.Rover == uuid.Nil {
return uuid.UUID{}, fmt.Errorf("no rover spawned for account %s", account)
return "", fmt.Errorf("no account found for id: %s", account)
} else {
return this.Rover, nil
return this.Data[key], nil
}
}

18
pkg/accounts/accounts_test.go
View file

@ -43,7 +43,7 @@ func TestAccountant_RegisterAccount(t *testing.T) {
}
}
func TestAccountant_AssignGetRover(t *testing.T) {
func TestAccountant_AssignGetData(t *testing.T) {
accountant := NewAccountant()
if len(accountant.Accounts) != 0 {
t.Error("New accountant created with non-zero account number")
@ -55,16 +55,12 @@ func TestAccountant_AssignGetRover(t *testing.T) {
t.Error(err)
}
inst := uuid.New()
err = accountant.AssignRover(a.Name, inst)
err = accountant.AssignData(a.Name, "key", "value")
if err != nil {
t.Error("Failed to set rover for created account")
} else if accountant.Accounts[a.Name].Rover != inst {
t.Error("Rover for assigned account is incorrect")
} else if id, err := accountant.GetRover(a.Name); err != nil {
t.Error("Failed to get rover for account")
} else if id != inst {
t.Error("Fetched rover is incorrect for account")
t.Error("Failed to set data for created account")
} else if id, err := accountant.GetData(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")
}
}