alby-hub/nip47/controllers/create_connection_controller_test.go
Roland 3e1d16d423
Feat: NWA auth for self-hosted hubs (#1016)
* feat: nwc create_connection command (WIP)

* feat: allow creating superuser apps from the ui

* fix: pass methods rather than scopes in create_connection method

* chore: add extra tests

* fix: use browser router in http mode

* fix: update links to not use hash router

* fix: add redirect from hash router url

* fix: return nostrWalletConnectUrl in nwc connection success event and message

* feat: publish nwa event

* chore: use nwc info event instead of nwa event

* feat: create custom alby go detail page

* chore: allow creating/editing apps with the same name

* fix: convert budget from msats to sats (#1111)

* chore: address NWA feedback

* chore: address feedback

- adjust button copy
- make create_connection methods consistent with http deeplink flow
- remove unused constant
- add empty string check before adding lud16 tag
- fix test

* feat: add support for notification_types in create_connection method

* fix: shorter button copy

* fix: do not include lud16 tag in published info event

* Feat: add lud16 to get_info response (#1128)

feat: add lud16 to get_info response

* chore: minor alby go screen improvements

* fix: incorrect unlock password error message to create app with superuser access

* chore: minor ui improvements on alby go detail page

* chore: avoid duplicate app names by adding a suffix

* chore: move scopes check to apps service, add new tests, DRY controller test setup

* fix: scopes component full access scopes and isolated scope group check

* fix: use supported capabilities for Alby Go

* fix: return correct app name

* chore: address minor comments

---------

Co-authored-by: im-adithya <imadithyavardhan@gmail.com>
2025-02-27 19:12:55 +07:00

363 lines
11 KiB
Go

package controllers
import (
"context"
"encoding/json"
"fmt"
"testing"
"github.com/nbd-wtf/go-nostr"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/getAlby/hub/apps"
"github.com/getAlby/hub/config"
"github.com/getAlby/hub/constants"
"github.com/getAlby/hub/db"
"github.com/getAlby/hub/nip47/models"
"github.com/getAlby/hub/tests"
)
func TestHandleCreateConnectionEvent(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
svc.Cfg.SetUpdate("LNBackendType", config.LDKBackendType, "")
require.NoError(t, err)
defer svc.Remove()
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123",
"request_methods": ["get_info", "pay_invoice"],
"notification_types": ["payment_received"],
"max_amount": 100000000,
"budget_renewal": "monthly",
"isolated": true
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.Nil(t, publishedResponse.Error)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
createAppResult := publishedResponse.Result.(createConnectionResponse)
assert.NotNil(t, createAppResult.WalletPubkey)
app := db.App{}
err = svc.DB.First(&app).Error
assert.NoError(t, err)
assert.Equal(t, pairingPublicKey, app.AppPubkey)
assert.Equal(t, createAppResult.WalletPubkey, *app.WalletPubkey)
permissions := []db.AppPermission{}
err = svc.DB.Find(&permissions).Error
assert.NoError(t, err)
assert.Equal(t, 3, len(permissions))
assert.Equal(t, constants.GET_INFO_SCOPE, permissions[0].Scope)
assert.Equal(t, constants.PAY_INVOICE_SCOPE, permissions[1].Scope)
assert.Equal(t, constants.NOTIFICATIONS_SCOPE, permissions[2].Scope)
assert.True(t, app.Isolated)
assert.Equal(t, 100_000, permissions[1].MaxAmountSat)
assert.Equal(t, constants.BUDGET_RENEWAL_MONTHLY, permissions[1].BudgetRenewal)
}
func TestHandleCreateConnectionEvent_IsolatedUnsupportedBackendType(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
require.NoError(t, err)
defer svc.Remove()
svc.Cfg.SetUpdate("BackendType", config.CashuBackendType, "")
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123",
"request_methods": ["get_info", "pay_invoice"],
"notification_types": ["payment_received"],
"max_amount": 100000000,
"budget_renewal": "monthly",
"isolated": true
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.NotNil(t, publishedResponse.Error)
assert.Equal(t, constants.ERROR_INTERNAL, publishedResponse.Error.Code)
assert.Equal(t, "sub-wallets are currently not supported on your node backend. Try LDK or LND", publishedResponse.Error.Message)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
}
func TestHandleCreateConnectionEvent_PubkeyAlreadyExists(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
require.NoError(t, err)
defer svc.Remove()
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
appsSvc := apps.NewAppsService(svc.DB, svc.EventPublisher, svc.Keys, svc.Cfg)
_, _, err = appsSvc.CreateApp("Existing App", pairingPublicKey, 0, constants.BUDGET_RENEWAL_NEVER, nil, []string{models.GET_INFO_METHOD}, false, nil)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123",
"request_methods": ["get_info"]
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.NotNil(t, publishedResponse.Error)
assert.Equal(t, constants.ERROR_INTERNAL, publishedResponse.Error.Code)
assert.Equal(t, "duplicated key not allowed", publishedResponse.Error.Message)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
assert.Nil(t, publishedResponse.Result)
}
func TestHandleCreateConnectionEvent_NoMethods(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
require.NoError(t, err)
defer svc.Remove()
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123"
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.NotNil(t, publishedResponse.Error)
assert.Equal(t, constants.ERROR_INTERNAL, publishedResponse.Error.Code)
assert.Equal(t, "No request methods provided", publishedResponse.Error.Message)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
assert.Nil(t, publishedResponse.Result)
}
func TestHandleCreateConnectionEvent_UnsupportedMethod(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
require.NoError(t, err)
defer svc.Remove()
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123",
"request_methods": ["non_existent"]
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.NotNil(t, publishedResponse.Error)
assert.Equal(t, constants.ERROR_INTERNAL, publishedResponse.Error.Code)
assert.Equal(t, "One or more methods are not supported by the current LNClient", publishedResponse.Error.Message)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
assert.Nil(t, publishedResponse.Result)
}
func TestHandleCreateConnectionEvent_UnsupportedNotificationType(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
require.NoError(t, err)
defer svc.Remove()
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123",
"request_methods": ["get_info"],
"notification_types": ["non_existent"]
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.NotNil(t, publishedResponse.Error)
assert.Equal(t, constants.ERROR_INTERNAL, publishedResponse.Error.Code)
assert.Equal(t, "One or more notification types are not supported by the current LNClient", publishedResponse.Error.Message)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
assert.Nil(t, publishedResponse.Result)
}
func TestHandleCreateConnectionEvent_DoNotAllowCreateConnectionMethod(t *testing.T) {
ctx := context.TODO()
svc, err := tests.CreateTestService(t)
require.NoError(t, err)
defer svc.Remove()
pairingSecretKey := nostr.GeneratePrivateKey()
pairingPublicKey, err := nostr.GetPublicKey(pairingSecretKey)
require.NoError(t, err)
nip47CreateConnectionJson := fmt.Sprintf(`
{
"method": "create_connection",
"params": {
"pubkey": "%s",
"name": "Test 123",
"request_methods": ["create_connection"]
}
}
`, pairingPublicKey)
nip47Request := &models.Request{}
err = json.Unmarshal([]byte(nip47CreateConnectionJson), nip47Request)
assert.NoError(t, err)
dbRequestEvent := &db.RequestEvent{}
err = svc.DB.Create(&dbRequestEvent).Error
assert.NoError(t, err)
var publishedResponse *models.Response
publishResponse := func(response *models.Response, tags nostr.Tags) {
publishedResponse = response
}
NewTestNip47Controller(svc).
HandleCreateConnectionEvent(ctx, nip47Request, dbRequestEvent.ID, publishResponse)
assert.NotNil(t, publishedResponse.Error)
assert.Equal(t, constants.ERROR_INTERNAL, publishedResponse.Error.Code)
assert.Equal(t, "cannot create a new app that has create_connection permission via NWC", publishedResponse.Error.Message)
assert.Equal(t, models.CREATE_CONNECTION_METHOD, publishedResponse.ResultType)
assert.Nil(t, publishedResponse.Result)
}