features/support/env.rb in testlab-0.7.6 vs features/support/env.rb in testlab-0.8.0

@@ -1,9 +1,17 @@
 require 'aruba/cucumber'
 
 ENV['PATH'] = "#{File.expand_path(File.dirname(__FILE__) + '/../../bin')}#{File::PATH_SEPARATOR}#{ENV['PATH']}"
 LIB_DIR = File.join(File.expand_path(File.dirname(__FILE__)),'..','..','lib')
+
 TEST_REPO = File.dirname(__FILE__)
+
+if ENV['USER'] == 'vagrant'
+  TEST_LABFILE = File.join(TEST_REPO, 'Labfile.local')
+else
+  TEST_LABFILE = File.join(TEST_REPO, 'Labfile.vagrant')
+end
+
 
 Before do
   # Using "announce" causes massive warnings on 1.9.2
   @aruba_timeout_seconds = 3600
   @puts = true