pEpObjCTests/PEPSessionTest.m
author Dirk Zimmermann <dirk@pep-project.org>
Fri, 09 Feb 2018 15:01:51 +0100
changeset 441 b951138b7dd1
parent 440 d8cf857a5fc2
child 443 10ec442ed3ad
permissions -rw-r--r--
IOS-937 trying to make testOutgoingColors more stable
andreas@392
     1
//
andreas@392
     2
//  PEPSessionTest.m
andreas@392
     3
//  pEpObjCAdapterTests
andreas@392
     4
//
andreas@392
     5
//  Created by Andreas Buff on 18.01.18.
andreas@392
     6
//  Copyright © 2018 p≡p. All rights reserved.
andreas@392
     7
//
andreas@392
     8
andreas@392
     9
#import <XCTest/XCTest.h>
andreas@392
    10
andreas@392
    11
#import "PEPObjCAdapter.h"
andreas@392
    12
#import "NSDictionary+Extension.h"
andreas@392
    13
#import "PEPIdentity.h"
andreas@392
    14
#import "PEPMessage.h"
andreas@392
    15
andreas@392
    16
#import "PEPTestUtils.h"
andreas@392
    17
#import "PEPTestSyncDelegate.h"
andreas@392
    18
andreas@392
    19
@interface PEPSessionTest : XCTestCase
andreas@392
    20
@end
andreas@392
    21
andreas@392
    22
@implementation PEPSessionTest
andreas@392
    23
andreas@392
    24
- (void)setUp
andreas@392
    25
{
andreas@392
    26
    [super setUp];
andreas@416
    27
    [PEPObjCAdapter setUnecryptedSubjectEnabled:NO];
andreas@416
    28
andreas@392
    29
    [self pEpCleanUp];
andreas@392
    30
}
andreas@392
    31
andreas@392
    32
- (void)tearDown {
andreas@392
    33
    [self pEpCleanUp];
andreas@392
    34
    [super tearDown];
andreas@392
    35
}
andreas@392
    36
andreas@392
    37
- (void)testSyncSession
andreas@392
    38
{
andreas@392
    39
    PEPSession *session = [PEPSession new];
andreas@392
    40
    PEPTestSyncDelegate *syncDelegate = [[PEPTestSyncDelegate alloc] init];
andreas@392
    41
andreas@392
    42
    // This should attach session just created
andreas@392
    43
    [PEPObjCAdapter startSync:syncDelegate];
andreas@392
    44
andreas@392
    45
    PEPIdentity *identMe = [[PEPIdentity alloc]
andreas@392
    46
                            initWithAddress:@"pep.test.iosgenkey@pep-project.org"
andreas@392
    47
                            userID:@"Me"
andreas@392
    48
                            userName:@"pEp Test iOS GenKey"
andreas@392
    49
                            isOwn:YES];
andreas@392
    50
andreas@392
    51
    [session mySelf:identMe];
andreas@392
    52
andreas@392
    53
    bool res = [syncDelegate waitUntilSent:1];
andreas@392
    54
andreas@392
    55
    // Can't currently work, engine doesn't contain sync.
andreas@392
    56
    XCTAssertFalse(res);
andreas@392
    57
andreas@392
    58
    // This should detach session just created
andreas@392
    59
    [PEPObjCAdapter stopSync];
andreas@392
    60
}
andreas@392
    61
andreas@392
    62
- (void)testTrustWords
andreas@392
    63
{
andreas@392
    64
    PEPSession *session = [PEPSession new];
andreas@392
    65
andreas@392
    66
    NSArray *trustwords = [session trustwords:@"DB47DB47DB47DB47DB47DB47DB47DB47DB47DB47"
andreas@392
    67
                                  forLanguage:@"en" shortened:false];
andreas@392
    68
    XCTAssertEqual([trustwords count], 10);
andreas@392
    69
andreas@392
    70
    for(id word in trustwords)
andreas@392
    71
        XCTAssertEqualObjects(word, @"BAPTISMAL");
andreas@392
    72
}
andreas@392
    73
andreas@392
    74
- (void)testGenKey
andreas@392
    75
{
andreas@392
    76
    PEPSession *session = [PEPSession new];
andreas@392
    77
andreas@392
    78
    PEPIdentity *identMe = [[PEPIdentity alloc]
andreas@392
    79
                            initWithAddress:@"pep.test.iosgenkey@pep-project.org"
andreas@392
    80
                            userID:@"Me"
andreas@392
    81
                            userName:@"pEp Test iOS GenKey"
andreas@392
    82
                            isOwn:YES];
andreas@392
    83
andreas@392
    84
    [session mySelf:identMe];
andreas@392
    85
andreas@392
    86
    XCTAssertNotNil(identMe.fingerPrint);
andreas@392
    87
    XCTAssertNotEqual(identMe.commType, PEP_ct_unknown);
andreas@392
    88
andreas@392
    89
    // check that the comm type is not a PGP one
andreas@392
    90
    XCTAssertFalse([identMe containsPGPCommType]);
andreas@392
    91
}
andreas@392
    92
andreas@392
    93
- (void)testMySelfCommType
andreas@392
    94
{
andreas@392
    95
    PEPSession *session = [PEPSession new];
andreas@392
    96
andreas@392
    97
    PEPIdentity *identMe = [[PEPIdentity alloc]
andreas@392
    98
                            initWithAddress:@"pep.test.iosgenkey@pep-project.org"
andreas@392
    99
                            userID:@"Me"
andreas@392
   100
                            userName:@"pEp Test iOS GenKey"
andreas@392
   101
                            isOwn:YES];
andreas@392
   102
andreas@392
   103
    [session mySelf:identMe];
andreas@392
   104
andreas@392
   105
    XCTAssertNotNil(identMe.fingerPrint);
andreas@392
   106
    XCTAssertNotEqual(identMe.commType, PEP_ct_unknown);
andreas@392
   107
andreas@392
   108
    // check that the comm type is not a PGP one
andreas@392
   109
    XCTAssertFalse([identMe containsPGPCommType]);
andreas@392
   110
andreas@392
   111
    dispatch_queue_t queue = dispatch_get_global_queue(QOS_CLASS_USER_INITIATED, 0);
andreas@392
   112
    dispatch_sync(queue, ^{
andreas@392
   113
        PEPSession *session2 = [PEPSession new];
andreas@392
   114
andreas@392
   115
        // Now simulate an update from the app, which usually only caches
andreas@392
   116
        // kPepUsername, kPepAddress and optionally kPepUserID.
andreas@392
   117
        PEPIdentity *identMe2 = [[PEPIdentity alloc]
andreas@392
   118
                                 initWithAddress:identMe.address
andreas@392
   119
                                 userID:identMe.userID
andreas@392
   120
                                 userName:identMe.userName
andreas@392
   121
                                 isOwn:NO];
andreas@392
   122
        [session2 mySelf:identMe2];
andreas@392
   123
        XCTAssertNotNil(identMe2.fingerPrint);
andreas@392
   124
        XCTAssertFalse([identMe2 containsPGPCommType]);
andreas@392
   125
        XCTAssertEqualObjects(identMe2.fingerPrint, identMe.fingerPrint);
andreas@392
   126
andreas@392
   127
        // Now pretend the app only knows kPepUsername and kPepAddress
andreas@411
   128
        PEPIdentity *identMe3 = [PEPTestUtils foreignPepIdentityWithAddress:identMe.address
andreas@411
   129
                                                                   userName:identMe.userName];
andreas@392
   130
        [session2 mySelf:identMe3];
andreas@392
   131
        XCTAssertNotNil(identMe3.fingerPrint);
andreas@392
   132
        XCTAssertFalse([identMe3 containsPGPCommType]);
andreas@392
   133
        XCTAssertEqualObjects(identMe3.fingerPrint, identMe.fingerPrint);
andreas@392
   134
andreas@392
   135
        XCTAssertEqualObjects(identMe.address, identMe2.address);
andreas@392
   136
        XCTAssertEqualObjects(identMe.address, identMe3.address);
andreas@392
   137
        XCTAssertEqual(identMe.commType, identMe2.commType);
andreas@392
   138
        XCTAssertEqual(identMe.commType, identMe3.commType);
andreas@392
   139
    });
andreas@392
   140
}
andreas@392
   141
