|
3 | 3 | //
|
4 | 4 |
|
5 | 5 | #import <AxolotlKit/AxolotlExceptions.h>
|
6 |
| -#import <AxolotlKit/AxolotlInMemoryStore.h> |
7 | 6 | #import <AxolotlKit/NSData+keyVersionByte.h>
|
| 7 | +#import <AxolotlKit/SPKMockProtocolStore.h> |
8 | 8 | #import <AxolotlKit/SessionBuilder.h>
|
9 | 9 | #import <AxolotlKit/SessionCipher.h>
|
10 | 10 | #import <Curve25519Kit/Ed25519.h>
|
@@ -47,10 +47,10 @@ - (void)testBasicPreKey {
|
47 | 47 | NSString *BOB_RECIPIENT_ID = @"+3828923892";
|
48 | 48 | NSString *ALICE_RECIPIENT_ID = @"alice@gmail.com";
|
49 | 49 |
|
50 |
| - AxolotlInMemoryStore *aliceStore = [AxolotlInMemoryStore new]; |
| 50 | + SPKMockProtocolStore *aliceStore = [SPKMockProtocolStore new]; |
51 | 51 | SessionBuilder *aliceSessionBuilder = [[SessionBuilder alloc] initWithAxolotlStore:aliceStore recipientId:BOB_RECIPIENT_ID deviceId:1];
|
52 | 52 |
|
53 |
| - AxolotlInMemoryStore *bobStore = [AxolotlInMemoryStore new]; |
| 53 | + SPKMockProtocolStore *bobStore = [SPKMockProtocolStore new]; |
54 | 54 | ECKeyPair *bobPreKeyPair = [Curve25519 generateKeyPair];
|
55 | 55 | ECKeyPair *bobSignedPreKeyPair = [Curve25519 generateKeyPair];
|
56 | 56 | NSData *bobSignedPreKeySignature =
|
@@ -99,10 +99,10 @@ - (void)testBasicPreKeyMITM {
|
99 | 99 |
|
100 | 100 | NSString *BOB_RECIPIENT_ID = @"+3828923892";
|
101 | 101 |
|
102 |
| - AxolotlInMemoryStore *aliceStore = [AxolotlInMemoryStore new]; |
| 102 | + SPKMockProtocolStore *aliceStore = [SPKMockProtocolStore new]; |
103 | 103 | SessionBuilder *aliceSessionBuilder = [[SessionBuilder alloc] initWithAxolotlStore:aliceStore recipientId:BOB_RECIPIENT_ID deviceId:1];
|
104 | 104 |
|
105 |
| - AxolotlInMemoryStore *bobStore = [AxolotlInMemoryStore new]; |
| 105 | + SPKMockProtocolStore *bobStore = [SPKMockProtocolStore new]; |
106 | 106 | ECKeyPair *bobIdentityKeyPair1 = [Curve25519 generateKeyPair];
|
107 | 107 | ECKeyPair *bobPreKeyPair1 = [Curve25519 generateKeyPair];
|
108 | 108 | ECKeyPair *bobSignedPreKeyPair1 = [Curve25519 generateKeyPair];
|
|
0 commit comments