BaseTools: Replace BSD License with BSD+Patent License

[mirror_edk2.git] / BaseTools / Tests / TestRegularExpression.py

## @file\r
# Routines for generating Pcd Database\r
#\r
# Copyright (c) 2018, Intel Corporation. All rights reserved.<BR>\r
# SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
import unittest\r
from Common.Misc import RemoveCComments\r
from Workspace.BuildClassObject import ArrayIndex\r
\r
class TestRe(unittest.TestCase):\r
    def test_ccomments(self):\r
        TestStr1 = """ {0x01,0x02} """\r
        self.assertEquals(TestStr1, RemoveCComments(TestStr1))\r
\r
        TestStr2 = """ L'TestString' """\r
        self.assertEquals(TestStr2, RemoveCComments(TestStr2))\r
\r
        TestStr3 = """ 'TestString' """\r
        self.assertEquals(TestStr3, RemoveCComments(TestStr3))\r
\r
        TestStr4 = """\r
            {CODE({\r
              {0x01, {0x02, 0x03, 0x04 }},// Data comment\r
              {0x01, {0x02, 0x03, 0x04 }},// Data comment\r
              })\r
            }  /*\r
               This is multiple line comments\r
               The seconde line comment\r
               */\r
            // This is a comment\r
        """\r
        Expect_TestStr4 = """{CODE({\r
              {0x01, {0x02, 0x03, 0x04 }},\r
              {0x01, {0x02, 0x03, 0x04 }},\r
              })\r
            }"""\r
        self.assertEquals(Expect_TestStr4, RemoveCComments(TestStr4).strip())\r
\r
    def Test_ArrayIndex(self):\r
        TestStr1 = """[1]"""\r
        self.assertEquals(['[1]'], ArrayIndex.findall(TestStr1))\r
\r
        TestStr2 = """[1][2][0x1][0x01][]"""\r
        self.assertEquals(['[1]','[2]','[0x1]','[0x01]','[]'], ArrayIndex.findall(TestStr2))\r
\r
if __name__ == '__main__':\r
    unittest.main()\r