dirk@435
   142
- (void)testPartnerWithoutFingerPrint
dirk@435
   143
{
dirk@435
   144
    PEPSession *session = [PEPSession new];
dirk@435
   145
dirk@435
   146
    PEPIdentity *identRandom = [[PEPIdentity alloc]
dirk@435
   147
                                initWithAddress:@"does_not_exist@example.com"
dirk@435
   148
                                userID:@"got_out"
dirk@435
   149
                                userName:@"No Way Not Even Alice"
dirk@435
   150
                                isOwn:NO];
dirk@435
   151
dirk@435
   152
    [session updateIdentity:identRandom];
dirk@435
   153
    XCTAssertNil(identRandom.fingerPrint);
dirk@435
   154
}
dirk@435
   155
dirk@437
   156
- (void)testImportPartnerKeys
dirk@436
   157
{
dirk@437
   158
    [self checkImportingKeyFilePath:@"6FF00E97_sec.asc" address:@"pep.test.alice@pep-project.org"
dirk@437
   159
                             userID:@"This Is Alice"
dirk@437
   160
                        fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
dirk@436
   161
dirk@437
   162
    [self checkImportingKeyFilePath:@"0xC9C2EE39.asc" address:@"pep.test.bob@pep-project.org"
dirk@437
   163
                             userID:@"This Is Bob"
dirk@437
   164
                        fingerPrint:@"BFCDB7F301DEEEBBF947F29659BFF488C9C2EE39"];
dirk@436
   165
}
dirk@436
   166
dirk@439
   167
- (void)testIdentityRating
dirk@439
   168
{
dirk@439
   169
    PEPSession *session = [PEPSession new];
dirk@439
   170
dirk@439
   171
    PEPIdentity *me = [self
dirk@439
   172
                       checkMySelfImportingKeyFilePath:@"6FF00E97_sec.asc"
dirk@439
   173
                       address:@"pep.test.alice@pep-project.org"
dirk@439
   174
                       userID:@"Alice_User_ID"
dirk@439
   175
                       fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
dirk@439
   176
    XCTAssertEqual([session identityRating:me], PEP_rating_trusted_and_anonymized);
dirk@439
   177
dirk@439
   178
    PEPIdentity *alice = [self
dirk@439
   179
                          checkImportingKeyFilePath:@"6FF00E97_sec.asc"
dirk@439
   180
                          address:@"pep.test.alice@pep-project.org"
dirk@439
   181
                          userID:@"This Is Alice"
dirk@439
   182
                          fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
dirk@439
   183
    XCTAssertEqual([session identityRating:alice], PEP_rating_reliable);
dirk@439
   184
}
dirk@439
   185
andreas@392
   186
- (void)testOutgoingColors
andreas@392
   187
{
andreas@392
   188
    PEPSession *session = [PEPSession new];
andreas@392
   189
andreas@392
   190
    // Our test user :
andreas@392
   191
    // pEp Test Alice (test key don't use) <pep.test.alice@pep-project.org>
andreas@392
   192
    // 4ABE3AAF59AC32CFE4F86500A9411D176FF00E97
andreas@392
   193
    [PEPTestUtils importBundledKey:@"6FF00E97_sec.asc"];
andreas@392
   194
andreas@392
   195
    // Our test user :
dirk@440
   196
    PEPIdentity *identAlice = [self
dirk@440
   197
                               checkMySelfImportingKeyFilePath:@"6FF00E97_sec.asc"
dirk@440
   198
                               address:@"pep.test.alice@pep-project.org"
dirk@440
   199
                               userID:@"Alice_User_ID"
andreas@392
   200
                               fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
andreas@392
   201
andreas@392
   202
    //Message
andreas@392
   203
dirk@441
   204
    {
dirk@441
   205
        PEPIdentity *identUnknownBob = [[PEPIdentity alloc]
dirk@441
   206
                                        initWithAddress:@"pep.test.unknown.bob@pep-project.org"
dirk@441
   207
                                        userID:@"4242" userName:@"pEp Test Bob Unknown"
dirk@441
   208
                                        isOwn:NO];
dirk@441
   209
dirk@441
   210
        PEPMessage *msgGray = [PEPMessage new];
dirk@441
   211
        msgGray.from = identAlice;
dirk@441
   212
        msgGray.to = @[identUnknownBob];
dirk@441
   213
        msgGray.shortMessage = @"All Gray Test";
dirk@441
   214
        msgGray.longMessage = @"This is a text content";
dirk@441
   215
        msgGray.direction = PEP_dir_outgoing;
dirk@441
   216
dirk@441
   217
        // Test with unknown Bob
dirk@441
   218
        PEP_rating clr = [session outgoingColorForMessage:msgGray];
dirk@441
   219
        XCTAssertEqual(clr, PEP_rating_unencrypted);
dirk@441
   220
    }
dirk@441
   221
dirk@441
   222
    PEPIdentity *identBob = [self
dirk@441
   223
                             checkImportingKeyFilePath:@"0xC9C2EE39.asc"
dirk@441
   224
                             address:@"pep.test.bob@pep-project.org"
dirk@441
   225
                             userID:@"42"
dirk@441
   226
                             fingerPrint:@"BFCDB7F301DEEEBBF947F29659BFF488C9C2EE39"];
dirk@438
   227
andreas@392
   228
    PEPMessage *msg = [PEPMessage new];
andreas@392
   229
    msg.from = identAlice;
dirk@438
   230
    msg.to = @[identBob];
dirk@441
   231
    msg.shortMessage = @"All Gray Test";
andreas@392
   232
    msg.longMessage = @"This is a text content";
andreas@392
   233
    msg.direction = PEP_dir_outgoing;
andreas@392
   234
dirk@441
   235
    // Should be yellow, since no handshake happened.
andreas@392
   236
    PEP_rating clr = [session outgoingColorForMessage:msg];
dirk@420
   237
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   238
andreas@392
   239
    clr = [session identityRating:identBob];
dirk@420
   240
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   241
andreas@392
   242
    // Let' say we got that handshake, set PEP_ct_confirmed in Bob's identity
andreas@392
   243
    [session trustPersonalKey:identBob];
andreas@392
   244
andreas@392
   245
    // This time it should be green
andreas@392
   246
    clr = [session outgoingColorForMessage:msg];
dirk@420
   247
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   248
andreas@392
   249
    clr = [session identityRating:identBob];
dirk@420
   250
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   251
andreas@392
   252
    // Let' say we undo handshake
andreas@392
   253
    [session keyResetTrust:identBob];
andreas@392
   254
andreas@392
   255
    // Yellow ?
andreas@392
   256
    clr = [session outgoingColorForMessage:msg];
dirk@420
   257
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   258
andreas@392
   259
    // mistrust Bob
andreas@392
   260
    [session keyMistrusted:identBob];
andreas@392
   261
dirk@441
   262
    [session updateIdentity:identBob];
dirk@441
   263
    XCTAssertNil(identBob.fingerPrint);
dirk@441
   264
andreas@392
   265
    // Gray == PEP_rating_unencrypted
andreas@392
   266
    clr = [session outgoingColorForMessage:msg];
dirk@420
   267
    XCTAssertEqual(clr, PEP_rating_unencrypted);
andreas@392
   268
andreas@392
   269
    // Forget
andreas@392
   270
    [session keyResetTrust:identBob];
andreas@392
   271
andreas@392
   272
    // Back to yellow
andreas@392
   273
    clr = [session outgoingColorForMessage:msg];
dirk@420
   274
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   275
andreas@392
   276
    // Trust again
andreas@392
   277
    [session trustPersonalKey:identBob];
andreas@392
   278
andreas@392
   279
    // Back to green
andreas@392
   280
    clr = [session outgoingColorForMessage:msg];
dirk@420
   281
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   282
andreas@392
   283
    // Now let see if it turns back yellow if we add an unconfirmed folk.
andreas@392
   284
    // pEp Test John (test key, don't use) <pep.test.john@pep-project.org>
andreas@392
   285
    // AA2E4BEB93E5FE33DEFD8BE1135CD6D170DCF575
andreas@392
   286
    [PEPTestUtils importBundledKey:@"0x70DCF575.asc"];
andreas@392
   287
andreas@392
   288
    PEPIdentity *identJohn = [[PEPIdentity alloc]
andreas@392
   289
                              initWithAddress:@"pep.test.john@pep-project.org"
andreas@392
   290
                              userID:@"101" userName:@"pEp Test John"
andreas@392
   291
                              isOwn:NO
andreas@392
   292
                              fingerPrint:@"AA2E4BEB93E5FE33DEFD8BE1135CD6D170DCF575"];
andreas@392
   293
andreas@392
   294
    [session updateIdentity:identJohn];
andreas@392
   295
andreas@411
   296
    msg.cc = @[[PEPTestUtils foreignPepIdentityWithAddress:@"pep.test.john@pep-project.org"
andreas@411
   297
                                                  userName:@"pEp Test John"]];
andreas@392
   298
    // Yellow ?
andreas@392
   299
    clr = [session outgoingColorForMessage:msg];
dirk@420
   300
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   301
andreas@392
   302
    PEPMessage *encmsg;
andreas@392
   303
    PEP_STATUS status = [session encryptMessage:msg extra:@[] dest:&encmsg];
andreas@392
   304
andreas@392
   305
    XCTAssertNotNil(encmsg);
andreas@392
   306
    XCTAssertEqualObjects(encmsg.shortMessage, @"p≡p");
andreas@392
   307
    XCTAssertTrue([encmsg.longMessage containsString:@"p≡p"]);
andreas@392
   308
dirk@420
   309
    XCTAssertEqual(status, PEP_STATUS_OK);
andreas@392
   310
}
andreas@392
   311
andreas@392
   312
andreas@392
   313
- (void)testOutgoingBccColors
andreas@392
   314
{
andreas@392
   315
    PEPSession *session = [PEPSession new];
andreas@392
   316
andreas@392
   317
    // Our test user :
andreas@392
   318
    // pEp Test Alice (test key don't use) <pep.test.alice@pep-project.org>
andreas@392
   319
    // 4ABE3AAF59AC32CFE4F86500A9411D176FF00E97
andreas@392
   320
    [PEPTestUtils importBundledKey:@"6FF00E97_sec.asc"];
andreas@392
   321
andreas@392
   322
    PEPIdentity *identAlice = [[PEPIdentity alloc]
andreas@392
   323
                               initWithAddress:@"pep.test.alice@pep-project.org"
andreas@392
   324
                               userID:ownUserId
andreas@392
   325
                               userName:@"pEp Test Alice"
andreas@392
   326
                               isOwn:YES
andreas@392
   327
                               fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
andreas@392
   328
andreas@392
   329
    [session mySelf:identAlice];
andreas@392
   330
andreas@392
   331
    PEPMessage *msg = [PEPMessage new];
andreas@392
   332
    msg.from = identAlice;
andreas@392
   333
    msg.to = @[[[PEPIdentity alloc] initWithAddress:@"pep.test.bob@pep-project.org"
andreas@392
   334
                                             userID:@"42" userName:@"pEp Test Bob" isOwn:NO]];
andreas@392
   335
    msg.shortMessage = @"All Green Test";
andreas@392
   336
    msg.longMessage = @"This is a text content";
andreas@392
   337
    msg.direction = PEP_dir_outgoing;
andreas@392
   338
andreas@392
   339
    // Test with unknown Bob
andreas@392
   340
    PEP_rating clr = [session outgoingColorForMessage:msg];
dirk@420
   341
    XCTAssertEqual(clr, PEP_rating_unencrypted);
andreas@392
   342
andreas@392
   343
    // Now let see with bob's pubkey already known
andreas@392
   344
    // pEp Test Bob (test key, don't use) <pep.test.bob@pep-project.org>
andreas@392
   345
    // BFCDB7F301DEEEBBF947F29659BFF488C9C2EE39
andreas@392
   346
    [PEPTestUtils importBundledKey:@"0xC9C2EE39.asc"];
andreas@392
   347
andreas@392
   348
    PEPIdentity *identBob = [[PEPIdentity alloc]
andreas@392
   349
                             initWithAddress:@"pep.test.bob@pep-project.org"
andreas@392
   350
                             userID:@"42" userName:@"pEp Test Bob"
andreas@392
   351
                             isOwn:NO
andreas@392
   352
                             fingerPrint:@"BFCDB7F301DEEEBBF947F29659BFF488C9C2EE39"];
andreas@392
   353
andreas@392
   354
    [session updateIdentity:identBob];
andreas@392
   355
andreas@392
   356
    // Should be yellow, since no handshake happened.
andreas@392
   357
    clr = [session outgoingColorForMessage:msg];
dirk@420
   358
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   359
andreas@392
   360
    clr = [session identityRating:identBob];
dirk@420
   361
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   362
andreas@392
   363
    // Let' say we got that handshake, set PEP_ct_confirmed in Bob's identity
andreas@392
   364
    [session trustPersonalKey:identBob];
andreas@392
   365
andreas@392
   366
    // This time it should be green
andreas@392
   367
    clr = [session outgoingColorForMessage:msg];
dirk@420
   368
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   369
andreas@392
   370
    clr = [session identityRating:identBob];
dirk@420
   371
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   372
andreas@392
   373
    // Now let see if it turns back yellow if we add an unconfirmed folk.
andreas@392
   374
    // pEp Test John (test key, don't use) <pep.test.john@pep-project.org>
andreas@392
   375
    // AA2E4BEB93E5FE33DEFD8BE1135CD6D170DCF575
andreas@392
   376
    [PEPTestUtils importBundledKey:@"0x70DCF575.asc"];
andreas@392
   377
andreas@392
   378
    PEPIdentity *identJohn = [[PEPIdentity alloc]
andreas@392
   379
                              initWithAddress:@"pep.test.john@pep-project.org"
andreas@392
   380
                              userID:@"101" userName:@"pEp Test John"
andreas@392
   381
                              isOwn:NO
andreas@392
   382
                              fingerPrint:@"AA2E4BEB93E5FE33DEFD8BE1135CD6D170DCF575"];
andreas@392
   383
andreas@392
   384
    [session updateIdentity:identJohn];
andreas@392
   385
andreas@392
   386
    msg.bcc = @[[[PEPIdentity alloc] initWithAddress:@"pep.test.john@pep-project.org"
andreas@392
   387
                                              userID:@"101" userName:@"pEp Test John" isOwn:NO]];
andreas@392
   388
andreas@392
   389
    // Yellow ?
andreas@392
   390
    clr = [session outgoingColorForMessage:msg];
dirk@420
   391
    XCTAssertEqual(clr, PEP_rating_reliable);
andreas@392
   392
andreas@392
   393
    [session trustPersonalKey:identJohn];
andreas@392
   394
andreas@392
   395
    // This time it should be green
andreas@392
   396
    clr = [session outgoingColorForMessage:msg];
dirk@420
   397
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   398
andreas@392
   399
    clr = [session identityRating:identJohn];
dirk@420
   400
    XCTAssertEqual(clr, PEP_rating_trusted);
andreas@392
   401
}
andreas@392
   402
andreas@392
   403
- (void)testDontEncryptForMistrusted
andreas@392
   404
{
andreas@392
   405
    PEPSession *session = [PEPSession new];
andreas@392
   406
andreas@392
   407
    // Our test user :
andreas@392
   408
    // pEp Test Alice (test key don't use) <pep.test.alice@pep-project.org>
andreas@392
   409
    // 4ABE3AAF59AC32CFE4F86500A9411D176FF00E97
andreas@392
   410
    [PEPTestUtils importBundledKey:@"6FF00E97_sec.asc"];
andreas@392
   411
andreas@392
   412
    PEPIdentity *identAlice = [[PEPIdentity alloc]
andreas@392
   413
                               initWithAddress:@"pep.test.alice@pep-project.org"
andreas@392
   414
                               userID:ownUserId
andreas@392
   415
                               userName:@"pEp Test Alice"
andreas@392
   416
                               isOwn:YES
andreas@392
   417
                               fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
andreas@392
   418
andreas@392
   419
    [session mySelf:identAlice];
andreas@392
   420
andreas@392
   421
    // pEp Test Bob (test key, don't use) <pep.test.bob@pep-project.org>
andreas@392
   422
    // BFCDB7F301DEEEBBF947F29659BFF488C9C2EE39
andreas@392
   423
    [PEPTestUtils importBundledKey:@"0xC9C2EE39.asc"];
andreas@392
   424
andreas@392
   425
    PEPIdentity *identBob = [[PEPIdentity alloc]
andreas@392
   426
                             initWithAddress:@"pep.test.bob@pep-project.org"
andreas@392
   427
                             userID:@"42" userName:@"pEp Test Bob"
andreas@392
   428
                             isOwn:NO
andreas@392
   429
                             fingerPrint:@"BFCDB7F301DEEEBBF947F29659BFF488C9C2EE39"];
andreas@392
   430
andreas@392
   431
    [session updateIdentity:identBob];
andreas@392
   432
andreas@392
   433
    // mistrust Bob
andreas@392
   434
    [session keyMistrusted:identBob];
andreas@392
   435
andreas@392
   436
    PEPMessage *msg = [PEPMessage new];
andreas@392
   437
    msg.from = identAlice;
andreas@392
   438
    msg.to = @[[[PEPIdentity alloc] initWithAddress:@"pep.test.bob@pep-project.org" userID:@"42"
andreas@392
   439
                                           userName:@"pEp Test Bob" isOwn:NO]];
andreas@392
   440
    msg.shortMessage = @"All Green Test";
andreas@392
   441
    msg.longMessage = @"This is a text content";
andreas@392
   442
    msg.direction = PEP_dir_outgoing;
andreas@392
   443
andreas@392
   444
    // Gray == PEP_rating_unencrypted
andreas@392
   445
    PEP_rating clr = [session outgoingColorForMessage:msg];
dirk@420
   446
    XCTAssertEqual(clr, PEP_rating_unencrypted);
andreas@392
   447
andreas@392
   448
    PEPMessage *encmsg;
andreas@392
   449
    PEP_STATUS status = [session encryptMessage:msg extra:@[] dest:&encmsg];
andreas@392
   450
dirk@420
   451
    XCTAssertEqual(status, PEP_UNENCRYPTED);
andreas@392
   452
andreas@392
   453
    XCTAssertNotEqualObjects(encmsg.attachments[0][@"mimeType"], @"application/pgp-encrypted");
andreas@392
   454
andreas@392
   455
    [self pEpCleanUp];
andreas@392
   456
}
andreas@392
   457
andreas@392
   458
- (void)testRenewExpired
andreas@392
   459
{
andreas@392
   460
    PEPSession *session = [PEPSession new];
andreas@392
   461
andreas@392
   462
    // Our expired test user :
andreas@392
   463
    // pEp Test Hector (old test key don't use) <pep.test.hector@pep-project.org>
andreas@392
   464
    [PEPTestUtils importBundledKey:@"5CB2C182_sec.asc"];
andreas@392
   465
andreas@392
   466
    PEPIdentity *identHector = [[PEPIdentity alloc]
andreas@392
   467
                                initWithAddress:@"pep.test.hector@pep-project.org"
andreas@392
   468
                                userID:@"fc2d33" userName:@"pEp Test Hector"
andreas@392
   469
                                isOwn:NO
andreas@392
   470
                                fingerPrint:@"EEA655839E347EC9E10A5DE2E80CB3FD5CB2C182"];
andreas@392
   471
andreas@392
   472
    // Check that this key is indeed expired
andreas@392
   473
    [session updateIdentity:identHector];
andreas@392
   474
    XCTAssertEqual(PEP_ct_key_expired, identHector.commType);
andreas@392
   475
andreas@392
   476
    PEPIdentity *identHectorOwn = [[PEPIdentity alloc]
andreas@392
   477
                                   initWithAddress:@"pep.test.hector@pep-project.org"
andreas@392
   478
                                   userID:ownUserId userName:@"pEp Test Hector"
andreas@392
   479
                                   isOwn:YES
andreas@392
   480
                                   fingerPrint:@"EEA655839E347EC9E10A5DE2E80CB3FD5CB2C182"];
andreas@392
   481
andreas@392
   482
    // Myself automatically renew expired key.
andreas@392
   483
    [session mySelf:identHectorOwn];
andreas@392
   484
    XCTAssertEqual(PEP_ct_pEp, identHectorOwn.commType);
andreas@392
   485
andreas@392
   486
    [self pEpCleanUpRestoringBackupNamed:@"Bob"];
andreas@392
   487
andreas@392
   488
andreas@392
   489
    [self pEpSetUp:@"Bob"];
andreas@392
   490
andreas@392
   491
    PEPIdentity *_identHector = [[PEPIdentity alloc]
andreas@392
   492
                                 initWithAddress:@"pep.test.hector@pep-project.org"
andreas@392
   493
                                 userID:@"khkhkh" userName:@"pEp Test Hector"
andreas@392
   494
                                 isOwn:NO
andreas@392
   495
                                 fingerPrint:@"EEA655839E347EC9E10A5DE2E80CB3FD5CB2C182"];
andreas@392
   496
andreas@392
   497
    [session updateIdentity:_identHector];
andreas@392
   498
    XCTAssertEqual(PEP_ct_OpenPGP_unconfirmed, _identHector.commType);
andreas@392
   499
}
andreas@392
   500
andreas@392
   501
- (void)testRevoke
andreas@392
   502
{
andreas@392
   503
    PEPSession *session = [PEPSession new];
andreas@392
   504
andreas@392
   505
    // Our test user :
andreas@392
   506
    // pEp Test Alice (test key don't use) <pep.test.alice@pep-project.org>
andreas@392
   507
    // 4ABE3AAF59AC32CFE4F86500A9411D176FF00E97
andreas@392
   508
    [PEPTestUtils importBundledKey:@"6FF00E97_sec.asc"];
andreas@392
   509
    NSString *fpr = @"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97";
andreas@392
   510
andreas@392
   511
    PEPIdentity *identAlice = [[PEPIdentity alloc]
andreas@392
   512
                               initWithAddress:@"pep.test.alice@pep-project.org"
andreas@392
   513
                               userID:ownUserId
andreas@392
   514
                               userName:@"pEp Test Alice"
andreas@392
   515
                               isOwn:YES
andreas@392
   516
                               fingerPrint:fpr];
andreas@392
   517
andreas@392
   518
    [session mySelf:identAlice];
andreas@392
   519
andreas@392
   520
    PEPIdentity *identAlice2 = [identAlice mutableCopy];
andreas@392
   521
andreas@392
   522
    // This will revoke key
andreas@392
   523
    [session keyMistrusted:identAlice2];
andreas@416
   524
    identAlice2.fingerPrint = nil;
andreas@416
   525
    [session mySelf:identAlice];
andreas@416
   526
andreas@392
   527
andreas@392
   528
    // Check fingerprint is different
andreas@392
   529
    XCTAssertNotEqualObjects(identAlice2.fingerPrint, fpr);
andreas@392
   530
}
andreas@392
   531
andreas@392
   532
- (void)testMailToMyself
andreas@392
   533
{
andreas@392
   534
    PEPSession *session = [PEPSession new];
andreas@392
   535
andreas@392
   536
    // Our test user :
andreas@392
   537
    // pEp Test Alice (test key don't use) <pep.test.alice@pep-project.org>
andreas@392
   538
    // 4ABE3AAF59AC32CFE4F86500A9411D176FF00E97
andreas@392
   539
    [PEPTestUtils importBundledKey:@"6FF00E97_sec.asc"];
andreas@392
   540
andreas@392
   541
    PEPIdentity *identAlice = [[PEPIdentity alloc]
andreas@392
   542
                               initWithAddress:@"pep.test.alice@pep-project.org"
andreas@392
   543
                               userID:ownUserId
andreas@392
   544
                               userName:@"pEp Test Alice"
andreas@392
   545
                               isOwn:YES
andreas@392
   546
                               fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
andreas@392
   547
andreas@392
   548
    [session mySelf:identAlice];
andreas@392
   549
andreas@392
   550
    PEPMessage *msg = [PEPMessage new];
andreas@392
   551
    msg.from = identAlice;
andreas@392
   552
    msg.to = @[identAlice];
andreas@392
   553
    msg.shortMessage = @"Mail to Myself";
andreas@392
   554
    msg.longMessage = @"This is a text content";
andreas@392
   555
    msg.direction = PEP_dir_outgoing;
andreas@392
   556
andreas@392
   557
    PEP_rating clr = [session outgoingColorForMessage:msg];
dirk@420
   558
    XCTAssertEqual(clr, PEP_rating_trusted_and_anonymized);
andreas@392
   559
andreas@392
   560
    PEPMessage *encmsg;
andreas@392
   561
    PEP_STATUS status = [session encryptMessage:msg extra:@[] dest:&encmsg];
andreas@392
   562
dirk@420
   563
    XCTAssertEqual(status, PEP_STATUS_OK);
andreas@392
   564
andreas@392
   565
    NSArray* keys;
andreas@392
   566
    PEPMessage *decmsg;
andreas@392
   567
andreas@392
   568
    clr = [session decryptMessage:encmsg dest:&decmsg keys:&keys];
dirk@420
   569
    XCTAssertEqual(clr, PEP_rating_trusted_and_anonymized);
andreas@392
   570
}
andreas@392
   571
andreas@392
   572
- (void)testEncryptedMailFromMutt
andreas@392
   573
{
andreas@392
   574
    PEPSession *session = [PEPSession new];
andreas@392
   575
andreas@392
   576
    // This is the public key for test001@peptest.ch
andreas@392
   577
    [PEPTestUtils importBundledKey:@"A3FC7F0A.asc"];
andreas@392
   578
andreas@392
   579
    // This is the secret key for test001@peptest.ch
andreas@392
   580
    [PEPTestUtils importBundledKey:@"A3FC7F0A_sec.asc"];
andreas@392
   581
andreas@392
   582
    // Mail from mutt, already processed into message dict by the app.
dirk@434
   583
    NSMutableDictionary *msgDict = [PEPTestUtils
dirk@434
   584
                                    unarchiveDictionary:@"msg_to_A3FC7F0A_from_mutt.ser"].mutableCopy;
andreas@392
   585
    [msgDict removeObjectForKey:kPepLongMessage];
andreas@392
   586
    [msgDict removeObjectForKey:kPepLongMessageFormatted];
andreas@392
   587
andreas@392
   588
    // Also extracted "live" from the app.
dirk@434
   589
    NSMutableDictionary *accountDict = [PEPTestUtils
dirk@434
   590
                                        unarchiveDictionary:@"account_A3FC7F0A.ser"].mutableCopy;
andreas@392
   591
    [accountDict removeObjectForKey:kPepCommType];
andreas@392
   592
    [accountDict removeObjectForKey:kPepFingerprint];
andreas@392
   593
    PEPIdentity *identMe = [[PEPIdentity alloc] initWithDictionary:accountDict];
andreas@392
   594
andreas@392
   595
    [session mySelf:identMe];
andreas@392
   596
    XCTAssertNotNil(identMe.fingerPrint);
andreas@392
   597
andreas@392
   598
    NSArray* keys;
andreas@392
   599
    PEPMessage *msg = [PEPMessage new];
andreas@392
   600
    [msg setValuesForKeysWithDictionary:msgDict];
andreas@392
   601
    PEPMessage *pepDecryptedMail;
andreas@392
   602
    [session decryptMessage:msg dest:&pepDecryptedMail keys:&keys];
andreas@392
   603
    XCTAssertNotNil(pepDecryptedMail.longMessage);
andreas@392
   604
}
andreas@392
   605
andreas@392
   606
- (void)testOutgoingContactColor
andreas@392
   607
{
andreas@392
   608
    PEPSession *session = [PEPSession new];
andreas@392
   609
andreas@411
   610
    PEPIdentity *partner1Orig = [PEPTestUtils foreignPepIdentityWithAddress:@"partner1@dontcare.me"
andreas@411
   611
                                                                   userName:@"Partner 1"];
andreas@392
   612
    NSString *pubKeyPartner1 = [PEPTestUtils loadResourceByName:@"partner1_F2D281C2789DD7F6_pub.asc"];
andreas@392
   613
    XCTAssertNotNil(pubKeyPartner1);
andreas@392
   614
    [session importKey:pubKeyPartner1];
andreas@392
   615
andreas@392
   616
    PEP_rating color = [session identityRating:partner1Orig];
andreas@392
   617
    XCTAssertEqual(color, PEP_rating_reliable);
andreas@392
   618
}
andreas@392
   619
andreas@392
   620
- (void)testMessageTrustwordsWithMySelf
andreas@392
   621
{
andreas@392
   622
    PEPSession *session = [PEPSession new];
andreas@392
   623
andreas@392
   624
    PEPStringList *keys = nil;
andreas@392
   625
    PEPMessage *decryptedDict = [self internalEncryptToMySelfKeys:&keys];
andreas@392
   626
    XCTAssertNotNil(keys);
andreas@392
   627
    XCTAssert(keys.count > 0);
andreas@392
   628
andreas@392
   629
    PEPIdentity *receiver = decryptedDict.to[0];
andreas@392
   630
    [session updateIdentity:receiver];
andreas@392
   631
    XCTAssertNotNil(receiver);
andreas@392
   632
    PEP_STATUS trustwordsStatus;
andreas@392
   633
andreas@392
   634
    NSString *trustwords = [session getTrustwordsForMessage:decryptedDict
andreas@392
   635
                                                   receiver:receiver
andreas@392
   636
                                                  keysArray:keys language:@"en"
andreas@392
   637
                                                       full:YES
andreas@392
   638
                                            resultingStatus: &trustwordsStatus];
andreas@392
   639
    // No trustwords with yourself
andreas@392
   640
    XCTAssertEqual(trustwordsStatus, PEP_TRUSTWORDS_DUPLICATE_FPR);
andreas@392
   641
    XCTAssertNil(trustwords);
andreas@392
   642
}
andreas@392
   643
andreas@392
   644
- (void)testGetTrustwords
andreas@392
   645
{
andreas@392
   646
    PEPSession *session = [PEPSession new];
andreas@392
   647
andreas@392
   648
    PEPIdentity *partner1Orig = [[PEPIdentity alloc]
andreas@392
   649
                                 initWithAddress:@"partner1@dontcare.me" userID:@"partner1"
andreas@392
   650
                                 userName:@"partner1"
andreas@392
   651
                                 isOwn:NO fingerPrint:@"F0CD3F7B422E5D587ABD885BF2D281C2789DD7F6"];
andreas@392
   652
andreas@392
   653
    PEPIdentity *meOrig = [[PEPIdentity alloc]
andreas@392
   654
                           initWithAddress:@"me@dontcare.me" userID:@"me"
andreas@392
   655
                           userName:@"me"
andreas@392
   656
                           isOwn:NO fingerPrint:@"CC1F73F6FB774BF08B197691E3BFBCA9248FC681"];
andreas@392
   657
andreas@392
   658
    NSString *pubKeyPartner1 = [PEPTestUtils loadResourceByName:@"partner1_F2D281C2789DD7F6_pub.asc"];
andreas@392
   659
    XCTAssertNotNil(pubKeyPartner1);
andreas@392
   660
    NSString *pubKeyMe = [PEPTestUtils loadResourceByName:@"meATdontcare_E3BFBCA9248FC681_pub.asc"];
andreas@392
   661
    XCTAssertNotNil(pubKeyMe);
andreas@392
   662
    NSString *secKeyMe = [PEPTestUtils loadResourceByName:@"meATdontcare_E3BFBCA9248FC681_sec.asc"];
andreas@392
   663
    XCTAssertNotNil(secKeyMe);
andreas@392
   664
andreas@392
   665
    NSString *trustwordsFull = [session getTrustwordsIdentity1:meOrig identity2:partner1Orig
andreas@392
   666
                                                      language:nil full:YES];
andreas@392
   667
    XCTAssertEqualObjects(trustwordsFull,
andreas@392
   668
                          @"EMERSON GASPER TOKENISM BOLUS COLLAGE DESPISE BEDDED ENCRYPTION IMAGINE BEDFORD");
andreas@392
   669
andreas@392
   670
    NSString *trustwordsFullEnglish = [session getTrustwordsIdentity1:meOrig identity2:partner1Orig
andreas@392
   671
                                                             language:@"en" full:YES];
andreas@392
   672
    XCTAssertEqualObjects(trustwordsFullEnglish, trustwordsFull);
andreas@392
   673
andreas@392
   674
    NSString *trustwordsUndefined = [session getTrustwordsIdentity1:meOrig identity2:partner1Orig
andreas@392
   675
                                                           language:@"ZZ" full:YES];
andreas@392
   676
    XCTAssertNil(trustwordsUndefined);
andreas@392
   677
}
andreas@392
   678
andreas@416
   679
#pragma mark - configUnencryptedSubject
andreas@416
   680
andreas@416
   681
- (void)testConfigUnencryptedSubject
andreas@416
   682
{
andreas@416
   683
    // Setup Config to encrypt subject
andreas@416
   684
    [PEPObjCAdapter setUnecryptedSubjectEnabled:NO];
andreas@416
   685
andreas@416
   686
    // Write mail to yourself ...
andreas@416
   687
    PEPMessage *encMessage = [self mailWrittenToMySelf];
andreas@416
   688
andreas@416
   689
    // ... and assert subject is encrypted
andreas@416
   690
    XCTAssertEqualObjects(encMessage.shortMessage, @"p≡p", @"Subject should be encrypted");
andreas@416
   691
}
andreas@416
   692
andreas@416
   693
- (void)testConfigUnencryptedSubject_encryptedSubjectDisabled
andreas@416
   694
{
andreas@416
   695
    // Setup Config to not encrypt subject
andreas@416
   696
    [PEPObjCAdapter setUnecryptedSubjectEnabled:YES];
andreas@416
   697
andreas@416
   698
    // Write mail to yourself ...
andreas@416
   699
    PEPMessage *encMessage = [self mailWrittenToMySelf];
andreas@416
   700
andreas@416
   701
    // ... and assert the subject is not encrypted
andreas@416
   702
    XCTAssertNotEqualObjects(encMessage.shortMessage, @"p≡p", @"Subject should not be encrypted");
andreas@416
   703
}
andreas@416
   704
dirk@418
   705
- (void)testStringToRating
dirk@418
   706
{
dirk@418
   707
    PEPSession *session = [PEPSession new];
dirk@418
   708
    XCTAssertEqual([session ratingFromString:@"cannot_decrypt"], PEP_rating_cannot_decrypt);
dirk@418
   709
    XCTAssertEqual([session ratingFromString:@"have_no_key"], PEP_rating_have_no_key);
dirk@418
   710
    XCTAssertEqual([session ratingFromString:@"unencrypted"], PEP_rating_unencrypted);
dirk@418
   711
    XCTAssertEqual([session ratingFromString:@"unencrypted_for_some"],
dirk@418
   712
                   PEP_rating_unencrypted_for_some);
dirk@418
   713
    XCTAssertEqual([session ratingFromString:@"unreliable"], PEP_rating_unreliable);
dirk@418
   714
    XCTAssertEqual([session ratingFromString:@"reliable"], PEP_rating_reliable);
dirk@418
   715
    XCTAssertEqual([session ratingFromString:@"trusted"], PEP_rating_trusted);
dirk@418
   716
    XCTAssertEqual([session ratingFromString:@"trusted_and_anonymized"],
dirk@418
   717
                   PEP_rating_trusted_and_anonymized);
dirk@418
   718
    XCTAssertEqual([session ratingFromString:@"fully_anonymous"], PEP_rating_fully_anonymous);
dirk@418
   719
    XCTAssertEqual([session ratingFromString:@"mistrust"], PEP_rating_mistrust);
dirk@418
   720
    XCTAssertEqual([session ratingFromString:@"b0rken"], PEP_rating_b0rken);
dirk@418
   721
    XCTAssertEqual([session ratingFromString:@"under_attack"], PEP_rating_under_attack);
dirk@418
   722
    XCTAssertEqual([session ratingFromString:@"undefined"], PEP_rating_undefined);
dirk@418
   723
    XCTAssertEqual([session ratingFromString:@"does not exist111"], PEP_rating_undefined);
dirk@418
   724
}
dirk@418
   725
dirk@418
   726
- (void)testRatingToString
dirk@418
   727
{
dirk@418
   728
    PEPSession *session = [PEPSession new];
dirk@418
   729
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_cannot_decrypt], @"cannot_decrypt");
dirk@418
   730
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_have_no_key], @"have_no_key");
dirk@418
   731
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_unencrypted], @"unencrypted");
dirk@418
   732
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_unencrypted_for_some],
dirk@418
   733
                          @"unencrypted_for_some");
