group file functions to utils.FileUtils

tests/test_utils.py

@@ -1,11 +1,146 @@
 import os
 import shutil
+import unittest
 
 from httprunner import exception, utils
 from httprunner.compat import OrderedDict
+from httprunner.utils import FileUtils
 from tests.base import ApiServerUnittest
 
 
+class TestFileUtils(unittest.TestCase):
+
+    def test_load_yaml_file_file_format_error(self):
+        yaml_tmp_file = "tests/data/tmp.yml"
+        # create empty yaml file
+        with open(yaml_tmp_file, 'w') as f:
+            f.write("")
+
+        with self.assertRaises(exception.FileFormatError):
+            FileUtils._load_yaml_file(yaml_tmp_file)
+
+        os.remove(yaml_tmp_file)
+
+        # create invalid format yaml file
+        with open(yaml_tmp_file, 'w') as f:
+            f.write("abc")
+
+        with self.assertRaises(exception.FileFormatError):
+            FileUtils._load_yaml_file(yaml_tmp_file)
+
+        os.remove(yaml_tmp_file)
+
+
+    def test_load_json_file_file_format_error(self):
+        json_tmp_file = "tests/data/tmp.json"
+        # create empty file
+        with open(json_tmp_file, 'w') as f:
+            f.write("")
+
+        with self.assertRaises(exception.FileFormatError):
+            FileUtils._load_json_file(json_tmp_file)
+
+        os.remove(json_tmp_file)
+
+        # create empty json file
+        with open(json_tmp_file, 'w') as f:
+            f.write("{}")
+
+        with self.assertRaises(exception.FileFormatError):
+            FileUtils._load_json_file(json_tmp_file)
+
+        os.remove(json_tmp_file)
+
+        # create invalid format json file
+        with open(json_tmp_file, 'w') as f:
+            f.write("abc")
+
+        with self.assertRaises(exception.FileFormatError):
+            FileUtils._load_json_file(json_tmp_file)
+
+        os.remove(json_tmp_file)
+
+    def test_load_testcases_bad_filepath(self):
+        testcase_file_path = os.path.join(os.getcwd(), 'tests/data/demo')
+        with self.assertRaises(FileNotFoundError):
+            FileUtils.load_file(testcase_file_path)
+
+    def test_load_json_testcases(self):
+        testcase_file_path = os.path.join(
+            os.getcwd(), 'tests/data/demo_testset_hardcode.json')
+        testcases = FileUtils.load_file(testcase_file_path)
+        self.assertEqual(len(testcases), 3)
+        test = testcases[0]["test"]
+        self.assertIn('name', test)
+        self.assertIn('request', test)
+        self.assertIn('url', test['request'])
+        self.assertIn('method', test['request'])
+
+    def test_load_yaml_testcases(self):
+        testcase_file_path = os.path.join(
+            os.getcwd(), 'tests/data/demo_testset_hardcode.yml')
+        testcases = FileUtils.load_file(testcase_file_path)
+        self.assertEqual(len(testcases), 3)
+        test = testcases[0]["test"]
+        self.assertIn('name', test)
+        self.assertIn('request', test)
+        self.assertIn('url', test['request'])
+        self.assertIn('method', test['request'])
+
+    def test_load_csv_file_one_parameter(self):
+        csv_file_path = os.path.join(
+            os.getcwd(), 'tests/data/user_agent.csv')
+        csv_content = FileUtils.load_file(csv_file_path)
+        self.assertEqual(
+            csv_content,
+            [
+                {'user_agent': 'iOS/10.1'},
+                {'user_agent': 'iOS/10.2'},
+                {'user_agent': 'iOS/10.3'}
+            ]
+        )
+
+    def test_load_csv_file_multiple_parameters(self):
+        csv_file_path = os.path.join(
+            os.getcwd(), 'tests/data/account.csv')
+        csv_content = FileUtils.load_file(csv_file_path)
+        self.assertEqual(
+            csv_content,
+            [
+                {'username': 'test1', 'password': '111111'},
+                {'username': 'test2', 'password': '222222'},
+                {'username': 'test3', 'password': '333333'}
+            ]
+        )
+
+    def test_load_folder_files(self):
+        folder = os.path.join(os.getcwd(), 'tests')
+        file1 = os.path.join(os.getcwd(), 'tests', 'test_utils.py')
+        file2 = os.path.join(os.getcwd(), 'tests', 'data', 'demo_binds.yml')
+
+        files = FileUtils.load_folder_files(folder, recursive=False)
+        self.assertNotIn(file2, files)
+
+        files = FileUtils.load_folder_files(folder)
+        self.assertIn(file2, files)
+        self.assertNotIn(file1, files)
+
+        files_1 = FileUtils.load_folder_files(folder)
+        api_file = os.path.join(os.getcwd(), 'tests', 'api', 'demo.yml')
+        self.assertEqual(files_1[0], api_file)
+
+        files_2 = FileUtils.load_folder_files(folder)
+        api_file = os.path.join(os.getcwd(), 'tests', 'api', 'demo.yml')
+        self.assertEqual(files_2[0], api_file)
+        self.assertEqual(len(files_1), len(files_2))
+
+        files = FileUtils.load_folder_files("not_existed_foulder", recursive=False)
+        self.assertEqual([], files)
+
+        files = FileUtils.load_folder_files(file2, recursive=False)
+        self.assertEqual([], files)
+
+
 class TestUtils(ApiServerUnittest):
 
     def test_remove_prefix(self):
@@ -16,33 +151,6 @@ class TestUtils(ApiServerUnittest):
             "/post/123"
         )
 
-    def test_load_folder_files(self):
-        folder = os.path.join(os.getcwd(), 'tests')
-        file1 = os.path.join(os.getcwd(), 'tests', 'test_utils.py')
-        file2 = os.path.join(os.getcwd(), 'tests', 'data', 'demo_binds.yml')
-
-        files = utils.load_folder_files(folder, recursive=False)
-        self.assertNotIn(file2, files)
-
-        files = utils.load_folder_files(folder)
-        self.assertIn(file2, files)
-        self.assertNotIn(file1, files)
-
-        files_1 = utils.load_folder_files(folder)
-        api_file = os.path.join(os.getcwd(), 'tests', 'api', 'demo.yml')
-        self.assertEqual(files_1[0], api_file)
-
-        files_2 = utils.load_folder_files(folder)
-        api_file = os.path.join(os.getcwd(), 'tests', 'api', 'demo.yml')
-        self.assertEqual(files_2[0], api_file)
-        self.assertEqual(len(files_1), len(files_2))
-
-        files = utils.load_folder_files("not_existed_foulder", recursive=False)
-        self.assertEqual([], files)
-
-        files = utils.load_folder_files(file2, recursive=False)
-        self.assertEqual([], files)
-
     def test_query_json(self):
         json_content = {
             "ids": [1, 2, 3, 4],