import 'package:omemo_dart/omemo_dart.dart'; import 'package:test/test.dart'; void main() { test('Test serialising and deserialising the Device', () async { // Generate a random session final oldSession = await OmemoSessionManager.generateNewIdentity(opkAmount: 1); final oldDevice = await oldSession.getDevice(); final serialised = await oldDevice.toJson(); final newDevice = Device.fromJson(serialised); expect(oldDevice.id, newDevice.id); expect(await oldDevice.ik.equals(newDevice.ik), true); expect(await oldDevice.spk.equals(newDevice.spk), true); expect(listsEqual(oldDevice.spkSignature, newDevice.spkSignature), true); expect(oldDevice.spkId, newDevice.spkId); // Check the Ontime-Prekeys expect(oldDevice.opks.length, newDevice.opks.length); for (final entry in oldDevice.opks.entries) { expect(await newDevice.opks[entry.key]!.equals(entry.value), true); } }); test('Test serialising and deserialising the OmemoDoubleRatchet', () async { // Generate a random ratchet const aliceJid = 'alice@server.example'; const bobJid = 'bob@other.server.example'; final aliceSession = await OmemoSessionManager.generateNewIdentity(opkAmount: 1); final bobSession = await OmemoSessionManager.generateNewIdentity(opkAmount: 1); final aliceMessage = await aliceSession.encryptToJid( bobJid, 'Hello Bob!', newSessions: [ await (await bobSession.getDevice()).toBundle(), ], ); await bobSession.decryptMessage( aliceMessage.ciphertext, aliceJid, (await aliceSession.getDevice()).id, aliceMessage.encryptedKeys, ); final aliceOld = aliceSession.getRatchet((await bobSession.getDevice()).id); final aliceSerialised = await aliceOld.toJson(); final aliceNew = OmemoDoubleRatchet.fromJson(aliceSerialised); expect(await aliceOld.dhs.equals(aliceNew.dhs), true); if (aliceOld.dhr == null) { expect(aliceNew.dhr, null); } else { expect(await aliceOld.dhr!.equals(aliceNew.dhr!), true); } expect(listsEqual(aliceOld.rk, aliceNew.rk), true); if (aliceOld.cks == null) { expect(aliceNew.cks, null); } else { expect(listsEqual(aliceOld.cks!, aliceNew.cks!), true); } if (aliceOld.ckr == null) { expect(aliceNew.ckr, null); } else { expect(listsEqual(aliceOld.ckr!, aliceNew.ckr!), true); } expect(aliceOld.ns, aliceNew.ns); expect(aliceOld.nr, aliceNew.nr); expect(aliceOld.pn, aliceNew.pn); expect(listsEqual(aliceOld.sessionAd, aliceNew.sessionAd), true); }); }