From f6ec88d24b5651511311b89e21584b7b00cc4353 Mon Sep 17 00:00:00 2001
From: Christopher DeCairos <cade@mozillafoundation.org>
Date: Tue, 24 Sep 2019 16:52:09 -0400
Subject: [PATCH] update gitignore, use a test braintree nonce

---
 .gitignore     |  3 ++-
 stress_test.py | 24 +++++++++++++++---------
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/.gitignore b/.gitignore
index ca5e3f2..edd6eb8 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 venv/
-.idea/
\ No newline at end of file
+.idea/
+__pycache__/
\ No newline at end of file
diff --git a/stress_test.py b/stress_test.py
index a865ce5..52b47d8 100644
--- a/stress_test.py
+++ b/stress_test.py
@@ -1,11 +1,13 @@
 import random
-import urllib.parse
 
 from locust import HttpLocust, TaskSet, seq_task
 from pyquery import PyQuery
 
 
 class DonorBehaviour(TaskSet):
+    # Default host to use, override with --host on the command line
+    host = 'http://localhost:8000'
+
     def __init__(self, parent):
         super().__init__(parent)
         self.donation_payload = {
@@ -19,11 +21,15 @@ class DonorBehaviour(TaskSet):
             'amount': 50,
             'landing_url': 'https://donate-wagtail-staging.herokuapp.com/en-CA/',
             'project': 'mozillafoundation',
-            'campaign_id': ''
+            'campaign_id': '',
+            # https://developers.braintreepayments.com/reference/general/testing/python#nonces-representing-cards
+            'braintree_nonce': 'fake-valid-nonce'
+        }
+        self.donation_params = {
+            'currency': 'cad',
+            'source_page_id': 3,
+            'amount': 50
         }
-
-    # Default host to use, override with --host on the command line
-    host = 'http://localhost:8000'
 
     @seq_task(1)
     def load_donate_form(self):
@@ -31,16 +37,16 @@ class DonorBehaviour(TaskSet):
 
     @seq_task(2)
     def load_payment_information_form(self):
-        resp = self.client.get(f'/en-CA/card/single/?source_page_id={self.d_source_page_id}&currency={self.d_currency}'
-                               f'&amount={self.d_amount}')
+        resp = self.client.get(f'/en-CA/card/single/', params=self.donation_params)
         pq = PyQuery(resp.content)
         self.donation_payload['csrfmiddlewaretoken'] = pq('input[name=csrfmiddlewaretoken]').val()
-        self.donation_payload['braintree_nonce'] = pq('input#id_braintree_nonce').val()
 
     @seq_task(3)
     def make_payment(self):
+        print('submitting payment')
         headers = {'content-type': 'application/x-www-form-encoded'}
-        resp = self.client.post('/en-US/card/single', data=self.donation_payload, headers=headers)
+        resp = self.client.post('/en-US/card/single/', params=self.donation_params,
+                                data=self.donation_payload, headers=headers)
 
 
 class DonorUser(HttpLocust):