qase-python-commons 3.1.9__py3-none-any.whl → 4.1.9__py3-none-any.whl

qase/commons/models/step.py

@@ -5,6 +5,7 @@ from enum import Enum
 from typing import Optional, Union, Dict, List, Type
 from .attachment import Attachment
 from .basemodel import BaseModel
+from .. import QaseUtils
 
 
 class StepType(Enum):
@@ -20,6 +21,7 @@ class StepTextData(BaseModel):
     def __init__(self, action: str, expected_result: Optional[str] = None):
         self.action = action
         self.expected_result = expected_result
+        self.input_data = None
 
 
 class StepAssertData(BaseModel):
@@ -30,10 +32,11 @@ class StepAssertData(BaseModel):
 
 
 class StepGherkinData(BaseModel):
-    def __init__(self, keyword: str, name: str, line: int):
+    def __init__(self, keyword: str, name: str, line: int, data: Optional[str] = None):
         self.keyword = keyword
         self.name = name
         self.line = line
+        self.data = data
 
 
 class StepRequestData(BaseModel):
@@ -42,10 +45,20 @@ class StepRequestData(BaseModel):
         self.response_body = None
         self.status_code = None
         if isinstance(request_body, bytes):
-            request_body = request_body.decode('utf-8')
+            try:
+                request_body = request_body.decode('utf-8')
+            except UnicodeDecodeError:
+                # For binary data (like file uploads), keep as base64 encoded string
+                import base64
+                request_body = base64.b64encode(request_body).decode('ascii')
         self.request_body = request_body
         if isinstance(request_headers, bytes):
-            request_headers = request_headers.decode('utf-8')
+            try:
+                request_headers = request_headers.decode('utf-8')
+            except UnicodeDecodeError:
+                # For binary headers, keep as base64 encoded string
+                import base64
+                request_headers = base64.b64encode(request_headers).decode('ascii')
         self.request_headers = request_headers
         self.request_method = request_method
         self.request_url = request_url
@@ -55,16 +68,33 @@ class StepRequestData(BaseModel):
         self.status_code = status_code
 
         if isinstance(response_body, bytes):
-            response_body = response_body.decode('utf-8')
+            try:
+                response_body = response_body.decode('utf-8')
+            except UnicodeDecodeError:
+                # For binary data (like file downloads), keep as base64 encoded string
+                import base64
+                response_body = base64.b64encode(response_body).decode('ascii')
         self.response_body = response_body
         if isinstance(response_headers, bytes):
-            response_headers = response_headers.decode('utf-8')
+            try:
+                response_headers = response_headers.decode('utf-8')
+            except UnicodeDecodeError:
+                # For binary headers, keep as base64 encoded string
+                import base64
+                response_headers = base64.b64encode(response_headers).decode('ascii')
         self.response_headers = response_headers
 
 
 class StepDbQueryData(BaseModel):
-    def __init__(self, query: str, expected_result: str):
+    def __init__(self, query: str, expected_result: str = None, 
+                 database_type: str = None, execution_time: float = None,
+                 rows_affected: int = None, connection_info: str = None):
         self.query = query
+        self.expected_result = expected_result
+        self.database_type = database_type
+        self.execution_time = execution_time
+        self.rows_affected = rows_affected
+        self.connection_info = connection_info
 
 
 class StepSleepData(BaseModel):
@@ -74,21 +104,25 @@ class StepSleepData(BaseModel):
 
 class StepExecution(BaseModel):
     def __init__(self, status: Optional[str] = 'untested', end_time: int = 0, duration: int = 0):
-        self.start_time = time.time()
+        self.start_time = QaseUtils.get_real_time()
         self.status = status
         self.end_time = end_time
         self.duration = duration
+        self.attachments = []
 
     def set_status(self, status: Optional[str]):
-        if status in ['passed', 'failed', 'skipped', 'blocked', 'untested']:
+        if status in ['passed', 'failed', 'skipped', 'blocked', 'untested', 'invalid']:
             self.status = status
         else:
-            raise ValueError('Step status must be one of: passed, failed, skipped, blocked, untested')
+            raise ValueError('Step status must be one of: passed, failed, skipped, blocked, untested, invalid')
 
     def complete(self):
-        self.end_time = time.time()
+        self.end_time = QaseUtils.get_real_time()
         self.duration = int((self.end_time - self.start_time) * 1000)
 
+    def add_attachment(self, attachment: Attachment):
+        self.attachments.append(attachment)
+
 
 class Step(BaseModel):
     def __init__(self,
@@ -107,7 +141,6 @@ class Step(BaseModel):
         self.data = data
         self.parent_id = parent_id
         self.execution = StepExecution()
-        self.attachments = []
         self.steps = []
 
     def set_parent_id(self, parent_id: Optional[str]):
@@ -132,4 +165,4 @@ class Step(BaseModel):
         self.steps = steps
 
     def add_attachment(self, attachment: Attachment):
-        self.attachments.append(attachment)
+        self.execution.add_attachment(attachment)

qase/commons/profilers/__init__.py

@@ -1,10 +1,11 @@
 from .network import NetworkProfiler, NetworkProfilerSingleton
 from .sleep import SleepProfiler
-from .db import DbProfiler
+from .db import DatabaseProfiler, DatabaseProfilerSingleton
 
 __all__ = [
     NetworkProfiler,
     NetworkProfilerSingleton,
     SleepProfiler,
-    DbProfiler
-]
+    DatabaseProfiler,
+    DatabaseProfilerSingleton
+]