module ThoughtBot::Shoulda::ActiveRecord::Assertions

def get_instance_of(object_or_klass)
  if object_or_klass.is_a?(Class)
    klass = object_or_klass
    instance_variable_get("@#{klass.to_s.underscore}") || klass.new
  else
    object_or_klass
  end
end