From 1d4a86594b488d572e6007c009dfa8eeccd96ded Mon Sep 17 00:00:00 2001
From: Matt Fellenz <mattf53190@gmail.com>
Date: Mon, 4 Oct 2021 20:35:26 -0700
Subject: [PATCH] Add tests for UTMP methods

---
 kitty_tests/utmp.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)
 create mode 100644 kitty_tests/utmp.py

diff --git a/kitty_tests/utmp.py b/kitty_tests/utmp.py
new file mode 100644
index 000000000..2105ba13a
--- /dev/null
+++ b/kitty_tests/utmp.py
@@ -0,0 +1,13 @@
+from sys import platform
+
+if platform in ('linux', 'linux2'):
+    import subprocess
+    import re
+    from kitty.fast_data_types import num_users
+    from . import BaseTest
+
+    class UTMPTest(BaseTest):
+        def test_num_users(self):
+            # who -q is the control
+            expected = subprocess.run(['who'], capture_output=True).stdout.decode('utf-8').count('\n')
+            self.ae(num_users(), expected)