[mirror_edk2.git] / BaseTools / Source / Python / UPT / UnitTest / CommentGeneratingUnitTest.py

## @file\r
# This file contain unit test for CommentParsing\r
#\r
# Copyright (c) 2011 - 2018, Intel Corporation. All rights reserved.<BR>\r
#\r
# SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
import os\r
import unittest\r
\r
import Logger.Log as Logger\r
from GenMetaFile.GenInfFile import GenGuidSections\r
from GenMetaFile.GenInfFile import GenProtocolPPiSections\r
from GenMetaFile.GenInfFile import GenPcdSections\r
from GenMetaFile.GenInfFile import GenSpecialSections\r
from Library.CommentGenerating import GenGenericCommentF\r
from Library.CommentGenerating import _GetHelpStr\r
from Object.POM.CommonObject import TextObject\r
from Object.POM.CommonObject import GuidObject\r
from Object.POM.CommonObject import ProtocolObject\r
from Object.POM.CommonObject import PpiObject\r
from Object.POM.CommonObject import PcdObject\r
from Object.POM.ModuleObject import HobObject\r
\r
from Library.StringUtils import GetSplitValueList\r
from Library.DataType import TAB_SPACE_SPLIT\r
from Library.DataType import TAB_LANGUAGE_EN_US\r
from Library.DataType import TAB_LANGUAGE_ENG\r
from Library.DataType import ITEM_UNDEFINED\r
from Library.DataType import TAB_INF_FEATURE_PCD\r
from Library import GlobalData\r
from Library.Misc import CreateDirectory\r
\r
#\r
# Test _GetHelpStr\r
#\r
class _GetHelpStrTest(unittest.TestCase):\r
    def setUp(self):\r
        pass\r
\r
    def tearDown(self):\r
        pass\r
\r
    #\r
    # Normal case1: have one help text object with Lang = 'en-US'\r
    #\r
    def testNormalCase1(self):\r
        HelpStr = 'Hello world'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang(TAB_LANGUAGE_EN_US)\r
        HelpTextObj.SetString(HelpStr)\r
\r
        HelpTextList = [HelpTextObj]\r
        Result = _GetHelpStr(HelpTextList)\r
        self.assertEqual(Result, HelpStr)\r
\r
    #\r
    # Normal case2: have two help text object with Lang = 'en-US' and other\r
    #\r
    def testNormalCase2(self):\r
        HelpStr = 'Hello world'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang(TAB_LANGUAGE_ENG)\r
        HelpTextObj.SetString(HelpStr)\r
\r
        HelpTextList = [HelpTextObj]\r
\r
        ExpectedStr = 'Hello world1'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang(TAB_LANGUAGE_EN_US)\r
        HelpTextObj.SetString(ExpectedStr)\r
\r
        HelpTextList.append(HelpTextObj)\r
\r
        Result = _GetHelpStr(HelpTextList)\r
        self.assertEqual(Result, ExpectedStr)\r
\r
    #\r
    # Normal case3: have two help text object with Lang = '' and 'eng'\r
    #\r
    def testNormalCase3(self):\r
        HelpStr = 'Hello world'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
\r
        HelpTextList = [HelpTextObj]\r
\r
        ExpectedStr = 'Hello world1'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang(TAB_LANGUAGE_ENG)\r
        HelpTextObj.SetString(ExpectedStr)\r
\r
        HelpTextList.append(HelpTextObj)\r
\r
        Result = _GetHelpStr(HelpTextList)\r
        self.assertEqual(Result, ExpectedStr)\r
\r
    #\r
    # Normal case4: have two help text object with Lang = '' and ''\r
    #\r
    def testNormalCase4(self):\r
\r
        ExpectedStr = 'Hello world1'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang(TAB_LANGUAGE_ENG)\r
        HelpTextObj.SetString(ExpectedStr)\r
        HelpTextList = [HelpTextObj]\r
\r
        HelpStr = 'Hello world'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        HelpTextList.append(HelpTextObj)\r
\r
        Result = _GetHelpStr(HelpTextList)\r
        self.assertEqual(Result, ExpectedStr)\r
\r
    #\r
    # Normal case: have three help text object with Lang = '','en', 'en-US'\r
    #\r
    def testNormalCase5(self):\r
\r
        ExpectedStr = 'Hello world1'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang(TAB_LANGUAGE_EN_US)\r
        HelpTextObj.SetString(ExpectedStr)\r
        HelpTextList = [HelpTextObj]\r
