feat: Improve rosemary maek:module command

core/environment/host.py

@@ -1,4 +1,7 @@
 import os
+from dotenv import load_dotenv
+
+load_dotenv()
 
 
 def get_host_for_testing(test_type):

docker/docker-compose.dev.yml

@@ -12,10 +12,9 @@ services:
       context: ../
       dockerfile: docker/images/Dockerfile.dev
     volumes:
-      - ./entrypoints/development_entrypoint.sh:/app/entrypoint.sh
       - ../:/app
       - /var/run/docker.sock:/var/run/docker.sock
-    command: [ "sh", "-c", "sh /app/entrypoint.sh" ]
+    command: [ "sh", "-c", "sh /app/docker/entrypoints/development_entrypoint.sh" ]
     networks:
       - uvlhub_network
 

rosemary/commands/make_module.py

@@ -50,7 +50,8 @@ def make_module(name):
         'forms.py': 'blueprint_forms.py.j2',
         'seeders.py': 'blueprint_seeders.py.j2',
         os.path.join('templates', name, 'index.html'): 'blueprint_templates_index.html.j2',
-        'tests/test_unit.py': 'blueprint_tests_test_unit.py.j2'
+        'tests/test_unit.py': 'blueprint_tests_test_unit.py.j2',
+        'tests/locustfile.py': 'blueprint_tests_locustfile.py.j2'
     }
 
     # Create the necessary directories, explicitly excluding 'tests' from the creation of subfolders.

rosemary/templates/blueprint_tests_locustfile.py.j2

@@ -0,0 +1,21 @@
+from locust import HttpUser, TaskSet, task
+from core.environment.host import get_host_for_locust_testing
+
+
+class {{ blueprint_name | pascalcase }}Behavior(TaskSet):
+    def on_start(self):
+        self.index()
+
+    @task
+    def index(self):
+        response = self.client.get("/{{ blueprint_name }}")
+
+        if response.status_code != 200:
+            print(f"{{ blueprint_name | pascalcase }} index failed: {response.status_code}")
+
+
+class {{ blueprint_name | pascalcase }}User(HttpUser):
+    tasks = [{{ blueprint_name | pascalcase }}Behavior]
+    min_wait = 5000
+    max_wait = 9000
+    host = get_host_for_locust_testing()