mirror of
https://github.com/matrix-org/dendrite.git
synced 2025-12-16 11:23:11 -06:00
131 lines
2.9 KiB
Go
131 lines
2.9 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"github.com/matrix-org/dendrite/roomserver/api"
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
var dataSource string
|
|
var insideCi = false
|
|
var insideDocker = false
|
|
|
|
func init() {
|
|
for _, val := range os.Environ() {
|
|
tokens := strings.Split(val, "=")
|
|
if tokens[0] == "CI" && tokens[1] == "true" {
|
|
insideCi = true
|
|
}
|
|
}
|
|
if !insideCi {
|
|
if _, err := os.Open("/.dockerenv"); err == nil {
|
|
insideDocker = true
|
|
}
|
|
}
|
|
|
|
if insideCi {
|
|
dataSource = "postgres://postgres@localhost/dendrite_syncapi?sslmode=disable"
|
|
} else if insideDocker {
|
|
dataSource = "postgres://dendrite:itsasecret@postgres/dendrite_syncapi?sslmode=disable"
|
|
} else {
|
|
dataSource = "postgres://dendrite:itsasecret@localhost:15432/dendrite_syncapi?sslmode=disable"
|
|
}
|
|
}
|
|
|
|
const testEventID = "$test-event-id:test-domain.example.com"
|
|
|
|
func Test_sanityCheckOutputRoomEvents(t *testing.T) {
|
|
db, err := NewSyncServerDatabase(dataSource)
|
|
assert.Nil(t, err)
|
|
|
|
err = db.events.prepare(db.db)
|
|
assert.Nil(t, err)
|
|
|
|
truncateTable(t, db)
|
|
insertTestEvent(t, db)
|
|
selectTestEvent(t, db)
|
|
truncateTable(t, db)
|
|
}
|
|
|
|
func TestSyncServerDatabase_selectEventsWithEventIDs(t *testing.T) {
|
|
db, err := NewSyncServerDatabase(dataSource)
|
|
assert.Nil(t, err)
|
|
insertTestEvent(t, db)
|
|
ctx := context.Background()
|
|
txn, err := db.db.Begin()
|
|
assert.Nil(t, err)
|
|
|
|
var eventIDs = []string{testEventID}
|
|
events, err := db.fetchMissingStateEvents(ctx, txn, eventIDs)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, events)
|
|
assert.Condition(t, func() bool {
|
|
return len(events) > 0
|
|
})
|
|
}
|
|
|
|
func insertTestEvent(t *testing.T, db *SyncServerDatabase) {
|
|
txn, err := db.db.Begin()
|
|
assert.Nil(t, err)
|
|
|
|
keyBytes := []byte("1122334455667788112233445566778811223344556677881122334455667788")
|
|
eventBuilder := gomatrixserverlib.EventBuilder{
|
|
RoomID: "!test_room_id:test-domain.example.com",
|
|
Content: []byte(`{"RawContent": "test-raw-content"}`),
|
|
Sender: "@test-user:test-domain.example.com",
|
|
}
|
|
event, err := eventBuilder.Build(
|
|
testEventID,
|
|
time.Now(),
|
|
"test-domain.example.com",
|
|
"test-key-id",
|
|
keyBytes)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
var addState, removeState []string
|
|
transactionID := api.TransactionID{
|
|
DeviceID: "test-device-id",
|
|
TransactionID: "test-transaction-id",
|
|
}
|
|
|
|
newEventID, err := db.events.insertEvent(
|
|
context.Background(),
|
|
txn,
|
|
&event,
|
|
addState,
|
|
removeState,
|
|
&transactionID)
|
|
|
|
assert.Nil(t, err)
|
|
err = txn.Commit()
|
|
assert.Nil(t, err)
|
|
|
|
assert.Condition(t, func() bool {
|
|
return newEventID > 0
|
|
})
|
|
}
|
|
|
|
func selectTestEvent(t *testing.T, db *SyncServerDatabase) {
|
|
ctx := context.Background()
|
|
|
|
var eventIDs = []string{testEventID}
|
|
res, err := db.Events(ctx, eventIDs)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, res)
|
|
|
|
assert.Condition(t, func() bool {
|
|
return len(res) > 0
|
|
})
|
|
}
|
|
|
|
func truncateTable(t *testing.T, db *SyncServerDatabase) {
|
|
_, err := db.db.Exec("TRUNCATE syncapi_output_room_events")
|
|
assert.Nil(t, err)
|
|
}
|