\r
        HelpStr = 'Hello unknown world'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        HelpTextList.append(HelpTextObj)\r
\r
        HelpStr = 'Hello mysterious world'\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        HelpTextList.append(HelpTextObj)\r
\r
        Result = _GetHelpStr(HelpTextList)\r
        self.assertEqual(Result, ExpectedStr)\r
\r
        HelpTextList.sort()\r
        self.assertEqual(Result, ExpectedStr)\r
\r
        HelpTextList.sort(reverse=True)\r
        self.assertEqual(Result, ExpectedStr)\r
\r
\r
#\r
# Test GenGuidSections\r
#\r
class GenGuidSectionsTest(unittest.TestCase):\r
    def setUp(self):\r
        pass\r
\r
    def tearDown(self):\r
        pass\r
\r
    #\r
    # This is the API to generate Guid Object to help UnitTest\r
    #\r
    def GuidFactory(self, CName, FFE, Usage, GuidType, VariableName, HelpStr):\r
        Guid = GuidObject()\r
        Guid.SetCName(CName)\r
        Guid.SetFeatureFlag(FFE)\r
        Guid.SetGuidTypeList([GuidType])\r
        Guid.SetUsage(Usage)\r
        Guid.SetVariableName(VariableName)\r
\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        Guid.SetHelpTextList([HelpTextObj])\r
\r
        return Guid\r
