Update Unit-tests

Updating unit-tests by adding trainingjobid argument in order to resolve
'tox -e code' error

Issue-id: AIMLFW-314
Change-Id: I03c88ede91943ee74bf9046350e23456ca6d0e6f
Signed-off-by: ashishj1729 <jain.ashish@samsung.com>

diff --git a/.github/workflows/gerrit-verify.yaml b/.github/workflows/gerrit-verify.yaml
index 78ed11e..ea40b97 100644 (file)
--- a/.github/workflows/gerrit-verify.yaml
+++ b/.github/workflows/gerrit-verify.yaml
@@ -82,6 +82,7 @@ jobs:
       GERRIT_REFSPEC: ${{ inputs.GERRIT_REFSPEC }}
       TOX_ENVS: '["code"]'
       PYTHON_VERSION: '3.10'
+      PARALLEL: 1
 
   component-test:
     needs: prepare

diff --git a/test/test_main.py b/test/test_main.py
index d003093..6555075 100644 (file)
--- a/test/test_main.py
+++ b/test/test_main.py
@@ -15,7 +15,6 @@
 #   limitations under the License.
 #
 # ==================================================================================
-
 import sys
 import os
 from dotenv import load_dotenv
@@ -46,7 +45,7 @@ class Test_feature_groups:
         
 
     def test_feature_group(self):
-        request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
+        request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, "trainingjob_id": "52", 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
         response = self.client.post("/feature-groups", data=json.dumps(request_json), content_type='application/json')
         empyty_tasks()
         assert response.content_type ==  'application/json'
@@ -64,9 +63,10 @@ class Test_task_status:
     def setup_method(self):
         self.client = main.app.test_client(self)
      
-    def test_task_status(self):  
+    def test_task_status(self):
+        trainingjob_id = 1  
         request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
-        main.task_map["unittest_task"] = Task(request_json ,"Accepted")
+        main.task_map["unittest_task"] = Task(request_json ,"Accepted", trainingjob_id)
         response = self.client.get("/task-status/unittest_task")
         main.task_map.clear()
         assert response.content_type ==  'application/json'
@@ -75,7 +75,8 @@ class Test_task_status:
     #Error test
     def test_negative_task_status(self):  
         request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
-        main.task_map["unittest_task"] = Task(request_json ,"Error")
+        trainingjob_id = 1
+        main.task_map["unittest_task"] = Task(request_json ,"Error", trainingjob_id)
         response = self.client.get("/task-status/unittest_task")
         main.task_map.clear()
        
@@ -95,8 +96,9 @@ class Test_delete_task_status:
         self.client = main.app.test_client(self)
 
     def test_delete_task_status(self):
+        trainingjob_id = 1
         request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
-        main.task_map["unittest_task"] = Task(request_json ,"Accepted")
+        main.task_map["unittest_task"] = Task(request_json ,"Accepted", trainingjob_id)
         response = self.client.delete("/delete-task-status/unittest_task")
         main.task_map.clear()
         assert response.content_type ==  'application/json'
@@ -116,7 +118,7 @@ class Test_async_code_worker:
     @patch('main.factory.get_batch_pipeline')
     def test_negative_async_code_worker_1(self,mock1,mock2):
         
-        main.infinte_loop_config["infinte_run"]= "True"
+        main.infinte_loop_config["infinte_run"]= "False"
         main.infinte_loop_config["unit_test_mode"]= "True"
         request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
         response = self.client.post("/feature-groups", data=json.dumps(request_json), content_type='application/json')
@@ -127,7 +129,7 @@ class Test_async_code_worker:
     #error
     def test_negative_async_code_worker_2(self):
         
-        main.infinte_loop_config["infinte_run"]= "True"
+        main.infinte_loop_config["infinte_run"]= "False"
         main.infinte_loop_config["unit_test_mode"]= "True"
         request_json = {'source': {'InfluxSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
         response = self.client.post("/feature-groups", data=json.dumps(request_json), content_type='application/json')
@@ -137,7 +139,7 @@ class Test_async_code_worker:
     #error with Cassandra Source
     def test_negative_async_code_worker_3(self):
 
-        main.infinte_loop_config["infinte_run"]= "True"
+        main.infinte_loop_config["infinte_run"]= "False"
         main.infinte_loop_config["unit_test_mode"]= "True"
         request_json = {'source': {'CassandraSource': {'query': 'from(bucket:"UEData") |> range(start: 0, stop: now()) |> filter(fn: (r) => r._measurement == "liveCell") |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")'}}, 'transform': [{'operation': 'SQLTransform', 'FeatureList': '*', 'SQLFilter': ''}], 'sink': {'CassandraSink': {'CollectionName': 'last_check3'}}}
         response = self.client.post("/feature-groups", data=json.dumps(request_json), content_type='application/json')