dirk@418
   734
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_unreliable], @"unreliable");
dirk@418
   735
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_reliable], @"reliable");
dirk@418
   736
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_trusted], @"trusted");
dirk@418
   737
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_trusted_and_anonymized],
dirk@418
   738
                          @"trusted_and_anonymized");
dirk@418
   739
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_fully_anonymous],
dirk@418
   740
                          @"fully_anonymous");
dirk@418
   741
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_mistrust], @"mistrust");
dirk@418
   742
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_b0rken], @"b0rken");
dirk@418
   743
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_under_attack], @"under_attack");
dirk@418
   744
    XCTAssertEqualObjects([session stringFromRating:PEP_rating_undefined], @"undefined");
dirk@418
   745
    XCTAssertEqualObjects([session stringFromRating:500], @"undefined");
dirk@418
   746
}
dirk@418
   747
dirk@431
   748
- (void)testIsPEPUser
dirk@431
   749
{
dirk@431
   750
    PEPSession *session = [PEPSession new];
dirk@431
   751
dirk@431
   752
    PEPIdentity *identMe = [[PEPIdentity alloc]
dirk@431
   753
                            initWithAddress:@"me-myself-and-i@pep-project.org"
dirk@431
   754
                            userID:@"me-myself-and-i"
dirk@431
   755
                            userName:@"pEp Test Alice"
dirk@431
   756
                            isOwn:YES];
dirk@431
   757
    [session mySelf:identMe];
dirk@431
   758
    XCTAssertNotNil(identMe.fingerPrint);
dirk@432
   759
dirk@432
   760
    // PEP_CANNOT_FIND_PERSON == 902
dirk@431
   761
    NSError *error;
dirk@431
   762
    XCTAssertTrue([session isPEPUser:identMe error:&error]);
dirk@431
   763
    XCTAssertNil(error);
dirk@431
   764
}
dirk@431
   765
