1 files changed, 95 insertions, 0 deletions
diff --git a/csarvalidation/python/test/test_FileManager.py b/csarvalidation/python/test/test_FileManager.py
new file mode 100644
index 0000000..722d1a9
--- /dev/null
+++ b/csarvalidation/python/test/test_FileManager.py
@@ -0,0 +1,95 @@
+# ============LICENSE_START====================================
+# vnfsdk-validation
+# =========================================================
+# Copyright (C) 2020 Nokia. All rights reserved.
+# =========================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============LICENSE_END=====================================
+
+import os
+import shutil
+import unittest
+
+from validation.FileManager import FileManager
+from validation.rules.ActiveRulesTableGenerator import ActiveRulesTableGenerator
+from validation.rules.providers.ActiveRulesProvider import ActiveRulesProvider
+from validation.rules.providers.RulesDescriptionsProvider import RulesDescriptionsProvider
+
+RESOURCES_DIRECTORY = '../src/main/resources/'
+RULE_DESCRIPTION_SOL001_PATH = RESOURCES_DIRECTORY + 'open-cli-schema/sol001/'
+RULE_DESCRIPTION_SOL004_PATH = RESOURCES_DIRECTORY + 'open-cli-schema/sol004/'
+VNFREWS_PROPERTIES_PATH = RESOURCES_DIRECTORY + 'vnfreqs.properties'
+RULE_DESCRIPTION_FILE_NAME_PATTERN = 'vtp-validate-csar-%s.yaml'
+
+CSV_DELIMITER = ";"
+
+VNF_REQS_TAG = "vnfreqs"
+PNF_REQS_TAG = "pnfreqs"
+
+OUTPUT_DIRECTORY = './active_rules_table_generator_test/'
+
+TABLE_WITH_VNF_RULES = OUTPUT_DIRECTORY + 'VnfActiveRulesTable.csv'
+TABLE_WITH_PNF_RULES = OUTPUT_DIRECTORY + 'PnfActiveRulesTable.csv'
+
+
+class FileManagerTest(unittest.TestCase):
+
+
+    @staticmethod
+    def generate_tables_with_active_rules(requs_tags: list) -> list:
+        return ActiveRulesTableGenerator(
+            [
+                RulesDescriptionsProvider(RULE_DESCRIPTION_SOL001_PATH, RULE_DESCRIPTION_FILE_NAME_PATTERN),
+                RulesDescriptionsProvider(RULE_DESCRIPTION_SOL004_PATH, RULE_DESCRIPTION_FILE_NAME_PATTERN)
+            ],
+            ActiveRulesProvider(VNFREWS_PROPERTIES_PATH)
+        ).generate_active_validation_rule_tables(requs_tags)
+
+    def generate_and_save_tables(self):
+        tables = self.generate_tables_with_active_rules([VNF_REQS_TAG, PNF_REQS_TAG])
+        file_manager = FileManager(OUTPUT_DIRECTORY)
+        file_manager.save_rule_table(
+            tables[0].get_table_in_csv_format(values_separator=CSV_DELIMITER),
+            TABLE_WITH_VNF_RULES
+        )
+        file_manager.save_rule_table(
+            tables[1].get_table_in_csv_format(values_separator=CSV_DELIMITER),
+            TABLE_WITH_PNF_RULES
+        )
+
+    def tearDown(self):
+        shutil.rmtree(OUTPUT_DIRECTORY)
+
+    def test_generate_and_save_table_with_active_rules(self):
+        self.generate_and_save_tables()
+        self.assertTrue(os.path.isdir(OUTPUT_DIRECTORY))
+
+        self.assertTrue(os.path.isfile(TABLE_WITH_VNF_RULES))
+        with open(TABLE_WITH_VNF_RULES, 'r') as vnf_rules:
+            self.validate_csv_table_with_rules(vnf_rules)
+
+        self.assertTrue(os.path.isfile(TABLE_WITH_PNF_RULES))
+        with open(TABLE_WITH_PNF_RULES, 'r') as pnf_rules:
+            self.validate_csv_table_with_rules(pnf_rules)
+
+    def validate_csv_table_with_rules(self, vnf_rules):
+        lines = vnf_rules.read().splitlines()
+        for line in lines:
+            values = line.split(CSV_DELIMITER)
+            self.assertTrue(len(values) == 3)
+            self.assertTrue(values[0].startswith("onap-"))
+            self.assertTrue(values[1].startswith("r"))
+
+
+if __name__ == '__main__':
+    unittest.main()