diff --git a/test/tests/test_brakeman.rb b/test/tests/test_brakeman.rb index f2eca802cd8cd4494860257ce795212b02676836..08462065e9f8aac5afa9416acc9f832e1260245f 100644 --- a/test/tests/test_brakeman.rb +++ b/test/tests/test_brakeman.rb @@ -41,3 +41,33 @@ class SexpTests < Test::Unit::TestCase assert_equal call.args, Sexp.new() end end + +class BaseCheckTests < Test::Unit::TestCase + FakeTracker = Struct.new(:config) + + def setup + @tracker = FakeTracker.new + @check = Brakeman::BaseCheck.new @tracker + end + + def version_between? version, high, low + @tracker.config = { :rails_version => version } + @check.send(:version_between?, high, low) + end + + def test_version_between + assert version_between?("2.3.8", "2.3.0", "2.3.8") + assert version_between?("2.3.8", "2.3.0", "2.3.14") + assert version_between?("2.3.8", "1.0.0", "5.0.0") + end + + def test_version_not_between + assert_equal false, version_between?("3.2.1", "2.0.0", "3.0.0") + assert_equal false, version_between?("3.2.1", "3.0.0", "3.2.0") + assert_equal false, version_between?("0.0.0", "3.0.0", "3.2.0") + end + + def test_version_between_longer + assert_equal false, version_between?("1.0.1.2", "1.0.0", "1.0.1") + end +end