andreas@392
   766
#pragma mark - Helpers
andreas@392
   767
dirk@439
   768
- (PEPIdentity *)checkImportingKeyFilePath:(NSString *)filePath address:(NSString *)address
dirk@439
   769
                                    userID:(NSString *)userID
dirk@439
   770
                               fingerPrint:(NSString *)fingerPrint
dirk@437
   771
{
dirk@437
   772
    PEPSession *session = [PEPSession new];
dirk@437
   773
dirk@437
   774
    [PEPTestUtils importBundledKey:filePath];
dirk@437
   775
dirk@437
   776
    // Our test user:
dirk@437
   777
    PEPIdentity *identTest = [[PEPIdentity alloc]
dirk@437
   778
                              initWithAddress:address
dirk@437
   779
                              userID:userID
dirk@439
   780
                              userName:[NSString stringWithFormat:@"Some User Name %@", userID]
dirk@437
   781
                              isOwn:NO];
dirk@437
   782
dirk@437
   783
    [session updateIdentity:identTest];
dirk@437
   784
    XCTAssertNotNil(identTest.fingerPrint);
dirk@437
   785
    XCTAssertEqualObjects(identTest.fingerPrint, fingerPrint);
dirk@439
   786
dirk@439
   787
    return identTest;
dirk@439
   788
}
dirk@439
   789
