mock driver: allow switching to storage-only driver

git-svn-id: https://svn.apache.org/repos/asf/incubator/deltacloud/trunk@1063024 13f79535-47bb-0310-9956-ffa450edef68

server/lib/deltacloud/drivers/mock/mock_driver.rb

@@ -25,8 +25,15 @@ module Drivers
     module Mock
 class MockDriver < Deltacloud::BaseDriver
 
+  # If the provider is set to storage, pretend to be a storage-only
+  # driver
   def supported_collections
-    DEFAULT_COLLECTIONS + [ :buckets, :keys]
+    endpoint = Thread.current[:provider] || ENV['API_PROVIDER']
+    if endpoint == 'storage'
+      [:buckets]
+    else
+      DEFAULT_COLLECTIONS + [:buckets, :keys]
+    end
   end
 
   ( REALMS = [

server/tests/drivers/mock/api_test.rb

@@ -34,5 +34,19 @@ def test_it_responses_to_json
       JSON::parse(last_response.body)['api'].class.should == Hash
     end
 
+    def test_it_switches_drivers
+      begin
+        ENV.delete("API_PROVIDER")
+        do_xml_request '/api'
+        (last_xml_response/"api/link[rel = 'instances']").first.should_not == nil
+
+        # Switch to storage-only mock driver
+        ENV["API_PROVIDER"] = "storage"
+        do_xml_request '/api'
+        (last_xml_response/"api/link[rel = 'instances']").first.should == nil
+      ensure
+        ENV.delete("API_PROVIDER")
+      end
+    end
   end
 end