Compare commits
No commits in common. "5a586eb67a676aa57e3c5019f147ff8e098b9282" and "73b501c8aa806019d1a7a31446511770332ce1b5" have entirely different histories.
5a586eb67a
...
73b501c8aa
@ -3218,142 +3218,3 @@ func TestResetOthersPin(t *testing.T) {
|
|||||||
|
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestResetValidPin(t *testing.T) {
|
|
||||||
ctx := context.Background()
|
|
||||||
|
|
||||||
fm, err := NewFlagManager(flagsPath)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
flag_valid_pin, _ := fm.GetFlag("flag_valid_pin")
|
|
||||||
|
|
||||||
expectedResult := resource.Result{
|
|
||||||
FlagReset: []uint32{flag_valid_pin},
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &MenuHandlers{
|
|
||||||
flagManager: fm,
|
|
||||||
}
|
|
||||||
|
|
||||||
res, err := h.ResetValidPin(ctx, "reset_valid_pin", []byte(""))
|
|
||||||
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
assert.Equal(t, expectedResult, res)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestResetUnregisteredNumber(t *testing.T) {
|
|
||||||
ctx := context.Background()
|
|
||||||
|
|
||||||
fm, err := NewFlagManager(flagsPath)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
flag_unregistered_number, _ := fm.GetFlag("flag_unregistered_number")
|
|
||||||
|
|
||||||
expectedResult := resource.Result{
|
|
||||||
FlagReset: []uint32{flag_unregistered_number},
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &MenuHandlers{
|
|
||||||
flagManager: fm,
|
|
||||||
}
|
|
||||||
|
|
||||||
res, err := h.ResetUnregisteredNumber(ctx, "reset_unregistered_number", []byte(""))
|
|
||||||
|
|
||||||
assert.NoError(t, err)
|
|
||||||
|
|
||||||
assert.Equal(t, expectedResult, res)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestConstructAccountAlias(t *testing.T) {
|
|
||||||
ctx, store := InitializeTestStore(t)
|
|
||||||
sessionId := "session123"
|
|
||||||
mockAccountService := new(mocks.MockAccountService)
|
|
||||||
|
|
||||||
ctx = context.WithValue(ctx, "SessionId", sessionId)
|
|
||||||
|
|
||||||
h := &MenuHandlers{
|
|
||||||
userdataStore: store,
|
|
||||||
accountService: mockAccountService,
|
|
||||||
}
|
|
||||||
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
firstName string
|
|
||||||
familyName string
|
|
||||||
publicKey string
|
|
||||||
expectedAlias string
|
|
||||||
aliasResponse *models.RequestAliasResult
|
|
||||||
aliasError error
|
|
||||||
expectedError error
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
name: "Valid alias construction",
|
|
||||||
firstName: "John",
|
|
||||||
familyName: "Doe",
|
|
||||||
publicKey: "pubkey123",
|
|
||||||
expectedAlias: "JohnDoeAlias",
|
|
||||||
aliasResponse: &models.RequestAliasResult{Alias: "JohnDoeAlias"},
|
|
||||||
aliasError: nil,
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "Account service fails to return alias",
|
|
||||||
firstName: "Jane",
|
|
||||||
familyName: "Smith",
|
|
||||||
publicKey: "pubkey456",
|
|
||||||
expectedAlias: "",
|
|
||||||
aliasResponse: nil,
|
|
||||||
aliasError: fmt.Errorf("service unavailable"),
|
|
||||||
expectedError: fmt.Errorf("Failed to retrieve alias: service unavailable"),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tt := range tests {
|
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
|
||||||
if tt.firstName != "" {
|
|
||||||
err := store.WriteEntry(ctx, sessionId, storedb.DATA_FIRST_NAME, []byte(tt.firstName))
|
|
||||||
require.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if tt.familyName != "" {
|
|
||||||
err := store.WriteEntry(ctx, sessionId, storedb.DATA_FAMILY_NAME, []byte(tt.familyName))
|
|
||||||
require.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if tt.publicKey != "" {
|
|
||||||
err := store.WriteEntry(ctx, sessionId, storedb.DATA_PUBLIC_KEY, []byte(tt.publicKey))
|
|
||||||
require.NoError(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
aliasInput := fmt.Sprintf("%s%s", tt.firstName, tt.familyName)
|
|
||||||
|
|
||||||
// Mock service behavior
|
|
||||||
mockAccountService.On(
|
|
||||||
"RequestAlias",
|
|
||||||
tt.publicKey,
|
|
||||||
aliasInput,
|
|
||||||
).Return(tt.aliasResponse, tt.aliasError)
|
|
||||||
|
|
||||||
// Call the function under test
|
|
||||||
err := h.constructAccountAlias(ctx)
|
|
||||||
|
|
||||||
// Assertions
|
|
||||||
if tt.expectedError != nil {
|
|
||||||
assert.EqualError(t, err, tt.expectedError.Error())
|
|
||||||
} else {
|
|
||||||
assert.NoError(t, err)
|
|
||||||
if tt.expectedAlias != "" {
|
|
||||||
storedAlias, err := store.ReadEntry(ctx, sessionId, storedb.DATA_ACCOUNT_ALIAS)
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Equal(t, tt.expectedAlias, string(storedAlias))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Ensure mock expectations were met
|
|
||||||
mockAccountService.AssertExpectations(t)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user