dirk@439
   790
- (PEPIdentity *)checkMySelfImportingKeyFilePath:(NSString *)filePath address:(NSString *)address
dirk@439
   791
                                          userID:(NSString *)userID
dirk@439
   792
                                     fingerPrint:(NSString *)fingerPrint
dirk@439
   793
{
dirk@439
   794
    PEPSession *session = [PEPSession new];
dirk@439
   795
dirk@439
   796
    [PEPTestUtils importBundledKey:filePath];
dirk@439
   797
dirk@439
   798
    // Our test user:
dirk@439
   799
    PEPIdentity *identTest = [[PEPIdentity alloc]
dirk@439
   800
                              initWithAddress:address
dirk@439
   801
                              userID:userID
dirk@439
   802
                              userName:[NSString stringWithFormat:@"Some User Name %@", userID]
dirk@439
   803
                              isOwn:YES
dirk@439
   804
                              fingerPrint: fingerPrint];
dirk@439
   805
dirk@439
   806
    [session mySelf:identTest];
dirk@439
   807
    XCTAssertNotNil(identTest.fingerPrint);
dirk@439
   808
    XCTAssertEqualObjects(identTest.fingerPrint, fingerPrint);
dirk@439
   809
dirk@439
   810
    return identTest;
dirk@437
   811
}
dirk@437
   812
