clean up push test

eperezrico · Aug 22, 2012 · 5454782 · 5454782
1 parent f85d9f2
commit 5454782
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/MixpanelEventSample/Unit Tests/Unit_Tests.m b/MixpanelEventSample/Unit Tests/Unit_Tests.m
@@ -187,10 +187,9 @@ - (void)testPushFormat {
     STAssertTrue([dict objectForKey:@"$token"] != nil, @"token not set");
 
     NSDictionary *operation = [dict objectForKey:@"$union"];
-    NSArray *ios_devices = [operation objectForKey:@"$ios_devices"];
+    NSDictionary *targetOperation = [NSDictionary dictionaryWithObjectsAndKeys:[NSArray arrayWithObject:@"74657374"], @"$ios_devices", nil];
 
-    STAssertTrue(ios_devices.count == 1, @"ios_devices should contain exactly 1 token");
-    STAssertTrue([(NSString*)[ios_devices objectAtIndex:0] isEqualToString:@"74657374"], @"push format is incorrect");
+    STAssertTrue([operation isEqualToDictionary:targetOperation], @"push format is incorrect");
 }
 
 @end