Fix bug for classes that have an export macro and are final at the sa…

…me time (myint#152)
dyadav7 · Nov 10, 2019 · aa73d06 · aa73d06
1 parent 8340aba
commit aa73d06
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/cpp/ast.py b/cpp/ast.py
@@ -1521,6 +1521,9 @@ def _get_class(self, class_type, templated_types):
             assert_parse(class_name, class_token)
 
         bases = None
+
+        if token.name == 'final':
+            token = self._get_next_token()
         if token.token_type == tokenize.PREPROCESSOR:
             token = self._get_next_token()
         if token.token_type == tokenize.SYNTAX:

diff --git a/test_ast.py b/test_ast.py
@@ -896,6 +896,12 @@ def test_class_virtual_inheritance_reverse(self):
         self.assertEqual(1, len(nodes), repr(nodes))
         self.assertEqual(Class('Foo', bases=[Type('Bar')], body=[]), nodes[0])
 
+    def test_class_macro_final(self):
+        code = 'class EXPORT_MACRO Foo final {};'
+        nodes = list(MakeBuilder(code).generate())
+        self.assertEqual(1, len(nodes), repr(nodes))
+        self.assertEqual(Class('Foo', body=[]), nodes[0])
+
     def test_constructor(self):
         code = 'Foo::Foo() {}'
         nodes = list(MakeBuilder(code).generate())