dirk@433
   813
/**
dirk@433
   814
 Verifies that a partner ID is really a correct Identity.
dirk@433
   815
 Usually used on identities imported as keys, since the engine has problems with them.
dirk@433
   816
 */
dirk@433
   817
- (void)updateAndVerifyPartnerIdentity:(PEPIdentity *)partnerIdentity session:(PEPSession *)session
dirk@433
   818
{
dirk@433
   819
    XCTAssertNotNil(partnerIdentity.fingerPrint);
dirk@433
   820
    [session updateIdentity:partnerIdentity];
dirk@433
   821
    XCTAssertNotNil(partnerIdentity.fingerPrint);
dirk@433
   822
    NSString *fingerprint = partnerIdentity.fingerPrint;
dirk@433
   823
    partnerIdentity.fingerPrint = nil;
dirk@433
   824
    [session updateIdentity:partnerIdentity];
dirk@433
   825
    XCTAssertNotNil(partnerIdentity.fingerPrint);
dirk@433
   826
    XCTAssertEqualObjects(partnerIdentity.fingerPrint, fingerprint);
dirk@433
   827
}
dirk@433
   828
andreas@416
   829
- (PEPMessage *)mailWrittenToMySelf
andreas@416
   830
{
andreas@416
   831
    PEPSession *session = [PEPSession new];
andreas@416
   832
andreas@416
   833
    // Write a e-mail to yourself ...
andreas@416
   834
    PEPIdentity *me = [PEPTestUtils ownPepIdentityWithAddress:@"me@peptest.ch"
andreas@416
   835
                                                     userName:@"userName"];
andreas@416
   836
    [session mySelf:me];
andreas@416
   837
andreas@416
   838
    NSString *shortMessage = @"Subject";
andreas@416
   839
    NSString *longMessage = @"Oh, this is a long body text!";
andreas@416
   840
    PEPMessage *mail = [PEPTestUtils mailFrom:me
andreas@416
   841
                                      toIdent:me
andreas@416
   842
                                 shortMessage:shortMessage
andreas@416
   843
                                  longMessage:longMessage
andreas@416
   844
                                     outgoing:YES];
andreas@416
   845
    PEPMessage *encMessage;
andreas@416
   846
    [session encryptMessage:mail identity:me dest:&encMessage];
andreas@416
   847
andreas@416
   848
    return encMessage;
andreas@416
   849
}
andreas@416
   850
