mirror of
https://github.com/awesome-print/awesome_print
synced 2023-03-27 23:22:34 -04:00
add open struct support
This commit is contained in:
parent
84bfda9d07
commit
a71586ad99
3 changed files with 59 additions and 0 deletions
|
@ -36,4 +36,5 @@ unless defined?(AwesomePrint::Inspector)
|
|||
require File.dirname(__FILE__) + "/awesome_print/ext/no_brainer" if defined?(NoBrainer)
|
||||
require File.dirname(__FILE__) + "/awesome_print/ext/ripple" if defined?(Ripple)
|
||||
require File.dirname(__FILE__) + "/awesome_print/ext/sequel" if defined?(Sequel)
|
||||
require File.dirname(__FILE__) + "/awesome_print/ext/ostruct" if defined?(OpenStruct)
|
||||
end
|
||||
|
|
27
lib/awesome_print/ext/ostruct.rb
Normal file
27
lib/awesome_print/ext/ostruct.rb
Normal file
|
@ -0,0 +1,27 @@
|
|||
# Copyright (c) 2010-2013 Michael Dvorkin
|
||||
#
|
||||
# Awesome Print is freely distributable under the terms of MIT license.
|
||||
# See LICENSE file or http://www.opensource.org/licenses/mit-license.php
|
||||
#------------------------------------------------------------------------------
|
||||
module AwesomePrint
|
||||
module OpenStruct
|
||||
def self.included(base)
|
||||
base.send :alias_method, :cast_without_ostruct, :cast
|
||||
base.send :alias_method, :cast, :cast_with_ostruct
|
||||
end
|
||||
|
||||
def cast_with_ostruct(object, type)
|
||||
cast = cast_without_ostruct(object, type)
|
||||
if (defined?(::OpenStruct)) && (object.is_a?(::OpenStruct))
|
||||
cast = :open_struct_instance
|
||||
end
|
||||
cast
|
||||
end
|
||||
|
||||
def awesome_open_struct_instance(object)
|
||||
"#{object.class} #{awesome_hash(object.marshal_dump)}"
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
AwesomePrint::Formatter.send(:include, AwesomePrint::OpenStruct)
|
31
spec/ext/ostruct_spec.rb
Normal file
31
spec/ext/ostruct_spec.rb
Normal file
|
@ -0,0 +1,31 @@
|
|||
require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
|
||||
|
||||
begin
|
||||
require 'ostruct'
|
||||
require 'awesome_print/ext/ostruct'
|
||||
|
||||
describe 'AwesomePrint Ostruct extension' do
|
||||
before do
|
||||
stub_dotfile!
|
||||
@ap = AwesomePrint::Inspector.new(:plain => true, :sort_keys => true)
|
||||
end
|
||||
|
||||
it "empty hash" do
|
||||
struct = OpenStruct.new
|
||||
@ap.send(:awesome, struct).should == "OpenStruct {}"
|
||||
end
|
||||
|
||||
it "plain multiline" do
|
||||
struct = OpenStruct.new :name => "Marshall Shen", :address => "605 Madion St."
|
||||
@ap.send(:awesome, struct).should == <<-EOS.strip
|
||||
OpenStruct {
|
||||
:address => "605 Madion St.",
|
||||
:name => "Marshall Shen"
|
||||
}
|
||||
EOS
|
||||
end
|
||||
end
|
||||
|
||||
rescue LoadError => error
|
||||
puts "Skipping OpenStruct specs: #{error}"
|
||||
end
|
Loading…
Reference in a new issue