urdt/ussd
5
0
Fork 0
Code Issues 50 Pull Requests 7 Actions Packages Projects Releases Wiki Activity

make the VoucherMetadata more descriptive

Browse Source
This commit is contained in:
Alfred Kamanda 2024-10-30 18:40:03 +03:00
parent caafe495be
commit 8b399781e8
Signed by: Alfred-mk
GPG Key ID: 7EA3D01708908703
4 changed files with 29 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
internal/handlers/ussd/menuhandler.go
View File

@ -58,14 +58,6 @@ func (fm *FlagManager) GetFlag(label string) (uint32, error) {
return fm.parser.GetFlag(label) return fm.parser.GetFlag(label)
} }
// VoucherMetadata helps organize voucher data fields
type VoucherMetadata struct {
Symbol string
Balance string
Decimal string
Address string
}
type Handlers struct { type Handlers struct {
pe *persist.Persister pe *persist.Persister
st *state.State st *state.State
@ -1122,10 +1114,10 @@ func (h *Handlers) CheckVouchers(ctx context.Context, sym string, input []byte)
// Store all voucher data // Store all voucher data
dataMap := map[string]string{ dataMap := map[string]string{
"sym": data.Symbol, "sym": data.Symbols,
"bal": data.Balance, "bal": data.Balances,
"deci": data.Decimal, "deci": data.Decimals,
"addr": data.Address, "addr": data.Addresses,
} }
for key, value := range dataMap { for key, value := range dataMap {

24
internal/handlers/ussd/menuhandler_test.go
View File

@ -2096,19 +2096,19 @@ func TestSetVoucher(t *testing.T) {
ctx := context.WithValue(context.Background(), "SessionId", sessionId) ctx := context.WithValue(context.Background(), "SessionId", sessionId)
// Define the temporary voucher data // Define the temporary voucher data
tempData := &VoucherMetadata{ tempData := &dataserviceapi.TokenHoldings{
Symbol: "SRF", TokenSymbol: "SRF",
Balance: "200", Balance: "200",
Decimal: "6", TokenDecimals: "6",
Address: "0xd4c288865Ce0985a481Eef3be02443dF5E2e4Ea9", ContractAddress: "0xd4c288865Ce0985a481Eef3be02443dF5E2e4Ea9",
} }
// Define the expected active entries // Define the expected active entries
activeEntries := map[utils.DataTyp][]byte{ activeEntries := map[utils.DataTyp][]byte{
utils.DATA_ACTIVE_SYM: []byte(tempData.Symbol), utils.DATA_ACTIVE_SYM: []byte(tempData.TokenSymbol),
utils.DATA_ACTIVE_BAL: []byte(tempData.Balance), utils.DATA_ACTIVE_BAL: []byte(tempData.Balance),
utils.DATA_ACTIVE_DECIMAL: []byte(tempData.Decimal), utils.DATA_ACTIVE_DECIMAL: []byte(tempData.TokenDecimals),
utils.DATA_ACTIVE_ADDRESS: []byte(tempData.Address), utils.DATA_ACTIVE_ADDRESS: []byte(tempData.ContractAddress),
} }
// Define the temporary entries to be cleared // Define the temporary entries to be cleared
@ -2120,10 +2120,10 @@ func TestSetVoucher(t *testing.T) {
} }
// Mocking ReadEntry calls for temporary data retrieval // Mocking ReadEntry calls for temporary data retrieval
mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_SYM).Return([]byte(tempData.Symbol), nil) mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_SYM).Return([]byte(tempData.TokenSymbol), nil)
mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_BAL).Return([]byte(tempData.Balance), nil) mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_BAL).Return([]byte(tempData.Balance), nil)
mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_DECIMAL).Return([]byte(tempData.Decimal), nil) mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_DECIMAL).Return([]byte(tempData.TokenDecimals), nil)
mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_ADDRESS).Return([]byte(tempData.Address), nil) mockDataStore.On("ReadEntry", ctx, sessionId, utils.DATA_TEMPORARY_ADDRESS).Return([]byte(tempData.ContractAddress), nil)
// Mocking WriteEntry calls for setting active data // Mocking WriteEntry calls for setting active data
for key, value := range activeEntries { for key, value := range activeEntries {
@ -2143,7 +2143,7 @@ func TestSetVoucher(t *testing.T) {
assert.NoError(t, err) assert.NoError(t, err)
assert.Equal(t, string(tempData.Symbol), res.Content) assert.Equal(t, string(tempData.TokenSymbol), res.Content)
mockDataStore.AssertExpectations(t) mockDataStore.AssertExpectations(t)
} }

18
internal/utils/vouchers.go
View File

@ -9,12 +9,12 @@ import (
dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api" dataserviceapi "github.com/grassrootseconomics/ussd-data-service/pkg/api"
) )
// VoucherMetadata helps organize voucher data fields // VoucherMetadata helps organize data fields
type VoucherMetadata struct { type VoucherMetadata struct {
Symbol string Symbols string
Balance string Balances string
Decimal string Decimals string
Address string Addresses string
} }
// ProcessVouchers converts holdings into formatted strings // ProcessVouchers converts holdings into formatted strings
@ -29,10 +29,10 @@ func ProcessVouchers(holdings []dataserviceapi.TokenHoldings) VoucherMetadata {
addresses = append(addresses, fmt.Sprintf("%d:%s", i+1, h.ContractAddress)) addresses = append(addresses, fmt.Sprintf("%d:%s", i+1, h.ContractAddress))
} }
data.Symbol = strings.Join(symbols, "\n") data.Symbols = strings.Join(symbols, "\n")
data.Balance = strings.Join(balances, "\n") data.Balances = strings.Join(balances, "\n")
data.Decimal = strings.Join(decimals, "\n") data.Decimals = strings.Join(decimals, "\n")
data.Address = strings.Join(addresses, "\n") data.Addresses = strings.Join(addresses, "\n")
return data return data
} }

8
internal/utils/vouchers_test.go
View File

@ -68,10 +68,10 @@ func TestProcessVouchers(t *testing.T) {
} }
expectedResult := VoucherMetadata{ expectedResult := VoucherMetadata{
Symbol: "1:SRF\n2:MILO", Symbols: "1:SRF\n2:MILO",
Balance: "1:100\n2:200", Balances: "1:100\n2:200",
Decimal: "1:6\n2:4", Decimals: "1:6\n2:4",
Address: "1:0xd4c288865Ce\n2:0x41c188d63Qa", Addresses: "1:0xd4c288865Ce\n2:0x41c188d63Qa",
} }
result := ProcessVouchers(holdings) result := ProcessVouchers(holdings)