diff --git a/importer/test_main.py b/importer/test_main.py index c9b87c9f..1bda7696 100644 --- a/importer/test_main.py +++ b/importer/test_main.py @@ -382,9 +382,11 @@ def test_build_payload_care_teams(self, mock_get_resource): } validate(payload_obj["entry"][0]["request"], request_schema) + @patch("main.save_image") @patch("main.get_resource") - def test_build_payload_group(self, mock_get_resource): + def test_build_payload_group(self, mock_get_resource, mock_save_image): mock_get_resource.return_value = "1" + mock_save_image.return_value = "f374a23a-3c6a-4167-9970-b10c16a91bbd" csv_file = "csv/import/product.csv" resource_list = read_csv(csv_file)