test/unit/peddler/test_headers.rb in peddler-1.6.7 vs test/unit/peddler/test_headers.rb in peddler-2.0.0

@@ -17,29 +17,39 @@
       'x-mws-timestamp' => '2017-01-29T23:55:25.356Z',
       'x-mws-response-context' => 'foo'
     }
   end
 
-  def test_quota
-    assert quota
-    assert_kind_of Integer, quota.max
-    assert_kind_of Integer, quota.remaining
-    assert_kind_of Time, quota.resets_on
+  def test_mws_quota_max
+    assert_kind_of Integer, mws_quota_max
   end
 
+  def test_mws_quota_remaining
+    assert_kind_of Integer, mws_quota_remaining
+  end
+
+  def test_mws_quota_resets_on
+    assert_kind_of Time, mws_quota_resets_on
+  end
+
   def test_request_id
-    assert request_id
+    assert mws_request_id
   end
 
   def test_timestamp
-    assert_kind_of Time, timestamp
+    assert_kind_of Time, mws_timestamp
   end
 
   def test_response_context
-    assert response_context
+    assert mws_response_context
   end
 
   def test_handles_no_quota
     @headers = {}
-    assert_nil quota
+    assert_nil mws_quota_max
+    assert_nil mws_quota_remaining
+    assert_nil mws_quota_resets_on
+    assert_nil mws_request_id
+    assert_nil mws_timestamp
+    assert_nil mws_response_context
   end
 end