matrix-dart-sdk/test/matrix_database_test.dart

/*
 *   Famedly Matrix SDK
 *   Copyright (C) 2020 Famedly GmbH
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU Affero General Public License as
 *   published by the Free Software Foundation, either version 3 of the
 *   License, or (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *   GNU Affero General Public License for more details.
 *
 *   You should have received a copy of the GNU Affero General Public License
 *   along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 */

import 'package:matrix/matrix.dart';

import 'package:test/test.dart';
import 'fake_database.dart';

void main() {
  group('Databse', () {
    Logs().level = Level.error;
    final room = Room(id: '!room:blubb', client: Client('testclient'));
    test('setupDatabase', () async {
      final database = await getDatabase(null);
      await database.insertClient(
        'testclient',
        'https://example.org',
        'blubb',
        '@test:example.org',
        null,
        null,
        null,
        null,
      );
    });

    test('storeEventUpdate', () async {
      final client = Client('testclient');
      final database = await getDatabase(client);
      // store a simple update
      var update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': <String, dynamic>{'blah': 'blubb'},
          'event_id': '\$event-1',
          'sender': '@blah:blubb',
        },
      );
      await database.storeEventUpdate(update, client);
      var event = await database.getEventById('\$event-1', room);
      expect(event?.eventId, '\$event-1');

      // insert a transaction id
      update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': <String, dynamic>{'blah': 'blubb'},
          'event_id': 'transaction-1',
          'sender': '@blah:blubb',
          'status': EventStatus.sending.intValue,
        },
      );
      await database.storeEventUpdate(update, client);
      event = await database.getEventById('transaction-1', room);
      expect(event?.eventId, 'transaction-1');
      update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': <String, dynamic>{'blah': 'blubb'},
          'event_id': '\$event-2',
          'sender': '@blah:blubb',
          'unsigned': <String, dynamic>{
            'transaction_id': 'transaction-1',
          },
          'status': EventStatus.sent.intValue,
        },
      );
      await database.storeEventUpdate(update, client);
      event = await database.getEventById('transaction-1', room);
      expect(event, null);
      event = await database.getEventById('\$event-2', room);

      // insert a transaction id if the event id for it already exists
      update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': {'blah': 'blubb'},
          'event_id': '\$event-3',
          'sender': '@blah:blubb',
          'status': EventStatus.sending.intValue,
        },
      );
      await database.storeEventUpdate(update, client);
      event = await database.getEventById('\$event-3', room);
      expect(event?.eventId, '\$event-3');
      update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': {'blah': 'blubb'},
          'event_id': '\$event-3',
          'sender': '@blah:blubb',
          'status': EventStatus.sent.intValue,
          'unsigned': <String, dynamic>{
            'transaction_id': 'transaction-2',
          },
        },
      );
      await database.storeEventUpdate(update, client);
      event = await database.getEventById('\$event-3', room);
      expect(event?.eventId, '\$event-3');
      expect(event?.status, EventStatus.sent);
      event = await database.getEventById('transaction-2', room);
      expect(event, null);

      // insert transaction id and not update status
      update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': {'blah': 'blubb'},
          'event_id': '\$event-4',
          'sender': '@blah:blubb',
          'status': EventStatus.synced.intValue,
        },
      );
      await database.storeEventUpdate(update, client);
      event = await database.getEventById('\$event-4', room);
      expect(event?.eventId, '\$event-4');
      update = EventUpdate(
        type: EventUpdateType.timeline,
        roomID: room.id,
        content: {
          'type': 'm.room.message',
          'origin_server_ts': 100,
          'content': {'blah': 'blubb'},
          'event_id': '\$event-4',
          'sender': '@blah:blubb',
          'status': EventStatus.sent.intValue,
          'unsigned': <String, dynamic>{
            'transaction_id': 'transaction-3',
          },
        },
      );
      await database.storeEventUpdate(update, client);
      event = await database.getEventById('\$event-4', room);
      expect(event?.eventId, '\$event-4');
      expect(event?.status, EventStatus.synced);
      event = await database.getEventById('transaction-3', room);
      expect(event, null);
    });
  });
}