Add helper methods for swagger

Author: Lee Richmond

lib/jsonapi_spec_helpers.rb

@@ -1,18 +1,23 @@
 require 'json'
 require 'jsonapi_spec_helpers/version'
-require 'jsonapi_spec_helpers/matchers'
 require 'jsonapi_spec_helpers/helpers'
 require 'jsonapi_spec_helpers/payload'
 require 'jsonapi_spec_helpers/payload_sanitizer'
 
 module JsonapiSpecHelpers
   def self.included(klass)
+    # don't load RSpec until included
+    require 'jsonapi_spec_helpers/matchers'
     klass.send(:include, Helpers)
     if defined?(Rails)
-      Dir[Rails.root.join('spec/payloads/**/*.rb')].each { |f| require f }
+      load_payloads!
     end
   end
 
+  def self.load_payloads!
+    Dir[Rails.root.join('spec/payloads/**/*.rb')].each { |f| require f }
+  end
+
   def assert_payload(name, record, json, &blk)
     unless payload = JsonapiSpecHelpers::Payload.registry[name]
       raise "No payloads registered for '#{name}'"

lib/jsonapi_spec_helpers/payload.rb

@@ -5,14 +5,24 @@ class << self
     end
     self.registry = {}
 
-    attr_accessor :keys, :no_keys
+    attr_accessor :name, :type, :keys, :no_keys
 
     def self.register(name, &blk)
       instance = new
       instance.instance_eval(&blk)
+      instance.name = name
       registry[name] = instance
     end
 
+    def self.by_type(type)
+      found = nil
+      registry.each_pair do |name, payload|
+        found = payload if payload.type == type
+      end
+      raise "Could not find payload for type #{type}" unless found
+      found
+    end
+
     def fork
       instance = self.class.new
       instance.keys = keys.dup
@@ -30,6 +40,14 @@ def no_key(name)
       @no_keys << name
     end
 
+    def type(val = nil)
+      if val
+        @type = val
+      else
+        @type || name.to_s.pluralize.to_sym
+      end
+    end
+
     def key(name, *args, &blk)
       options = args.last.is_a?(Hash) ? args.pop : {}
       options[:type] = args.first
@@ -41,8 +59,8 @@ def key(name, *args, &blk)
     end
 
     def timestamps!
-      @keys[:created_at] = key(:created_at)
-      @keys[:updated_at] = key(:updated_at)
+      @keys[:created_at] = key(:created_at, String)
+      @keys[:updated_at] = key(:updated_at, String)
     end
   end
 end