add test for finding uid 0

1 files changed, 4 insertions, 2 deletions

diff --git a/bin/uid-zero b/bin/uid-zero
index 2976cbb..9b2640f 100755
--- a/bin/uid-zero
+++ b/bin/uid-zero
@@ -1,7 +1,9 @@
 #!/bin/bash
 
+FILE=${1:-'/etc/passwd'}
+
 if [[ "$OSTYPE" == "darwin"* ]]; then
-  cat /etc/passwd | cut -d: -f1,3 | grep ':0'
+  cat $FILE | cut -d: -f1,3 | grep ':0'
 else
-  cat /etc/passwd | grep -P '^\w+:[x]:\d+:0.*$'
+  cat $FILE | grep -P '^\w+:[x]:\d+:0.*$'
 fi