183 lines
7.3 KiB
Python
183 lines
7.3 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
#
|
||
|
|
# Copyright 2010, Google Inc.
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
# Redistribution and use in source and binary forms, with or without
|
||
|
|
# modification, are permitted provided that the following conditions are
|
||
|
|
# met:
|
||
|
|
#
|
||
|
|
# * Redistributions of source code must retain the above copyright
|
||
|
|
# notice, this list of conditions and the following disclaimer.
|
||
|
|
# * Redistributions in binary form must reproduce the above
|
||
|
|
# copyright notice, this list of conditions and the following disclaimer
|
||
|
|
# in the documentation and/or other materials provided with the
|
||
|
|
# distribution.
|
||
|
|
# * Neither the name of Google Inc. nor the names of its
|
||
|
|
# contributors may be used to endorse or promote products derived from
|
||
|
|
# this software without specific prior written permission.
|
||
|
|
#
|
||
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
|
||
|
|
"""Tests for the Pump meta-programming tool."""
|
||
|
|
|
||
|
|
from google3.testing.pybase import googletest
|
||
|
|
import google3.third_party.googletest.googlemock.scripts.pump
|
||
|
|
|
||
|
|
pump = google3.third_party.googletest.googlemock.scripts.pump
|
||
|
|
Convert = pump.ConvertFromPumpSource
|
||
|
|
StripMetaComments = pump.StripMetaComments
|
||
|
|
|
||
|
|
|
||
|
|
class PumpTest(googletest.TestCase):
|
||
|
|
|
||
|
|
def testConvertsEmptyToEmpty(self):
|
||
|
|
self.assertEquals('', Convert('').strip())
|
||
|
|
|
||
|
|
def testConvertsPlainCodeToSame(self):
|
||
|
|
self.assertEquals('#include <stdio.h>\n',
|
||
|
|
Convert('#include <stdio.h>\n'))
|
||
|
|
|
||
|
|
def testConvertsLongIWYUPragmaToSame(self):
|
||
|
|
long_line = '// IWYU pragma: private, include "' + (80*'a') + '.h"\n'
|
||
|
|
self.assertEquals(long_line, Convert(long_line))
|
||
|
|
|
||
|
|
def testConvertsIWYUPragmaWithLeadingSpaceToSame(self):
|
||
|
|
long_line = ' // IWYU pragma: private, include "' + (80*'a') + '.h"\n'
|
||
|
|
self.assertEquals(long_line, Convert(long_line))
|
||
|
|
|
||
|
|
def testConvertsIWYUPragmaWithSlashStarLeaderToSame(self):
|
||
|
|
long_line = '/* IWYU pragma: private, include "' + (80*'a') + '.h"\n'
|
||
|
|
self.assertEquals(long_line, Convert(long_line))
|
||
|
|
|
||
|
|
def testConvertsIWYUPragmaWithSlashStarAndSpacesToSame(self):
|
||
|
|
long_line = ' /* IWYU pragma: private, include "' + (80*'a') + '.h"\n'
|
||
|
|
self.assertEquals(long_line, Convert(long_line))
|
||
|
|
|
||
|
|
def testIgnoresMetaComment(self):
|
||
|
|
self.assertEquals('',
|
||
|
|
Convert('$$ This is a Pump meta comment.\n').strip())
|
||
|
|
|
||
|
|
def testSimpleVarDeclarationWorks(self):
|
||
|
|
self.assertEquals('3\n',
|
||
|
|
Convert('$var m = 3\n'
|
||
|
|
'$m\n'))
|
||
|
|
|
||
|
|
def testVarDeclarationCanReferenceEarlierVar(self):
|
||
|
|
self.assertEquals('43 != 3;\n',
|
||
|
|
Convert('$var a = 42\n'
|
||
|
|
'$var b = a + 1\n'
|
||
|
|
'$var c = (b - a)*3\n'
|
||
|
|
'$b != $c;\n'))
|
||
|
|
|
||
|
|
def testSimpleLoopWorks(self):
|
||
|
|
self.assertEquals('1, 2, 3, 4, 5\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$range i 1..n\n'
|
||
|
|
'$for i, [[$i]]\n'))
|
||
|
|
|
||
|
|
def testSimpleLoopWithCommentWorks(self):
|
||
|
|
self.assertEquals('1, 2, 3, 4, 5\n',
|
||
|
|
Convert('$var n = 5 $$ This is comment 1.\n'
|
||
|
|
'$range i 1..n $$ This is comment 2.\n'
|
||
|
|
'$for i, [[$i]]\n'))
|
||
|
|
|
||
|
|
def testNonTrivialRangeExpressionsWork(self):
|
||
|
|
self.assertEquals('1, 2, 3, 4\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$range i (n/n)..(n - 1)\n'
|
||
|
|
'$for i, [[$i]]\n'))
|
||
|
|
|
||
|
|
def testLoopWithoutSeparatorWorks(self):
|
||
|
|
self.assertEquals('a + 1 + 2 + 3;\n',
|
||
|
|
Convert('$range i 1..3\n'
|
||
|
|
'a$for i [[ + $i]];\n'))
|
||
|
|
|
||
|
|
def testCanGenerateDollarSign(self):
|
||
|
|
self.assertEquals('$\n', Convert('$($)\n'))
|
||
|
|
|
||
|
|
def testCanIterpolateExpressions(self):
|
||
|
|
self.assertEquals('a[2] = 3;\n',
|
||
|
|
Convert('$var i = 1\n'
|
||
|
|
'a[$(i + 1)] = $(i*4 - 1);\n'))
|
||
|
|
|
||
|
|
def testConditionalWithoutElseBranchWorks(self):
|
||
|
|
self.assertEquals('true\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$if n > 0 [[true]]\n'))
|
||
|
|
|
||
|
|
def testConditionalWithElseBranchWorks(self):
|
||
|
|
self.assertEquals('true -- really false\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$if n > 0 [[true]]\n'
|
||
|
|
'$else [[false]] -- \n'
|
||
|
|
'$if n > 10 [[really true]]\n'
|
||
|
|
'$else [[really false]]\n'))
|
||
|
|
|
||
|
|
def testConditionalWithCascadingElseBranchWorks(self):
|
||
|
|
self.assertEquals('a\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$if n > 0 [[a]]\n'
|
||
|
|
'$elif n > 10 [[b]]\n'
|
||
|
|
'$else [[c]]\n'))
|
||
|
|
self.assertEquals('b\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$if n > 10 [[a]]\n'
|
||
|
|
'$elif n > 0 [[b]]\n'
|
||
|
|
'$else [[c]]\n'))
|
||
|
|
self.assertEquals('c\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'$if n > 10 [[a]]\n'
|
||
|
|
'$elif n > 8 [[b]]\n'
|
||
|
|
'$else [[c]]\n'))
|
||
|
|
|
||
|
|
def testNestedLexicalBlocksWork(self):
|
||
|
|
self.assertEquals('a = 5;\n',
|
||
|
|
Convert('$var n = 5\n'
|
||
|
|
'a = [[$if n > 0 [[$n]]]];\n'))
|
||
|
|
|
||
|
|
|
||
|
|
class StripMetaCommentsTest(googletest.TestCase):
|
||
|
|
|
||
|
|
def testReturnsSameStringIfItContainsNoComment(self):
|
||
|
|
self.assertEquals('', StripMetaComments(''))
|
||
|
|
self.assertEquals(' blah ', StripMetaComments(' blah '))
|
||
|
|
self.assertEquals('A single $ is fine.',
|
||
|
|
StripMetaComments('A single $ is fine.'))
|
||
|
|
self.assertEquals('multiple\nlines',
|
||
|
|
StripMetaComments('multiple\nlines'))
|
||
|
|
|
||
|
|
def testStripsSimpleComment(self):
|
||
|
|
self.assertEquals('yes\n', StripMetaComments('yes $$ or no?\n'))
|
||
|
|
|
||
|
|
def testStripsSimpleCommentWithMissingNewline(self):
|
||
|
|
self.assertEquals('yes', StripMetaComments('yes $$ or no?'))
|
||
|
|
|
||
|
|
def testStripsPureCommentLinesEntirely(self):
|
||
|
|
self.assertEquals('yes\n',
|
||
|
|
StripMetaComments('$$ a pure comment line.\n'
|
||
|
|
'yes $$ or no?\n'
|
||
|
|
' $$ another comment line.\n'))
|
||
|
|
|
||
|
|
def testStripsCommentsFromMultiLineText(self):
|
||
|
|
self.assertEquals('multi-\n'
|
||
|
|
'line\n'
|
||
|
|
'text is fine.',
|
||
|
|
StripMetaComments('multi- $$ comment 1\n'
|
||
|
|
'line\n'
|
||
|
|
'text is fine. $$ comment 2'))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
googletest.main()
|