From abf38106c0a35e0e77982f7c2cd49cf1f5a54c23 Mon Sep 17 00:00:00 2001 From: Chris Roberts Date: Tue, 7 Mar 2017 10:36:14 -0800 Subject: [PATCH] Add shared helper method to detect if running version is prerelease --- lib/vagrant/shared_helpers.rb | 8 ++++++++ test/unit/vagrant/shared_helpers_test.rb | 12 ++++++++++++ 2 files changed, 20 insertions(+) diff --git a/lib/vagrant/shared_helpers.rb b/lib/vagrant/shared_helpers.rb index ffe61e571..c5d6ea617 100644 --- a/lib/vagrant/shared_helpers.rb +++ b/lib/vagrant/shared_helpers.rb @@ -102,4 +102,12 @@ module Vagrant Pathname.new(path).expand_path end + + # This returns true/false if the running version of Vagrant is + # a pre-release version (development) + # + # @return [Boolean] + def self.prerelease? + Gem::Version.new(Vagrant::VERSION).prerelease? + end end diff --git a/test/unit/vagrant/shared_helpers_test.rb b/test/unit/vagrant/shared_helpers_test.rb index c64a26001..2ebfc76a5 100644 --- a/test/unit/vagrant/shared_helpers_test.rb +++ b/test/unit/vagrant/shared_helpers_test.rb @@ -131,4 +131,16 @@ describe Vagrant do end end end + + describe "#prerelease?" do + it "should return true when Vagrant version is development" do + stub_const("Vagrant::VERSION", "1.0.0.dev") + expect(subject.prerelease?).to be(true) + end + + it "should return false when Vagrant version is release" do + stub_const("Vagrant::VERSION", "1.0.0") + expect(subject.prerelease?).to be(false) + end + end end