mirror of
https://github.com/matrix-org/dendrite.git
synced 2024-11-27 00:31:55 -06:00
221 lines
7.6 KiB
Go
221 lines
7.6 KiB
Go
// Copyright 2022 The Matrix.org Foundation C.I.C.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package routing_test
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/matrix-org/dendrite/internal/sqlutil"
|
|
"github.com/matrix-org/dendrite/relayapi/internal"
|
|
"github.com/matrix-org/dendrite/relayapi/routing"
|
|
"github.com/matrix-org/dendrite/relayapi/storage/shared"
|
|
"github.com/matrix-org/dendrite/test"
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func createQuery(
|
|
userID gomatrixserverlib.UserID,
|
|
prevEntry gomatrixserverlib.RelayEntry,
|
|
) gomatrixserverlib.FederationRequest {
|
|
var federationPathPrefixV1 = "/_matrix/federation/v1"
|
|
path := federationPathPrefixV1 + "/relay_txn/" + userID.Raw()
|
|
request := gomatrixserverlib.NewFederationRequest("GET", userID.Domain(), "relay", path)
|
|
request.SetContent(prevEntry)
|
|
|
|
return request
|
|
}
|
|
|
|
func TestGetEmptyDatabaseReturnsNothing(t *testing.T) {
|
|
testDB := test.NewInMemoryRelayDatabase()
|
|
db := shared.Database{
|
|
Writer: sqlutil.NewDummyWriter(),
|
|
RelayQueue: testDB,
|
|
RelayQueueJSON: testDB,
|
|
}
|
|
httpReq := &http.Request{}
|
|
userID, err := gomatrixserverlib.NewUserID("@local:domain", false)
|
|
assert.NoError(t, err, "Invalid userID")
|
|
|
|
transaction := createTransaction()
|
|
|
|
_, err = db.StoreTransaction(context.Background(), transaction)
|
|
assert.NoError(t, err, "Failed to store transaction")
|
|
|
|
relayAPI := internal.NewRelayInternalAPI(
|
|
&db, nil, nil, nil, nil, false, "", true,
|
|
)
|
|
|
|
request := createQuery(*userID, gomatrixserverlib.RelayEntry{})
|
|
response := routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusOK, response.Code)
|
|
|
|
jsonResponse := response.JSON.(gomatrixserverlib.RespGetRelayTransaction)
|
|
assert.Equal(t, false, jsonResponse.EntriesQueued)
|
|
assert.Equal(t, gomatrixserverlib.Transaction{}, jsonResponse.Transaction)
|
|
|
|
count, err := db.GetTransactionCount(context.Background(), *userID)
|
|
assert.NoError(t, err)
|
|
assert.Zero(t, count)
|
|
}
|
|
|
|
func TestGetInvalidPrevEntryFails(t *testing.T) {
|
|
testDB := test.NewInMemoryRelayDatabase()
|
|
db := shared.Database{
|
|
Writer: sqlutil.NewDummyWriter(),
|
|
RelayQueue: testDB,
|
|
RelayQueueJSON: testDB,
|
|
}
|
|
httpReq := &http.Request{}
|
|
userID, err := gomatrixserverlib.NewUserID("@local:domain", false)
|
|
assert.NoError(t, err, "Invalid userID")
|
|
|
|
transaction := createTransaction()
|
|
|
|
_, err = db.StoreTransaction(context.Background(), transaction)
|
|
assert.NoError(t, err, "Failed to store transaction")
|
|
|
|
relayAPI := internal.NewRelayInternalAPI(
|
|
&db, nil, nil, nil, nil, false, "", true,
|
|
)
|
|
|
|
request := createQuery(*userID, gomatrixserverlib.RelayEntry{EntryID: -1})
|
|
response := routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusInternalServerError, response.Code)
|
|
}
|
|
|
|
func TestGetReturnsSavedTransaction(t *testing.T) {
|
|
testDB := test.NewInMemoryRelayDatabase()
|
|
db := shared.Database{
|
|
Writer: sqlutil.NewDummyWriter(),
|
|
RelayQueue: testDB,
|
|
RelayQueueJSON: testDB,
|
|
}
|
|
httpReq := &http.Request{}
|
|
userID, err := gomatrixserverlib.NewUserID("@local:domain", false)
|
|
assert.NoError(t, err, "Invalid userID")
|
|
|
|
transaction := createTransaction()
|
|
receipt, err := db.StoreTransaction(context.Background(), transaction)
|
|
assert.NoError(t, err, "Failed to store transaction")
|
|
|
|
err = db.AssociateTransactionWithDestinations(
|
|
context.Background(),
|
|
map[gomatrixserverlib.UserID]struct{}{
|
|
*userID: {},
|
|
},
|
|
transaction.TransactionID,
|
|
receipt)
|
|
assert.NoError(t, err, "Failed to associate transaction with user")
|
|
|
|
relayAPI := internal.NewRelayInternalAPI(
|
|
&db, nil, nil, nil, nil, false, "", true,
|
|
)
|
|
|
|
request := createQuery(*userID, gomatrixserverlib.RelayEntry{})
|
|
response := routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusOK, response.Code)
|
|
|
|
jsonResponse := response.JSON.(gomatrixserverlib.RespGetRelayTransaction)
|
|
assert.True(t, jsonResponse.EntriesQueued)
|
|
assert.Equal(t, transaction, jsonResponse.Transaction)
|
|
|
|
// And once more to clear the queue
|
|
request = createQuery(*userID, gomatrixserverlib.RelayEntry{EntryID: jsonResponse.EntryID})
|
|
response = routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusOK, response.Code)
|
|
|
|
jsonResponse = response.JSON.(gomatrixserverlib.RespGetRelayTransaction)
|
|
assert.False(t, jsonResponse.EntriesQueued)
|
|
assert.Equal(t, gomatrixserverlib.Transaction{}, jsonResponse.Transaction)
|
|
|
|
count, err := db.GetTransactionCount(context.Background(), *userID)
|
|
assert.NoError(t, err)
|
|
assert.Zero(t, count)
|
|
}
|
|
|
|
func TestGetReturnsMultipleSavedTransactions(t *testing.T) {
|
|
testDB := test.NewInMemoryRelayDatabase()
|
|
db := shared.Database{
|
|
Writer: sqlutil.NewDummyWriter(),
|
|
RelayQueue: testDB,
|
|
RelayQueueJSON: testDB,
|
|
}
|
|
httpReq := &http.Request{}
|
|
userID, err := gomatrixserverlib.NewUserID("@local:domain", false)
|
|
assert.NoError(t, err, "Invalid userID")
|
|
|
|
transaction := createTransaction()
|
|
receipt, err := db.StoreTransaction(context.Background(), transaction)
|
|
assert.NoError(t, err, "Failed to store transaction")
|
|
|
|
err = db.AssociateTransactionWithDestinations(
|
|
context.Background(),
|
|
map[gomatrixserverlib.UserID]struct{}{
|
|
*userID: {},
|
|
},
|
|
transaction.TransactionID,
|
|
receipt)
|
|
assert.NoError(t, err, "Failed to associate transaction with user")
|
|
|
|
transaction2 := createTransaction()
|
|
receipt2, err := db.StoreTransaction(context.Background(), transaction2)
|
|
assert.NoError(t, err, "Failed to store transaction")
|
|
|
|
err = db.AssociateTransactionWithDestinations(
|
|
context.Background(),
|
|
map[gomatrixserverlib.UserID]struct{}{
|
|
*userID: {},
|
|
},
|
|
transaction2.TransactionID,
|
|
receipt2)
|
|
assert.NoError(t, err, "Failed to associate transaction with user")
|
|
|
|
relayAPI := internal.NewRelayInternalAPI(
|
|
&db, nil, nil, nil, nil, false, "", true,
|
|
)
|
|
|
|
request := createQuery(*userID, gomatrixserverlib.RelayEntry{})
|
|
response := routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusOK, response.Code)
|
|
|
|
jsonResponse := response.JSON.(gomatrixserverlib.RespGetRelayTransaction)
|
|
assert.True(t, jsonResponse.EntriesQueued)
|
|
assert.Equal(t, transaction, jsonResponse.Transaction)
|
|
|
|
request = createQuery(*userID, gomatrixserverlib.RelayEntry{EntryID: jsonResponse.EntryID})
|
|
response = routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusOK, response.Code)
|
|
|
|
jsonResponse = response.JSON.(gomatrixserverlib.RespGetRelayTransaction)
|
|
assert.True(t, jsonResponse.EntriesQueued)
|
|
assert.Equal(t, transaction2, jsonResponse.Transaction)
|
|
|
|
// And once more to clear the queue
|
|
request = createQuery(*userID, gomatrixserverlib.RelayEntry{EntryID: jsonResponse.EntryID})
|
|
response = routing.GetTransactionFromRelay(httpReq, &request, relayAPI, *userID)
|
|
assert.Equal(t, http.StatusOK, response.Code)
|
|
|
|
jsonResponse = response.JSON.(gomatrixserverlib.RespGetRelayTransaction)
|
|
assert.False(t, jsonResponse.EntriesQueued)
|
|
assert.Equal(t, gomatrixserverlib.Transaction{}, jsonResponse.Transaction)
|
|
|
|
count, err := db.GetTransactionCount(context.Background(), *userID)
|
|
assert.NoError(t, err)
|
|
assert.Zero(t, count)
|
|
}
|