Fixed typo

MacPassTests/KPKTestLegacyWriting.h

@@ -9,9 +9,6 @@
 #import <SenTestingKit/SenTestingKit.h>
 @class KPKPassword;
 
-@interface KPKTestLegacyWriting : SenTestCase {
+@interface KPKTestLegacyWriting : SenTestCase
-  NSData *_data;
-  KPKPassword *_password;
-}
 
 @end

MacPassTests/KPKTestLegacyWriting.m

@@ -8,28 +8,34 @@
 
 #import "KPKTestLegacyWriting.h"
 
-
 #import "KPKPassword.h"
 #import "KPKTree+Serializing.h"
 
 @implementation KPKTestLegacyWriting
 
-- (void)setUp {
-  NSBundle *myBundle = [NSBundle bundleForClass:[self class]];
-  NSURL *url = [myBundle URLForResource:@"Test_Password_1234" withExtension:@"kdb"];
-  _data = [NSData dataWithContentsOfURL:url];
-  _password = [[KPKPassword alloc] initWithPassword:@"1234" key:nil];
-}
-
-- (void)tearDown {
-  _data = nil;
-  _password = nil;
-}
-
 - (void)testWriting {
-  NSError *error = nil;
+  NSError __autoreleasing *error = nil;
-  KPKTree *tree = [[KPKTree alloc] initWithData:_data password:_password error:&error];
+  NSURL *dbUrl = [self _urlForFile:@"CustomIcon_Password_1234" extension:@"kdbx"];
-  NSData *data = [tree encryptWithPassword:_password forVersion:KPKLegacyVersion error:&error];
+  KPKPassword *password = [[KPKPassword alloc] initWithPassword:@"1234" key:nil];
+  KPKTree *tree = [[KPKTree alloc] initWithContentsOfUrl:dbUrl password:password error:&error];
+  STAssertNotNil(tree, @"Tree should be created");
+  error = nil;
+  NSData *data = [tree encryptWithPassword:password forVersion:KPKLegacyVersion error:&error];
+  STAssertNotNil(data, @"Serialized Data shoudl be created");
+  NSString *tempFile = [NSTemporaryDirectory() stringByAppendingString:@"CustomIcon_Password_1234.kdb"];
+  NSLog(@"Saved to %@", tempFile);
+  [data writeToFile:tempFile atomically:YES];
+  KPKTree *loadTree = [[KPKTree alloc] initWithData:data password:password error:&error];
+  STAssertNotNil(loadTree, @"Tree should be loadable from kdb file data");
+}
+
+- (NSData *)_dataForFile:(NSString *)name extension:(NSString *)extension {
+  NSURL *url = [self _urlForFile:name extension:extension];
+  return [NSData dataWithContentsOfURL:url];
+}
+
+- (NSURL *)_urlForFile:(NSString *)file extension:(NSString *)extension {
+  return [[NSBundle bundleForClass:[self class]] URLForResource:file withExtension:extension];
 }
 
 

MacPassTests/KPKTestXmlWriting.m

@@ -28,7 +28,6 @@
   NSURL *url = [NSURL fileURLWithPath:tempFile];
   KPKTree *reloadedTree = [[KPKTree alloc] initWithContentsOfUrl:url password:password error:&error];
   STAssertNotNil(reloadedTree, @"Reloaded tree should not be nil");
-  re
 }
 
 - (NSData *)_loadTestDataBase:(NSString *)name extension:(NSString *)extension {