load generation

load-gen/load-gen.sh

@@ -4,7 +4,7 @@
 docker run \
     -it \
     --rm \
-    --network robotshop_robot-shop \
-    -e 'HOST=http://web:8080' \
+    --network=host \
+    -e 'HOST=http://localhost:8080' \
     steveww/rs-load
 

load-gen/robot-shop.py

@@ -1,4 +1,5 @@
 from locust import HttpLocust, TaskSet, task
+from random import choice
 
 class UserBehavior(TaskSet):
     def on_start(self):
@@ -6,13 +7,56 @@ class UserBehavior(TaskSet):
         print('Starting')
 
     @task
-    def index(self):
-        self.client.get("/")
+    def login(self):
+        credentials = {
+                'name': 'user',
+                'password': 'password'
+                }
+        res = self.client.post('/api/user/login', json=credentials)
+        print('login {}'.format(res.status_code))
+
 
     @task
-    def user(self):
-        res = self.client.get("/api/user/uniqueid")
-        print('User {}'.format(res.content))
+    def load(self):
+        self.client.get('/')
+        user = self.client.get('/api/user/uniqueid').json()
+        uniqueid = user['uuid']
+        print('User {}'.format(uniqueid))
+
+        self.client.get('/api/catalogue/categories')
+        # all products in catalogue
+        products = self.client.get('/api/catalogue/products').json()
+        for i in range(2):
+            item = None
+            while True:
+                item = choice(products)
+                if item['instock'] != 0:
+                    break
+
+            self.client.get('/api/catalogue/product/{}'.format(item['sku']))
+            self.client.get('/api/cart/add/{}/{}/1'.format(uniqueid, item['sku']))
+
+        cart = self.client.get('/api/cart/cart/{}'.format(uniqueid)).json()
+        item = choice(cart['items'])
+        self.client.get('/api/cart/update/{}/{}/2'.format(uniqueid, item['sku']))
+
+        # country codes
+        code = choice(self.client.get('/api/shipping/codes').json())
+        city = choice(self.client.get('/api/shipping/cities/{}'.format(code['code'])).json())
+        print('code {} city {}'.format(code, city))
+        shipping = self.client.get('/api/shipping/calc/{}'.format(city['uuid'])).json()
+        shipping['location'] = '{} {}'.format(code['name'], city['name'])
+        print('Shipping {}'.format(shipping))
+        # POST
+        cart = self.client.post('/api/shipping/confirm/{}'.format(uniqueid), json=shipping).json()
+        print('Final cart {}'.format(cart))
+
+        order = self.client.post('/api/payment/pay/{}'.format(uniqueid), json=cart).json()
+        print('Order {}'.format(order))
+
+        # delete cart
+        self.client.delete('/api/cart/cart/{}'.format(uniqueid))
+
 
 class WebsiteUser(HttpLocust):
     task_set = UserBehavior