use os agnostic method to get user for tests to make compatible with …

…windows.
jonhadfield · Mar 1, 2020 · eb179e4 · eb179e4
1 parent 376edea
commit eb179e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_hosts.py b/tests/test_hosts.py
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 import datetime
 import os
-import pwd
+import getpass
 import sys
 
 import pytest
@@ -139,7 +139,7 @@ def test_remove_existing_entry_using_address_only(tmpdir):
 
 
 def get_username():
-    return pwd.getpwuid(os.getuid())[0]
+    return getpass.getuser()
 
 
 def test_hosts_str(tmpdir):