andreas@392
   851
- (PEPMessage *)internalEncryptToMySelfKeys:(PEPStringList **)keys
andreas@392
   852
{
andreas@392
   853
    PEPSession *session = [PEPSession new];
andreas@411
   854
    
andreas@411
   855
    PEPIdentity *me = [PEPTestUtils ownPepIdentityWithAddress:@"me@peptest.ch"
andreas@411
   856
                                                     userName:@"userName"];
andreas@392
   857
    [session mySelf:me];
andreas@392
   858
    XCTAssertNotNil(me.fingerPrint);
andreas@392
   859
andreas@392
   860
    // Create draft
andreas@392
   861
    NSString *shortMessage = @"Subject";
andreas@392
   862
    NSString *longMessage = @"Oh, this is a long body text!";
andreas@392
   863
    PEPMessage *mail = [PEPTestUtils mailFrom:me toIdent:me shortMessage:shortMessage longMessage:longMessage outgoing:YES];
andreas@392
   864
andreas@392
   865
    PEPMessage *encMessage;
andreas@392
   866
    PEP_STATUS status = [session encryptMessage:mail identity:me dest:&encMessage];
andreas@392
   867
    XCTAssertEqual(status, 0);
andreas@392
   868
    XCTAssertEqualObjects(encMessage.shortMessage, @"p≡p");
andreas@392
   869
andreas@392
   870
    PEPMessage *unencDict;
andreas@392
   871
    PEP_rating rating = [session decryptMessage:encMessage dest:&unencDict keys:keys];
andreas@392
   872
    XCTAssertGreaterThanOrEqual(rating, PEP_rating_reliable);
andreas@392
   873
andreas@392
   874
    XCTAssertEqualObjects(unencDict.shortMessage, shortMessage);
andreas@392
   875
    XCTAssertEqualObjects(unencDict.longMessage, longMessage);
andreas@392
   876
andreas@392
   877
    return unencDict;
andreas@392
   878
}
andreas@392
   879
