Create Explicit Test mode API calls

Example/StripeExample/Constants.h

@@ -9,6 +9,7 @@
 #import <Foundation/Foundation.h>
 
 extern NSString * const StripePublishableKey;
+extern BOOL const StripeTestMode;
 extern NSString * const ParseApplicationId;
 extern NSString * const ParseClientKey;
 

Example/StripeExample/Constants.m

@@ -12,6 +12,7 @@
 
 // This can be found at https://dashboard.stripe.com/account/apikeys
 NSString * const StripePublishableKey = @"pk_YT1CEhhujd0bklb2KGQZiaL3iTzj3"; // TODO: replace nil with your own value
+BOOL const StripeTestMode = TRUE; // TODO: replace with FALSE for production/live
 
 // These can be found at https://www.parse.com/apps/stripe-test/edit#app_keys
 NSString * const ParseApplicationId = nil; // TODO: replace nil with your own value

Example/StripeExample/ViewController.m

@@ -12,7 +12,7 @@
 #import "Stripe.h"
 #import "Constants.h"
 
-@interface ViewController()<PKPaymentAuthorizationViewControllerDelegate>
+@interface ViewController()<PKPaymentAuthorizationViewControllerDelegate, STPTestPaymentAuthorizationViewControllerDelegate>
 @end
 
 @implementation ViewController
@@ -24,7 +24,13 @@ - (IBAction)beginPayment:(id)sender {
                                                                            currency:@"USD"
                                                                         description:@"Premium Llama Food"];
     if ([Stripe canSubmitPaymentRequest:paymentRequest]) {
-        UIViewController *paymentController = [Stripe paymentControllerWithRequest:paymentRequest delegate:self];
+        UIViewController *paymentController;
+
+        if (StripeTestMode)
+            paymentController = [Stripe testPaymentControllerWithRequest:paymentRequest delegate:self];
+        else
+            paymentController = [Stripe paymentControllerWithRequest:paymentRequest delegate:self];
+
         [self presentViewController:paymentController animated:YES completion:nil];
     }
     else {
@@ -38,6 +44,12 @@ - (void)paymentAuthorizationViewController:(PKPaymentAuthorizationViewController
     [self handlePaymentAuthorizationWithPayment:payment completion:completion];
 }
 
+- (void)testPaymentAuthorizationViewController:(STPTestPaymentAuthorizationViewController *)controller
+                       didAuthorizePayment:(PKPayment *)payment
+                                completion:(void (^)(PKPaymentAuthorizationStatus))completion {
+    [self handlePaymentAuthorizationWithPayment:payment completion:completion];
+}
+
 - (void)handlePaymentAuthorizationWithPayment:(PKPayment *)payment
                                    completion:(void (^)(PKPaymentAuthorizationStatus))completion {
     [Stripe createTokenWithPayment:payment
@@ -88,4 +100,8 @@ - (void)paymentAuthorizationViewControllerDidFinish:(PKPaymentAuthorizationViewC
     [self dismissViewControllerAnimated:YES completion:nil];
 }
 
+- (void)testPaymentAuthorizationViewControllerDidFinish:(UIViewController *)controller {
+    [self dismissViewControllerAnimated:YES completion:nil];
+}
+
 @end

Stripe/STPTestPaymentAuthorizationViewController.h

@@ -11,9 +11,20 @@
 #import <UIKit/UIKit.h>
 #import <PassKit/PassKit.h>
 
+@protocol STPTestPaymentAuthorizationViewControllerDelegate <NSObject>
+
+@required
+-(void)testPaymentAuthorizationViewController:(UIViewController *)controller
+                          didAuthorizePayment:(PKPayment *)payment
+                                   completion:(void (^)(PKPaymentAuthorizationStatus))completion;
+
+-(void)testPaymentAuthorizationViewControllerDidFinish:(UIViewController *)controller;
+@end
+
+
 @interface STPTestPaymentAuthorizationViewController : UIViewController
 
-@property(nonatomic, assign)id<PKPaymentAuthorizationViewControllerDelegate>delegate;
+@property(nonatomic, assign)id<STPTestPaymentAuthorizationViewControllerDelegate>delegate;
 
 @end
 

Stripe/STPTestPaymentAuthorizationViewController.m

@@ -41,11 +41,11 @@ - (void)makePaymentWithCardNumber:(NSString *)cardNumber {
     PKPaymentAuthorizationViewController *auth = (PKPaymentAuthorizationViewController *)self;
 
     [self.activityIndicator startAnimating];
-    [self.delegate paymentAuthorizationViewController:(PKPaymentAuthorizationViewController *)auth
+    [self.delegate testPaymentAuthorizationViewController:(STPTestPaymentAuthorizationViewController *)auth
                                   didAuthorizePayment:payment
                                            completion:^(PKPaymentAuthorizationStatus status) {
                                                [self.activityIndicator stopAnimating];
-                                               [self.delegate paymentAuthorizationViewControllerDidFinish:auth];
+                                               [self.delegate testPaymentAuthorizationViewControllerDidFinish:auth];
                                            }];
 }
 
@@ -67,7 +67,7 @@ - (void)actionSheet:(UIActionSheet *)actionSheet clickedButtonAtIndex:(NSInteger
 }
 
 - (void)actionSheetCancel:(UIActionSheet *)actionSheet {
-    [self.delegate paymentAuthorizationViewControllerDidFinish:(PKPaymentAuthorizationViewController *)self];
+    [self.delegate testPaymentAuthorizationViewControllerDidFinish:(PKPaymentAuthorizationViewController *)self];
 }
 
 @end

Stripe/Stripe.h

@@ -11,6 +11,7 @@
 #import "StripeError.h"
 #import "STPCard.h"
 #import "STPToken.h"
+#import "STPTestPaymentAuthorizationViewController.h"
 
 FOUNDATION_EXPORT NSString *const kStripeiOSVersion; // Version of this library.
 
@@ -51,6 +52,8 @@ typedef void (^STPCompletionBlock)(STPToken *token, NSError *error);
                                                   currency:(NSString *)currency
                                                description:(NSString *)description;
 
++ (UIViewController *)testPaymentControllerWithRequest:(PKPaymentRequest *)request
+                                          delegate:(id<STPTestPaymentAuthorizationViewControllerDelegate>)delegate;
 + (UIViewController *)paymentControllerWithRequest:(PKPaymentRequest *)request
                                           delegate:(id<PKPaymentAuthorizationViewControllerDelegate>)delegate;
 

Stripe/Stripe.m

@@ -324,13 +324,16 @@ + (PKPaymentRequest *)paymentRequestWithMerchantIdentifier:(NSString *)merchantI
     return paymentRequest;
 }
 
+
++ (UIViewController *)testPaymentControllerWithRequest:(PKPaymentRequest *)request
+                                          delegate:(id<STPTestPaymentAuthorizationViewControllerDelegate>)delegate {
+    STPTestPaymentAuthorizationViewController *test = [STPTestPaymentAuthorizationViewController new];
+    test.delegate = delegate;
+    return test;
+}
+
 + (UIViewController *)paymentControllerWithRequest:(PKPaymentRequest *)request
                                           delegate:(id<PKPaymentAuthorizationViewControllerDelegate>)delegate {
-    if ([self isSimulatorBuild]) {
-        STPTestPaymentAuthorizationViewController *test = [STPTestPaymentAuthorizationViewController new];
-        test.delegate = delegate;
-        return test;
-    }
     PKPaymentAuthorizationViewController *auth = [[PKPaymentAuthorizationViewController alloc] initWithPaymentRequest:request];
     auth.delegate = delegate;
     return auth;