1 files changed, 40 insertions, 0 deletions
diff --git a/tests/test_procfile.py b/tests/test_procfile.py
new file mode 100644
index 0000000..d919771
--- /dev/null
+++ b/tests/test_procfile.py
@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+"""
+    Basic ProcfileLexer Test
+    ~~~~~~~~~~~~~~~~~~~~
+
+    :copyright: Copyright 2006-2020 by the Pygments team, see AUTHORS.
+    :license: BSD, see LICENSE for details.
+"""
+
+import pytest
+
+from pygments.token import Name, Punctuation, Text
+from pygments.lexers.procfile import ProcfileLexer
+
+
+@pytest.fixture(scope='module')
+def lexer():
+    yield ProcfileLexer()
+
+
+def test_basic_line(lexer):
+    text = 'task: executable --options'
+
+    tokens = lexer.get_tokens(text)
+
+    for index, token in enumerate(tokens):
+        if index == 0:
+            assert token == (Name.Label, 'task')
+        elif index == 1:
+            assert token == (Punctuation, ':')
+        else:
+            assert token[0] in (Text, Text.Whitespace)
+
+
+def test_environment_variable(lexer):
+    text = '$XDG_SESSION_PATH'
+
+    token = list(lexer.get_tokens(text))[0]
+
+    assert token == (Name.Variable, text)