71 lines
2.6 KiB
Dart
71 lines
2.6 KiB
Dart
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);
|
|
});
|
|
}
|