test/test_helper.rb in vertica-0.9.1 vs test/test_helper.rb in vertica-0.9.2

@@ -4,6 +4,16 @@
 require 'yaml'
 require 'test/unit'
 
 require 'vertica'
 
-TEST_CONNECTION_HASH = YAML.load(File.read(File.expand_path("../connection.yml", __FILE__)))
+TEST_CONNECTION_FILENAME = File.expand_path("../connection.yml", __FILE__)
+
+if File.exist?(TEST_CONNECTION_FILENAME)
+  TEST_CONNECTION_HASH = YAML.load(File.read(TEST_CONNECTION_FILENAME))
+else
+  puts
+  puts "Create #{TEST_CONNECTION_FILENAME} with connection info "
+  puts "for your Vertica test database in order to run the test suite."
+  puts
+  exit(1)
+end
\ No newline at end of file