Add support for @import keyword in Objective-C

1 files changed, 12 insertions, 1 deletions

diff --git a/tests/test_objectiveclexer.py b/tests/test_objectiveclexer.py
index aee7db66..faadb08f 100644
--- a/tests/test_objectiveclexer.py
+++ b/tests/test_objectiveclexer.py
@@ -69,7 +69,7 @@ class ObjectiveCLexerTest(unittest.TestCase):
         ]
         self.assertEqual(expected, list(self.lexer.get_tokens(fragment)))
 
-    def testLieralNumberBoolExpression(self):
+    def testLiteralNumberBoolExpression(self):
         fragment = u'@(YES);\n'
         expected = [
             (Token.Literal, u'@('),
@@ -79,3 +79,14 @@ class ObjectiveCLexerTest(unittest.TestCase):
             (Token.Text, u'\n'),
         ]
         self.assertEqual(expected, list(self.lexer.get_tokens(fragment)))
+
+    def testModuleImport(self):
+        fragment = u'@import ModuleA;\n'
+        expected = [
+            (Token.Keyword, u'@import'),
+            (Token.Text, u' '),
+            (Token.Name, u'ModuleA'),
+            (Token.Punctuation, u';'),
+            (Token.Text, u'\n'),
+        ]
+        self.assertEqual(expected, list(self.lexer.get_tokens(fragment)))