1# Copyright (c) 2014-2015, Intel Corporation
2# All rights reserved.
3#
4# Redistribution and use in source and binary forms, with or without modification,
5# are permitted provided that the following conditions are met:
6#
7# 1. Redistributions of source code must retain the above copyright notice, this
8# list of conditions and the following disclaimer.
9#
10# 2. Redistributions in binary form must reproduce the above copyright notice,
11# this list of conditions and the following disclaimer in the documentation and/or
12# other materials provided with the distribution.
13#
14# 3. Neither the name of the copyright holder nor the names of its contributors
15# may be used to endorse or promote products derived from this software without
16# specific prior written permission.
17#
18# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
19# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
22# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
25# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
27# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
29import logging
30
31
32class TestVectorFactory:
33
34    """
35        Factory of Test Vectors Objects
36    """
37
38    def __init__(self, criterionClasses, consoleLogger):
39        """
40            Init function
41
42            :param criterionClasses: available criterion types
43            :type criterionClasses: list of classes
44            :param consoleLogger: console log handler
45            :type consoleLogger: Handler
46        """
47        self.__criterionClasses = criterionClasses
48        self.__logger = logging.getLogger(__name__)
49        self.__logger.addHandler(consoleLogger)
50
51    def generateTestVector(self, rawCriterions=None):
52        """
53            Function invoqued to generate TestVector object
54
55            :param rawCriterions: the desired criterions state descriptions
56            :type rawCriterions: dictionnary
57        """
58
59        criterions = []
60
61        for criterionClass in self.__criterionClasses:
62            # Instanciate the criterion class requested
63            newCriterion = criterionClass()
64
65            if rawCriterions:
66                try:
67                    newCriterion.currentValue = rawCriterions[
68                        criterionClass.__name__]
69                    criterions.append(newCriterion)
70                except KeyError as e:
71                    self.__logger.debug(
72                        "Missing Criterion {}, old value keeped".format(
73                            criterionClass.__name__))
74            else:
75                # if rawCriterions is None, we create an empty criterions lists
76                newCriterion.currentValue = newCriterion.noValue
77                criterions.append(newCriterion)
78        return criterions
79