diff --git a/app/models/janitor_trial.rb b/app/models/janitor_trial.rb index b098e753c..037c97256 100644 --- a/app/models/janitor_trial.rb +++ b/app/models/janitor_trial.rb @@ -91,4 +91,12 @@ class JanitorTrial < ActiveRecord::Base user.update_column(:level, original_level) self.create_feedback end + + def active? + status == "active" + end + + def inactive? + status == "inactive" + end end diff --git a/test/unit/janitor_trial_test.rb b/test/unit/janitor_trial_test.rb index aff540e2e..a2cf89146 100644 --- a/test/unit/janitor_trial_test.rb +++ b/test/unit/janitor_trial_test.rb @@ -47,9 +47,8 @@ class JanitorTrialTest < ActiveSupport::TestCase end should "destroy the trial object" do - assert_difference("JanitorTrial.count", -1) do - @janitor_trial.promote! - end + @janitor_trial.promote! + assert_equal(false, @janitor_trial.active?) end end end