updated test with check for previous txn

brianmc · brianmc · commit 01d42825a341 · 2016-06-02T17:52:21.000-07:00
diff --git a/test-runner.py b/test-runner.py
@@ -228,12 +228,13 @@ def update_customer_payment_profile(self):
 		modl = imp.load_source('modulename', 'CustomerProfiles/create-customer-payment-profile.py')
 		response = modl.create_customer_payment_profile(customerProfileId)
 
-		modl = imp.load_source('modulename', 'CustomerProfiles/update-customer-payment-profile.py')
-		response = modl.update_customer_payment_profile(customerProfileId, str(response.customerPaymentProfileId))
+		if hasattr(response, 'customerProfileId') == True:
+			modl = imp.load_source('modulename', 'CustomerProfiles/update-customer-payment-profile.py')
+			response = modl.update_customer_payment_profile(customerProfileId, str(response.customerPaymentProfileId))
 		
-		#delete newly create customer profile
-		modl = imp.load_source('modulename', 'CustomerProfiles/delete-customer-profile.py')
-		modl.delete_customer_profile(customerProfileId)
+		    #delete newly create customer profile
+		    modl = imp.load_source('modulename', 'CustomerProfiles/delete-customer-profile.py')
+		    modl.delete_customer_profile(customerProfileId)
 
 		return response
 
