lib/buttercms-ruby.rb in buttercms-ruby-1.1.0 vs lib/buttercms-ruby.rb in buttercms-ruby-1.1.1

@@ -23,12 +23,13 @@
 end
 
 module ButterCMS
   @api_url = 'https://api.buttercms.com/v2'
 
-  class <<self
+  class << self
     attr_accessor :api_token
+    attr_accessor :test_mode
     attr_reader :data_store
     attr_writer :logger
   end
 
   def self.logger
@@ -56,14 +57,22 @@
       raise ArgumentError.new "Invalid ButterCMS data store #{strategy}"
     end
   end
 
   def self.api_request(path, options = {})
+    base_options = {
+      auth_token: api_token
+    }
+
+    if test_mode
+      base_options[:test] = 1
+    end
+
     response = RestClient::Request.execute(
       method: :get,
       url: @api_url + path,
       headers: {
-        params: options.merge(auth_token: api_token)
+        params: options.merge(base_options)
       },
       verify_ssl: false
     )
 
     response.body