andreas@392
   880
- (void)pEpCleanUpRestoringBackupNamed:(NSString *)backup {
andreas@392
   881
    [PEPTestUtils deleteWorkFilesAfterBackingUpWithBackupName:backup];
andreas@392
   882
}
andreas@392
   883
andreas@392
   884
- (void)pEpCleanUp
andreas@392
   885
{
andreas@392
   886
    [PEPSession cleanup];
andreas@392
   887
    [self pEpCleanUpRestoringBackupNamed:NULL];
andreas@392
   888
}
andreas@392
   889
andreas@392
   890
- (void)pEpSetUp:(NSString *)restore
andreas@392
   891
{
andreas@392
   892
    // Must be the first thing you do before using anything pEp-related
andreas@392
   893
    // ... but this is now done in session, with a "dispatch_once"
andreas@392
   894
    // [PEPObjCAdapter setupTrustWordsDB:[NSBundle bundleForClass:[self class]]];
andreas@392
   895
andreas@392
   896
    [PEPTestUtils deleteWorkFilesAfterBackingUpWithBackupName:nil];
andreas@392
   897
    [PEPTestUtils restoreWorkFilesFromBackupNamed:restore];
andreas@392
   898
}
andreas@392
   899
andreas@392
   900
- (void)pEpSetUp
andreas@392
   901
{
andreas@392
   902
    [self pEpSetUp:NULL];
andreas@392
   903
}
andreas@392
   904
dirk@424
   905
- (void)helperXEncStatusForOutgoingEncryptdMailToSelf:(BOOL)toSelf
dirk@424
   906
                                       expectedRating:(PEP_rating)expectedRating
dirk@422
   907
{
dirk@422
   908
    PEPSession *session = [PEPSession new];
dirk@422
   909
dirk@422
   910
    // Partner pubkey for the test:
dirk@422
   911
    // pEp Test Alice (test key don't use) <pep.test.alice@pep-project.org>
dirk@422
   912
    // 4ABE3AAF59AC32CFE4F86500A9411D176FF00E97
dirk@422
   913
    [PEPTestUtils importBundledKey:@"0x6FF00E97.asc"];
dirk@422
   914
dirk@422
   915
    PEPIdentity *identAlice = [[PEPIdentity alloc]
dirk@422
   916
                               initWithAddress:@"pep.test.alice@pep-project.org"
dirk@422
   917
                               userID:ownUserId
dirk@422
   918
                               userName:@"pEp Test Alice"
dirk@422
   919
                               isOwn:NO
dirk@433
   920
                               fingerPrint:@"4ABE3AAF59AC32CFE4F86500A9411D176FF00E97"];
dirk@433
   921
    [self updateAndVerifyPartnerIdentity:identAlice session:session];
dirk@422
   922
dirk@422
   923
    PEPIdentity *identMe = [[PEPIdentity alloc]
dirk@422
   924
                               initWithAddress:@"me-myself-and-i@pep-project.org"
dirk@422
   925
                               userID:@"me-myself-and-i"
dirk@422
   926
                               userName:@"pEp Test Alice"
dirk@422
   927
                               isOwn:YES];
dirk@422
   928
    [session mySelf:identMe];
dirk@422
   929
    XCTAssertNotNil(identMe.fingerPrint);
dirk@422
   930
dirk@422
   931
    PEPMessage *msg = [PEPMessage new];
dirk@422
   932
    msg.from = identMe;
dirk@422
   933
    msg.to = @[identAlice];
dirk@422
   934
    msg.shortMessage = @"Mail to Alice";
dirk@422
   935
    msg.longMessage = @"Alice?";
dirk@422
   936
    msg.direction = PEP_dir_outgoing;
dirk@422
   937
dirk@422
   938
    PEP_rating clr = [session outgoingColorForMessage:msg];
dirk@424
   939
    XCTAssertEqual(clr, PEP_rating_reliable);
dirk@422
   940
dirk@423
   941
    PEPMessage *encMsg;
dirk@424
   942
dirk@424
   943
    PEP_STATUS statusEnc = PEP_VERSION_MISMATCH;
dirk@424
   944
    if (toSelf) {
dirk@424
   945
        statusEnc = [session encryptMessage:msg identity:identMe dest:&encMsg];
dirk@424
   946
    } else {
dirk@424
   947
         statusEnc = [session encryptMessage:msg extra:@[] dest:&encMsg];
dirk@424
   948
    }
dirk@422
   949
dirk@423
   950
    XCTAssertEqual(statusEnc, PEP_STATUS_OK);
dirk@423
   951
dirk@423
   952
    XCTAssertNotNil(encMsg);
dirk@423
   953
dirk@423
   954
    PEPMessage *decMsg;
dirk@423
   955
    PEPStringList *keys;
dirk@423
   956
    PEP_rating pEpRating = [session decryptMessage:encMsg dest:&decMsg keys:&keys];
dirk@424
   957
    XCTAssertEqual(pEpRating, expectedRating);
dirk@423
   958
    XCTAssertNotNil(decMsg);
dirk@423
   959
dirk@423
   960
    NSArray * encStatusField = nil;
dirk@423
   961
    for (NSArray *field in decMsg.optionalFields) {
dirk@423
   962
        NSString *header = [field[0] lowercaseString];
dirk@423
   963
        if ([header isEqualToString:@"x-encstatus"]) {
dirk@423
   964
            encStatusField = field;
dirk@423
   965
        }
dirk@423
   966
    }
dirk@423
   967
    XCTAssertNotNil(encStatusField);
dirk@423
   968
    if (encStatusField) {
dirk@423
   969
        PEP_rating outgoingRating = [session ratingFromString:encStatusField[1]];
dirk@424
   970
        XCTAssertEqual(outgoingRating, expectedRating);
dirk@423
   971
    }
dirk@422
   972
}
dirk@422
   973
dirk@424
   974
- (void)testXEncStatusForOutgoingEncryptedMail
dirk@424
   975
{
dirk@424
   976
    [self helperXEncStatusForOutgoingEncryptdMailToSelf:NO expectedRating:PEP_rating_reliable];
dirk@424
   977
}
dirk@424
   978
dirk@424
   979
- (void)testXEncStatusForOutgoingSelfEncryptedMail
dirk@424
   980
{
dirk@424
   981
    [self helperXEncStatusForOutgoingEncryptdMailToSelf:YES
dirk@424
   982
                                         expectedRating:PEP_rating_trusted_and_anonymized];
dirk@424
   983
}
dirk@424
   984
andreas@392
   985
@end