add a test for the hostname parameter

2 files changed, 11 insertions, 0 deletions

diff --git a/bin/problem-7.sh b/bin/problem-7.sh
index 1005ac3..4e53180 100755
--- a/bin/problem-7.sh
+++ b/bin/problem-7.sh
@@ -1,5 +1,10 @@
 #!/bin/sh
 
+if [ $# -eq 0 ]; then
+  echo "error: a hostname is required" >&2
+  exit 1
+fi
+
 destination="$1"
 host=$(host "$1" | grep 'has address')
 ip=$(echo "$host" | cut -d' ' -f4)
diff --git a/test/problem-7.bats b/test/problem-7.bats
index 57d657a..6b5e193 100644
--- a/test/problem-7.bats
+++ b/test/problem-7.bats
@@ -13,3 +13,9 @@ load test_helper
 
   assert_success "unknown is NOT on the network"
 }
+
+@test "invoke without a hostname argument" {
+  run problem-7.sh
+
+  assert_failure "error: a hostname is required"
+}