\r
    #\r
    # Normal case: have two GuidObject\r
    #\r
    def testNormalCase1(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 1'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'CONSUMES'\r
        GuidType = 'Variable'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 2'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
## PRODUCES ## Event # Usage comment line 1\r
## CONSUMES ## Variable: # Usage comment line 2\r
Guid1|FFE1'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # Normal case: have two GuidObject\r
    #\r
    def testNormalCase2(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 1'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = 'Generic comment line 1\n Generic comment line 2'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
## PRODUCES ## Event # Usage comment line 1\r
# Generic comment line 1\r
# Generic comment line 2\r
Guid1|FFE1'''\r
\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # Normal case: have two GuidObject, one help goes to generic help,\r
    # the other go into usage comment\r
    #\r
    def testNormalCase3(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = 'Generic comment'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 1'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
# Generic comment\r
## PRODUCES ## Event # Usage comment line 1\r
Guid1|FFE1'''\r
\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # Normal case: have one GuidObject, generic comment multiple lines\r
    #\r
    def testNormalCase5(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = 'Generic comment line1 \n generic comment line 2'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
# Generic comment line1\r
# generic comment line 2\r
Guid1|FFE1'''\r
\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # Normal case: have one GuidObject, usage comment multiple lines\r
    #\r
    def testNormalCase6(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 1\n Usage comment line 2'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
Guid1|FFE1 ## PRODUCES ## Event # Usage comment line 1  Usage comment line 2\r
'''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
    #\r
    # Normal case: have one GuidObject, usage comment one line\r
    #\r
    def testNormalCase7(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 1'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
Guid1|FFE1 # Usage comment line 1\r
'''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
    #\r
    # Normal case: have two GuidObject\r
    #\r
    def testNormalCase8(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 1\n Usage comment line 2'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 3'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
## PRODUCES ## Event # Usage comment line 1  Usage comment line 2\r
## PRODUCES ## Event # Usage comment line 3\r
Guid1|FFE1\r
'''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
    #\r
    # Normal case: have no GuidObject\r
    #\r
    def testNormalCase9(self):\r
        GuidList = []\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = ''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
    #\r
    # Normal case: have one GuidObject with no comment generated\r
    #\r
    def testNormalCase10(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = ''\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
Guid1|FFE1\r
'''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
    #\r
    # Normal case: have three GuidObject\r
    #\r
    def testNormalCase11(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = 'general comment line 1'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = 'Usage comment line 3'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'UNDEFINED'\r
        GuidType = 'UNDEFINED'\r
        VariableName = ''\r
        HelpStr = 'general comment line 2'\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
# general comment line 1\r
## PRODUCES ## Event # Usage comment line 3\r
# general comment line 2\r
Guid1|FFE1\r
'''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
    #\r
    # Normal case: have three GuidObject, with Usage/Type and no help\r
    #\r
    def testNormalCase12(self):\r
        GuidList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'GUID'\r
        VariableName = ''\r
        HelpStr = ''\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'PRODUCES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = ''\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
        Usage = 'CONSUMES'\r
        GuidType = 'Event'\r
        VariableName = ''\r
        HelpStr = ''\r
        Guid1 = self.GuidFactory(CName, FFE, Usage, GuidType,\r
                                 VariableName, HelpStr)\r
        GuidList.append(Guid1)\r
\r
        Result = GenGuidSections(GuidList)\r
        Expected = '''[Guids]\r
## PRODUCES ## GUID\r
## PRODUCES ## Event\r
## CONSUMES ## Event\r
Guid1|FFE1\r
'''\r
        self.assertEqual(Result.strip(), Expected.strip())\r
\r
#\r
# Test GenProtocolPPiSections\r
#\r
class GenProtocolPPiSectionsTest(unittest.TestCase):\r
    def setUp(self):\r
        pass\r
\r
    def tearDown(self):\r
        pass\r
\r
    #\r
    # This is the API to generate Protocol/Ppi Object to help UnitTest\r
    #\r
    def ObjectFactory(self, CName, FFE, Usage, Notify, HelpStr, IsProtocol):\r
        if IsProtocol:\r
            Object = ProtocolObject()\r
        else:\r
            Object = PpiObject()\r
\r
        Object.SetCName(CName)\r
        Object.SetFeatureFlag(FFE)\r
        Object.SetUsage(Usage)\r
        Object.SetNotify(Notify)\r
\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        Object.SetHelpTextList([HelpTextObj])\r
\r
        return Object\r
\r
    #    Usage        Notify    Help    INF Comment\r
    #1   UNDEFINED    true    Present    ## UNDEFINED ## NOTIFY # Help\r
    #2   UNDEFINED    true    Not Present    ## UNDEFINED ## NOTIFY\r
    #3   UNDEFINED    false    Present    ## UNDEFINED # Help\r
    #4   UNDEFINED    false     Not Present    ## UNDEFINED\r
    #5   UNDEFINED    Not Present    Present    # Help\r
    #6   UNDEFINED    Not Present    Not Present    <empty>\r
    #7   Other        true    Present    ## Other ## NOTIFY # Help\r
    #8   Other        true    Not Present    ## Other ## NOTIFY\r
    #9   Other        false    Present    ## Other # Help\r
    #A   Other        false     Not Present    ## Other\r
    #B   Other        Not Present    Present    ## Other # Help\r
    #C   Other        Not Present    Not Present    ## Other\r
\r
    def testNormalCase1(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'UNDEFINED'\r
        Notify = True\r
        HelpStr = 'Help'\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## UNDEFINED ## NOTIFY # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
        IsProtocol = False\r
        ObjectList = []\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Ppis]\r
Guid1|FFE1 ## UNDEFINED ## NOTIFY # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase2(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'UNDEFINED'\r
        Notify = True\r
        HelpStr = ''\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## UNDEFINED ## NOTIFY'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase3(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'UNDEFINED'\r
        Notify = False\r
        HelpStr = 'Help'\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## UNDEFINED # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase4(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'UNDEFINED'\r
        Notify = False\r
        HelpStr = ''\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## UNDEFINED'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase5(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'UNDEFINED'\r
        Notify = ''\r
        HelpStr = 'Help'\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase6(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'UNDEFINED'\r
        Notify = ''\r
        HelpStr = ''\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase7(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'PRODUCES'\r
        Notify = True\r
        HelpStr = 'Help'\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## PRODUCES ## NOTIFY # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase8(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'PRODUCES'\r
        Notify = True\r
        HelpStr = ''\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## PRODUCES ## NOTIFY'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase9(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'PRODUCES'\r
        Notify = False\r
        HelpStr = 'Help'\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## PRODUCES # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCaseA(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'PRODUCES'\r
        Notify = False\r
        HelpStr = ''\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## PRODUCES'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCaseB(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'PRODUCES'\r
        Notify = ''\r
        HelpStr = 'Help'\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## PRODUCES # Help'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCaseC(self):\r
        ObjectList = []\r
\r
        CName = 'Guid1'\r
        FFE = 'FFE1'\r
\r
        Usage = 'PRODUCES'\r
        Notify = ''\r
        HelpStr = ''\r
        IsProtocol = True\r
        Object = self.ObjectFactory(CName, FFE, Usage, Notify,\r
                                 HelpStr, IsProtocol)\r
        ObjectList.append(Object)\r
\r
\r
        Result = GenProtocolPPiSections(ObjectList, IsProtocol)\r
        Expected = '''[Protocols]\r
Guid1|FFE1 ## PRODUCES'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
#\r
# Test GenPcdSections\r
#\r
class GenPcdSectionsTest(unittest.TestCase):\r
    def setUp(self):\r
        pass\r
\r
    def tearDown(self):\r
        pass\r
\r
    #\r
    # This is the API to generate Pcd Object to help UnitTest\r
    #\r
    def ObjectFactory(self, ItemType, TSCName, CName, DValue, FFE, Usage, Str):\r
        Object = PcdObject()\r
        HelpStr = Str\r
\r
        Object.SetItemType(ItemType)\r
        Object.SetTokenSpaceGuidCName(TSCName)\r
        Object.SetCName(CName)\r
        Object.SetDefaultValue(DValue)\r
        Object.SetFeatureFlag(FFE)\r
        Object.SetValidUsage(Usage)\r
\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        Object.SetHelpTextList([HelpTextObj])\r
\r
        return Object\r
\r
\r
    #    Usage        Help    INF Comment\r
    #1   UNDEFINED    Present    # Help\r
    #2   UNDEFINED    Not Present    <empty>\r
    #3   Other        Present    ## Other # Help\r
    #4   Other        Not Present    ## Other\r
\r
    def testNormalCase1(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'UNDEFINED'\r
        Str = 'Help'\r
\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = \\r
            '[Pcd]\n' + \\r
            'TSCName.CName|DValue|FFE # Help'\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase2(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'UNDEFINED'\r
        Str = ''\r
\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '[Pcd]\nTSCName.CName|DValue|FFE'\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase3(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'CONSUMES'\r
        Str = 'Help'\r
\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '[Pcd]\nTSCName.CName|DValue|FFE ## CONSUMES # Help'\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    def testNormalCase4(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'CONSUMES'\r
        Str = ''\r
\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '[Pcd]\nTSCName.CName|DValue|FFE ## CONSUMES'\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # multiple lines for normal usage\r
    #\r
    def testNormalCase5(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'CONSUMES'\r
        Str = 'commment line 1\ncomment line 2'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '''[Pcd]\r
TSCName.CName|DValue|FFE ## CONSUMES # commment line 1 comment line 2'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # multiple lines for UNDEFINED usage\r
    #\r
    def testNormalCase6(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'UNDEFINED'\r
        Str = 'commment line 1\ncomment line 2'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Usage = 'UNDEFINED'\r
        Str = 'commment line 3'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '''[Pcd]\r
# commment line 1\r
# comment line 2\r
# commment line 3\r
TSCName.CName|DValue|FFE'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # multiple lines for UNDEFINED and normal usage\r
    #\r
    def testNormalCase7(self):\r
        ObjectList = []\r
        ItemType = 'Pcd'\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'UNDEFINED'\r
        Str = 'commment line 1\ncomment line 2'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Usage = 'CONSUMES'\r
        Str = 'Foo'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Usage = 'UNDEFINED'\r
        Str = 'commment line 3'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '''[Pcd]\r
# commment line 1\r
# comment line 2\r
## CONSUMES # Foo\r
# commment line 3\r
TSCName.CName|DValue|FFE'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #    Usage        Help    INF Comment\r
    #    CONSUMES    Present    # Help (keep <EOL> and insert '#' at beginning of each new line)\r
    #    CONSUMES    Not Present   <empty>\r
\r
    #\r
    # TAB_INF_FEATURE_PCD\r
    #\r
    def testNormalCase8(self):\r
        ObjectList = []\r
        ItemType = TAB_INF_FEATURE_PCD\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'CONSUMES'\r
        Str = 'commment line 1\ncomment line 2'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '''[FeaturePcd]\r
# commment line 1\r
# comment line 2\r
TSCName.CName|DValue|FFE'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # TAB_INF_FEATURE_PCD\r
    #\r
    def testNormalCase9(self):\r
        ObjectList = []\r
        ItemType = TAB_INF_FEATURE_PCD\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'CONSUMES'\r
        Str = ''\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '''[FeaturePcd]\r
TSCName.CName|DValue|FFE'''\r
        self.assertEqual(Result.strip(), Expected)\r
\r
    #\r
    # TAB_INF_FEATURE_PCD\r
    #\r
    def testNormalCase10(self):\r
        ObjectList = []\r
        ItemType = TAB_INF_FEATURE_PCD\r
        TSCName = 'TSCName'\r
        CName = 'CName'\r
        DValue = 'DValue'\r
        FFE = 'FFE'\r
\r
        Usage = 'PRODUCES'\r
        Str = 'commment line 1\ncomment line 2'\r
        Object = self.ObjectFactory(ItemType, TSCName, CName, DValue, FFE,\r
                                    Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenPcdSections(ObjectList)\r
        Expected = '''\r
\r
[FeaturePcd]\r
# commment line 1\r
# comment line 2\r
TSCName.CName|DValue|FFE\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
\r
#\r
# Test GenSpecialSections of Hob\r
#\r
class GenHobSectionsTest(unittest.TestCase):\r
    def setUp(self):\r
        pass\r
\r
    def tearDown(self):\r
        pass\r
\r
    #\r
    # This is the API to generate Event Object to help UnitTest\r
    #\r
    def ObjectFactory(self, SupArchList, Type, Usage, Str):\r
        Object = HobObject()\r
        HelpStr = Str\r
\r
        Object.SetHobType(Type)\r
        Object.SetUsage(Usage)\r
        Object.SetSupArchList(SupArchList)\r
\r
        HelpTextObj = TextObject()\r
        HelpTextObj.SetLang('')\r
        HelpTextObj.SetString(HelpStr)\r
        Object.SetHelpTextList([HelpTextObj])\r
\r
        return Object\r
\r
    def testNormalCase1(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = 'Help'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# # Help\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase2(self):\r
        ObjectList = []\r
        SupArchList = []\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = 'Help'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob]\r
# ##\r
# # Help\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase3(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\nComment Line 1\n\n'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# # Comment Line 1\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase4(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\nComment Line 1\n'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# # Comment Line 1\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase5(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = 'Comment Line 1\n\n'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# # Comment Line 1\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase6(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = ''\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase7(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\nNew Stack HoB'\r
\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# # New Stack HoB\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase8(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\nNew Stack HoB\n\nTail Comment'\r
\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# # New Stack HoB\r
# #\r
# # Tail Comment\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase9(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\n\n'\r
\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# #\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase10(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\n'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# #\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase11(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\n\n\n'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# #\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase12(self):\r
        ObjectList = []\r
        SupArchList = ['X64']\r
        Type = 'Foo'\r
        Usage = 'UNDEFINED'\r
        Str = '\n\n\n\n'\r
\r
        Object = self.ObjectFactory(SupArchList, Type, Usage, Str)\r
        ObjectList.append(Object)\r
\r
        Result = GenSpecialSections(ObjectList, 'Hob')\r
        Expected = '''# [Hob.X64]\r
# ##\r
# #\r
# #\r
# #\r
# Foo ## UNDEFINED\r
#\r
#\r
'''\r
        self.assertEqual(Result, Expected)\r
\r
#\r
# Test GenGenericCommentF\r
#\r
class GenGenericCommentFTest(unittest.TestCase):\r
    def setUp(self):\r
        pass\r
\r
    def tearDown(self):\r
        pass\r
\r
    def testNormalCase1(self):\r
        CommentLines = 'Comment Line 1'\r
        Result = GenGenericCommentF(CommentLines)\r
        Expected = '# Comment Line 1\n'\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase2(self):\r
        CommentLines = '\n'\r
        Result = GenGenericCommentF(CommentLines)\r
        Expected = '#\n'\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase3(self):\r
        CommentLines = '\n\n\n'\r
        Result = GenGenericCommentF(CommentLines)\r
        Expected = '#\n#\n#\n'\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase4(self):\r
        CommentLines = 'coment line 1\n'\r
        Result = GenGenericCommentF(CommentLines)\r
        Expected = '# coment line 1\n'\r
        self.assertEqual(Result, Expected)\r
\r
    def testNormalCase5(self):\r
        CommentLines = 'coment line 1\n coment line 2\n'\r
        Result = GenGenericCommentF(CommentLines)\r
        Expected = '# coment line 1\n# coment line 2\n'\r
        self.assertEqual(Result, Expected)\r
\r
if __name__ == '__main__':\r
    Logger.Initialize()\r
    unittest.main()\r