tests: Add --version tests, fix required installfile tests

1 files changed, 5 insertions, 1 deletions

diff --git a/tests/spec/validator_spec.rb b/tests/spec/validator_spec.rb
index 963eecf..365a8f3 100644
--- a/tests/spec/validator_spec.rb
+++ b/tests/spec/validator_spec.rb
@@ -11,7 +11,11 @@ RSpec.describe 'HorizonScript validation', :type => :aruba do
     context "Utility argument passing" do
         it "requires an installfile to be specified" do
             run_command 'hscript-validate'
-            expect(last_command_started).to have_output(/Allowed options/)
+            expect(last_command_started).to have_output(/must specify an installfile/)
+        end
+        it "supports version output" do
+            run_command 'hscript-validate --version'
+            expect(last_command_started).to_not have_output(/Allowed options/)
         end
         it "accepts -i flag" do
             run_command 'hscript-validate foo -i'