Merge pull request #34 from keszybz/test-skipping

Improve test skipping on old systems.

1 files changed, 48 insertions, 0 deletions

diff --git a/systemd/test/test_login.py b/systemd/test/test_login.py
new file mode 100644
index 0000000..afb5f45
--- /dev/null
+++ b/systemd/test/test_login.py
@@ -0,0 +1,48 @@
+from __future__ import print_function
+import select
+import contextlib
+import errno
+
+from systemd import login
+
+import pytest
+
+@contextlib.contextmanager
+def skip_oserror(code):
+    try:
+        yield
+    except (OSError, IOError) as e:
+        if e.errno == code:
+            pytest.skip()
+        raise
+
+def test_seats():
+    # just check that we get some sequence back
+    with skip_oserror(errno.ENOENT):
+        seats = login.seats()
+        assert len(seats) >= 0
+
+def test_sessions():
+    with skip_oserror(errno.ENOENT):
+        sessions = login.sessions()
+        assert len(sessions) >= 0
+
+def test_machine_names():
+    with skip_oserror(errno.ENOENT):
+        machine_names = login.machine_names()
+        assert len(machine_names) >= 0
+
+def test_uids():
+    with skip_oserror(errno.ENOENT):
+        uids = login.uids()
+        assert len(uids) >= 0
+
+def test_monitor():
+    p = select.poll()
+
+    with skip_oserror(errno.ENOENT):
+        m = login.Monitor("machine")
+        p.register(m, m.get_events())
+        login.machine_names()
+        p.poll(1)
+        login.machine_names()