[ci] Unfreeze time in StatusHistoryTest

openSUSE · Oct 19, 2015 · bacde3f · bacde3f
1 parent 2cd4395
commit bacde3f
Showing 1 changed file with 26 additions and 24 deletions.
diff --git a/src/api/test/models/status_history_test.rb b/src/api/test/models/status_history_test.rb
@@ -11,38 +11,40 @@ def teardown
   end
 
   test "rescale" do
-    Timecop.freeze(2010, 7, 12)
-    now = Time.now.to_i - 2.days
-    StatusHistory.transaction do
-      1000.times do |i|
-        StatusHistory.create time: now + i, key: 'idle_x86_64', value: i
+    Timecop.freeze(2010, 7, 12) do
+      now = Time.now.to_i - 2.days
+      StatusHistory.transaction do
+        1000.times do |i|
+          StatusHistory.create time: now + i, key: 'idle_x86_64', value: i
+        end
       end
-    end
 
-    StatusHistory.create time: Time.now.to_i, key: 'busy_x86_64', value: 100
+      StatusHistory.create time: Time.now.to_i, key: 'busy_x86_64', value: 100
 
-    assert_equal 1001, StatusHistory.count
-    StatusHistoryRescaler.new.rescale
-    assert_equal 2, StatusHistory.count
+      assert_equal 1001, StatusHistory.count
+      StatusHistoryRescaler.new.rescale
+      assert_equal 2, StatusHistory.count
 
-    rel = StatusHistory.where(key: 'idle_x86_64')
-    assert_equal 1, rel.count
-    assert_equal 499.5, rel.first.value
+      rel = StatusHistory.where(key: 'idle_x86_64')
+      assert_equal 1, rel.count
+      assert_equal 499.5, rel.first.value
+    end
   end
 
   test "history_by_key_and_hours" do
-    Timecop.freeze(2010, 7, 12)
-    day_before_yesterday = Time.now.to_i - 2.days
-    yesterday = Time.now.to_i - 1.days
-    StatusHistory.transaction do
-      10.times do |i|
-        StatusHistory.create time: day_before_yesterday + i, key: 'squeue_low_aarch64', value: i
-      end
-      5.times do |i|
-        StatusHistory.create time: yesterday + i, key: 'squeue_low_aarch64', value: i
+    Timecop.freeze(2010, 7, 12) do
+      day_before_yesterday = Time.now.to_i - 2.days
+      yesterday = Time.now.to_i - 1.days
+      StatusHistory.transaction do
+        10.times do |i|
+          StatusHistory.create time: day_before_yesterday + i, key: 'squeue_low_aarch64', value: i
+        end
+        5.times do |i|
+          StatusHistory.create time: yesterday + i, key: 'squeue_low_aarch64', value: i
+        end
       end
+      assert_equal 5, StatusHistory.history_by_key_and_hours('squeue_low_aarch64', 25).count
+      assert_equal 15, StatusHistory.history_by_key_and_hours('squeue_low_aarch64', 49).count
     end
-    assert_equal 5, StatusHistory.history_by_key_and_hours('squeue_low_aarch64', 25).count
-    assert_equal 15, StatusHistory.history_by_key_and_hours('squeue_low_aarch64', 49).count
   end
 end