aljaz/lndhub.go
1
1
Fork 0
mirror of https://github.com/getAlby/lndhub.go.git synced 2025-12-21 14:44:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Check with github action

Browse Source
This commit is contained in:
Lucas Rouckhout
2023-02-04 15:49:14 +01:00
parent 6f7e7456a2
commit 1b6d50a1db
1 changed files with 52 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
integration_tests/rabbitmq_test.go
View File

@@ -7,8 +7,14 @@ import (
"log"
"testing"
"github.com/getAlby/lndhub.go/controllers"
"github.com/getAlby/lndhub.go/db/models"
"github.com/getAlby/lndhub.go/lib"
"github.com/getAlby/lndhub.go/lib/responses"
"github.com/getAlby/lndhub.go/lib/service"
"github.com/getAlby/lndhub.go/lib/tokens"
"github.com/go-playground/validator/v10"
"github.com/labstack/echo/v4"
amqp "github.com/rabbitmq/amqp091-go"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/suite"
@@ -20,6 +26,7 @@ type RabbitMQTestSuite struct {
invoiceUpdateSubCancelFn context.CancelFunc
userToken string
svc *service.LndhubService
testQueueName string
}
func (suite *RabbitMQTestSuite) SetupSuite() {
@@ -30,72 +37,90 @@ func (suite *RabbitMQTestSuite) SetupSuite() {
}
suite.mlnd = mlnd
suite.testQueueName = "test_invoice"
_, userTokens, err := createUsers(svc, 1)
if err != nil {
log.Fatalf("Error creating test users: %v", err)
log.Fatalf("error creating test users: %v", err)
}
suite.userToken = userTokens[0]
ctx, cancel := context.WithCancel(context.Background())
suite.invoiceUpdateSubCancelFn = cancel
go svc.InvoiceUpdateSubscription(ctx)
suite.svc = svc
e := echo.New()
e.HTTPErrorHandler = responses.HTTPErrorHandler
e.Validator = &lib.CustomValidator{Validator: validator.New()}
suite.echo = e
suite.echo.Use(tokens.Middleware(suite.svc.Config.JWTSecret))
suite.echo.POST("/addinvoice", controllers.NewAddInvoiceController(suite.svc).AddInvoice)
go svc.StartRabbitMqPublisher(ctx)
}
func (suite *RabbitMQTestSuite) TestPublishInvoice(t *testing.T) {
// create incoming invoice and fund account
invoice := suite.createAddInvoiceReq(1000, "integration test webhook", suite.userToken)
err := suite.mlnd.mockPaidInvoice(invoice, 0, false, nil)
assert.NoError(suite.T(), err)
// Check consume from rabbit
func (suite *RabbitMQTestSuite) TestPublishInvoice() {
conn, err := amqp.Dial(suite.svc.Config.RabbitMQUri)
if err != nil {
t.Error(err)
}
assert.NoError(suite.T(), err)
defer conn.Close()
ch, err := conn.Channel()
if err != nil {
t.Error(err)
}
assert.NoError(suite.T(), err)
defer ch.Close()
q, err := ch.QueueDeclare(
"test_invoice",
false,
suite.testQueueName,
true,
false,
false,
false,
nil,
)
if err != nil {
t.Error(err)
}
assert.NoError(suite.T(), err)
err = ch.QueueBind(q.Name, "#", suite.svc.Config.RabbitMQInvoiceExchange, false, nil)
assert.NoError(suite.T(), err)
invoice := suite.createAddInvoiceReq(1000, "integration test rabbitmq", suite.userToken)
err = suite.mlnd.mockPaidInvoice(invoice, 0, false, nil)
assert.NoError(suite.T(), err)
m, err := ch.Consume(
q.Name,
"",
false,
true,
false,
false,
false,
nil,
)
if err != nil {
t.Error(err)
}
assert.NoError(suite.T(), err)
t.Log("Blocking on message channel")
msg := <-m
t.Logf("%s/n", string(msg.Body))
var recievedInvoice models.Invoice
r := bytes.NewReader(msg.Body)
json.NewDecoder(r).Decode(&recievedInvoice)
err = json.NewDecoder(r).Decode(&recievedInvoice)
assert.NoError(suite.T(), err)
assert.Equal(t, invoice.RHash, recievedInvoice.RHash)
assert.Equal(suite.T(), invoice.RHash, recievedInvoice.RHash)
}
func (suite *RabbitMQTestSuite) TearDownSuite() {
conn, err := amqp.Dial(suite.svc.Config.RabbitMQUri)
assert.NoError(suite.T(), err)
defer conn.Close()
ch, err := conn.Channel()
assert.NoError(suite.T(), err)
defer ch.Close()
_, err = ch.QueueDelete(suite.testQueueName, false, false, false)
assert.NoError(suite.T(), err)
err = ch.ExchangeDelete(suite.svc.Config.RabbitMQInvoiceExchange, true, false)
assert.NoError(suite.T(), err)
}
func TestRabbitMQTestSuite(t *testing.T) {