beefcake 0.1.6 → 0.2.0

data/lib/beefcake.rb

@@ -104,7 +104,16 @@ module Beefcake
       end
 
       def valid_enum?(mod, val)
-        mod.constants.any? {|name| mod.const_get(name) == val }
+        !!name_for(mod, val)
+      end
+
+      def name_for(mod, val)
+        mod.constants.each do |name|
+          if mod.const_get(name) == val
+            return name
+          end
+        end
+        nil
       end
 
       def validate!
@@ -204,7 +213,19 @@ module Beefcake
 
     def inspect
       set = fields.values.select {|fld| self[fld.name] != nil }
-      flds = set.map {|fld| "#{fld.name}: #{self[fld.name].inspect}" }
+
+      flds = set.map do |fld|
+        val = self[fld.name]
+
+        case fld.type
+        when Module
+          title = name_for(fld.type, val) || "-NA-"
+          "#{fld.name}: #{title}(#{val.inspect})"
+        else
+          "#{fld.name}: #{val.inspect}"
+        end
+      end
+
       "<#{self.class.name} #{flds.join(", ")}>"
     end
 

data/test/message_test.rb

@@ -318,4 +318,11 @@ class MessageTest < Test::Unit::TestCase
     assert_equal "<SimpleMessage b: \"testing\">", msg.inspect
   end
 
+  def test_inspect_enums
+    msg = EnumsMessage.new :a => 1
+    assert_equal "<EnumsMessage a: A(1)>", msg.inspect
+    msg.a = 2
+    assert_equal "<EnumsMessage a: -NA-(2)>", msg.inspect
+  end
+
 end

metadata

@@ -5,9 +5,9 @@ version: !ruby/object:Gem::Version
   prerelease: 
   segments: 
   - 0
-  - 1
-  - 6
-  version: 0.1.6
+  - 2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Blake Mizerany