Merge pull request #47 from kishikawakatsumi/tests

Add tests
kishikawakatsumi · Jan 12, 2015 · 6511518 · 6511518
2 parents f20551d + 6a7a35f
commit 6511518
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/Lib/UICKeyChainStoreTests/UICKeyChainStoreTests.m b/Lib/UICKeyChainStoreTests/UICKeyChainStoreTests.m
@@ -804,15 +804,34 @@ - (void)testSetDataClassMethod
     XCTAssertEqualObjects([UICKeyChainStore dataForKey:@"username"], usernameData, @"stored username");
     XCTAssertEqualObjects([UICKeyChainStore dataForKey:@"password"], passwordData, @"stored password");
 
-    [UICKeyChainStore removeItemForKey:@"username" service:[UICKeyChainStore defaultService]];
+    [UICKeyChainStore removeItemForKey:@"username"];
     XCTAssertNil([UICKeyChainStore dataForKey:@"username"], @"removed username");
     XCTAssertEqualObjects([UICKeyChainStore dataForKey:@"password"], passwordData, @"left password");
 
-    [UICKeyChainStore removeItemForKey:@"password" service:[UICKeyChainStore defaultService]];
+    [UICKeyChainStore removeItemForKey:@"password"];
     XCTAssertNil([UICKeyChainStore dataForKey:@"username"], @"removed username");
     XCTAssertNil([UICKeyChainStore dataForKey:@"password"], @"removed password");
 }
 
+- (void)testSetDataWithServiceClassMethod
+{
+    NSData *usernameData = [@"kishikawakatsumi" dataUsingEncoding:NSUTF8StringEncoding];
+    NSData *passwordData = [@"password1234" dataUsingEncoding:NSUTF8StringEncoding];
+
+    [UICKeyChainStore setData:usernameData forKey:@"username" service:@"Twitter"];
+    [UICKeyChainStore setData:passwordData forKey:@"password" service:@"Twitter"];
+    XCTAssertEqualObjects([UICKeyChainStore dataForKey:@"username" service:@"Twitter"], usernameData, @"stored username");
+    XCTAssertEqualObjects([UICKeyChainStore dataForKey:@"password" service:@"Twitter"], passwordData, @"stored password");
+
+    [UICKeyChainStore removeItemForKey:@"username" service:@"Twitter"];
+    XCTAssertNil([UICKeyChainStore dataForKey:@"username" service:@"Twitter"], @"removed username");
+    XCTAssertEqualObjects([UICKeyChainStore dataForKey:@"password" service:@"Twitter"], passwordData, @"left password");
+
+    [UICKeyChainStore removeItemForKey:@"password" service:@"Twitter"];
+    XCTAssertNil([UICKeyChainStore dataForKey:@"username" service:@"Twitter"], @"removed username");
+    XCTAssertNil([UICKeyChainStore dataForKey:@"password" service:@"Twitter"], @"removed password");
+}
+
 #if TARGET_OS_IPHONE
 - (void)testSetDataClassMethodAndError
 {