Browse Source

Pytest NestedLoops bug resolved and pytest increased

Akash 3 years ago
parent
commit
396c693179
2 changed files with 46 additions and 6 deletions
  1. 3 4
      baangt/TestDataGenerator/TestDataGenerator.py
  2. 43 2
      tests/test_SendReports.py

+ 3 - 4
baangt/TestDataGenerator/TestDataGenerator.py

@@ -638,15 +638,14 @@ class TestDataGenerator:
         for sheet in sheet_lis:
             sheet_df[sheet] = self.get_str_sheet(wb, sheet)
             sheet_df[sheet].fillna("", inplace=True)
+        if return_json:
+            for df in sheet_df.keys():
+                sheet_df[df] = json.loads(sheet_df[df].to_json(orient="records"))
         if sheet_name == "":
             base_sheet = sheet_df[sheet_lis[0]]
         else:
             assert sheet_name in sheet_df, f"Excel file doesn't contain {sheet_name} sheet. Please recheck."
             base_sheet = sheet_df[sheet_name]
-        if return_json:
-            base_sheet = json.loads(base_sheet.to_json("records"))
-            for df in sheet_df:
-                sheet_df[df] = json.loads(sheet_df[df].to_json("records"))
         return sheet_df, base_sheet
 
     @staticmethod

+ 43 - 2
tests/test_SendReports.py

@@ -1,9 +1,9 @@
 from baangt.base.SendReports import Sender
 from pathlib import Path
 import configparser
-import requests
-import json
 import os
+from unittest.mock import patch
+from atlassian import Confluence
 
 
 def readConfig():
@@ -25,3 +25,44 @@ def test_telegram():
     messages = send_stats.sendTelegram(test=True)
     text = subject + "\n\n" + body
     assert text in messages
+
+
+def fake_response(text):
+    return '{"test@pytest": "Success"}'
+
+@patch.object(Confluence, "create_page")
+@patch.object(Confluence, "attach_file")
+def test_confluence(mock_attach, mock_create):
+    send_stats.globalSettings["Confluence-Base-Url"] = "xxxx"
+    send_stats.globalSettings["Confluence-Space"] = "xxxx"
+    send_stats.globalSettings["Confluence-Pagetitle"] = "xxxx"
+    send_stats.globalSettings["Confluence-Username"] = "xxxx"
+    send_stats.globalSettings["Confluence-Password"] = "xxxx"
+    send_stats.globalSettings["Confluence-Rootpage"] = "xxxx"
+    send_stats.globalSettings["Confluence-Remove_Headers"] = "xxxx"
+    send_stats.globalSettings["Confluence-Uploadoriginalfile"] = "xxxx"
+    send_stats.globalSettings["Confluence-Createsubpagesforeachxxentries"] = 11
+    send_stats.sendConfluence()
+    assert 1 == 1
+
+
+@patch("json.loads", fake_response)
+@patch("requests.post")
+def test_SendMail(mock_request):
+    send_stats.defaultSettings["SendMailTo"] = "test@pytest"
+    send_stats.sendMail()
+    assert mock_request.call_count == 1
+
+
+@patch("pymsteams.connectorcard")
+def test_SendMsTeams(mock_conn):
+    send_stats.defaultSettings["MsWebHook"] = "xxxx"
+    send_stats.sendMsTeam()
+    assert mock_conn.call_count == 1
+
+
+@patch("slack_webhook.Slack")
+def test_SendSlack(mock_conn):
+    send_stats.defaultSettings["SlackWebHook"] = "xxxx"
+    send_stats.sendSlack()
+    assert 1 == 1