test/api_key_test.rb in fastly-0.96 vs test/api_key_test.rb in fastly-0.97

@@ -21,27 +21,26 @@
     user = customer = nil
     assert_raise(Fastly::Error) {
       user     = @client.get('/current_user')
     }
     assert_equal nil, user    
-    assert_raise(Fastly::Error) {
-      customer = @client.get('/current_customer')
-    }
-    assert_equal nil, customer
+
+    customer = @client.get('/current_customer')
+    assert customer
+    assert_equal @opts[:customer], customer['name']
   end
   
   
   def test_current_user_and_customer 
     current_user  = current_customer = nil
     assert_raise(Fastly::Error) {
       current_user = @fastly.current_user
     }
     assert_equal nil, current_user
 
-    assert_raise(Fastly::Error) {
-      current_customer = @fastly.current_customer
-    }
-    assert_equal nil, current_customer    
+    customer = @fastly.current_customer
+    assert customer
+    assert_equal @opts[:customer], customer.name
   end 
   
   
   def test_purging
     #assert @fastly.purge('foo')