aboutsummaryrefslogtreecommitdiffstats
path: root/src/onaptests/scenario/cds_blueprint_enrichment.py
blob: a16169b9ce7d38cf5631b27ffa0b05a5f4ad7f08 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#!/usr/bin/env python
"""Simple CDS blueprint erichment test scenario."""

import logging
import time

from onapsdk.configuration import settings
from onapsdk.exceptions import SDKException
from onaptests.scenario.scenario_base import ScenarioBase
from onaptests.steps.onboard.cds import CbaEnrichStep
from onaptests.utils.exceptions import OnapTestException


class CDSBlueprintEnrichment(ScenarioBase):
    """Enrich simple blueprint using CDS blueprintprocessor."""

    __logger = logging.getLogger()

    def __init__(self, **kwargs):
        """Init CDS blueprint enrichment use case."""
        super().__init__('basic_cds', **kwargs)
        self.test = CbaEnrichStep(
            cleanup=settings.CLEANUP_FLAG)
        self.start_time = None
        self.stop_time = None
        self.result = 0

    def run(self):
        self.__logger.debug("CDS blueprint enrichment run")
        self.start_time = time.time()
        try:
            self.test.execute()
            if settings.CLEANUP_FLAG:
                self.test.cleanup()
            self.result = 100
        except OnapTestException as exc:
            self.result = 0
            self.__logger.exception(exc.error_message)
        except SDKException:
            self.result = 0
            self.__logger.exception("SDK Exception")
        finally:
            self.stop_time = time.time()

    def clean(self):
        """Clean Additional resources if needed."""
        self.__logger.info("Generate Test report")
        self.test.reports_collection.generate_report()