Commit: 748d22adc3e2b518047007c3ad42872d0b1984df - protobuf_bvgsdk (git)

超漢字アプリケーション開発用のProtocol Buffersコンパイラの開発

Commit MetaInfo

• Revision: 748d22adc3e2b518047007c3ad42872d0b1984df (tree)
• Time: 2015-03-09 02:19:39
• Author: ornse01 <ornse01@user...>
• Commiter: ornse01

Log Message

add ASTField.owner_message

Change Summary

• modified: src/protobuf/ast.rb
• modified: src/protobuf/test_ast.rb

Incremental Difference

--- a/src/protobuf/ast.rb
+++ b/src/protobuf/ast.rb
@@ -61,6 +61,13 @@ class Protobuf::ASTField < TreeNode
     @tag = tag
     super(nil)
   end
+
+  def owner_message
+    self
+      .enum_for(:ancestral_each)
+      .select{ |item| item.is_a? Protobuf::ASTMessage }
+      .first
+  end
 end
 
 class Protobuf::ASTMessage < TreeNode

--- a/src/protobuf/test_ast.rb
+++ b/src/protobuf/test_ast.rb
@@ -80,6 +80,26 @@ class ProtobufAstTest < Test::Unit::TestCase
     assert(message2.absolute_name == 'msg2', 'invalid absolute name 2') { }
   end
 
+  def test_field_owner_message
+    field0 = Protobuf::ASTField.new :required, :int32, "f1", 1
+    field1 = Protobuf::ASTField.new :optional, :string, "f2", 2
+    message0 = Protobuf::ASTMessage.new "msg0", [field0, field1]
+    field2 = Protobuf::ASTField.new :required, :int32, "f1", 1
+    field3 = Protobuf::ASTField.new :optional, :string, "f2", 2
+    message1 = Protobuf::ASTMessage.new "msg1", [field2, field3, message0]
+    field4 = Protobuf::ASTField.new :required, :int32, "f1", 1
+    field5 = Protobuf::ASTField.new :optional, :string, "f2", 2
+    message2 = Protobuf::ASTMessage.new "msg2", [field4, field5, message1]
+    root = Protobuf::ASTRoot.new [message2]
+
+    assert(field0.owner_message == message0, 'invalid owner 0') { }
+    assert(field1.owner_message == message0, 'invalid owner 1') { }
+    assert(field2.owner_message == message1, 'invalid owner 2') { }
+    assert(field3.owner_message == message1, 'invalid owner 3') { }
+    assert(field4.owner_message == message2, 'invalid owner 4') { }
+    assert(field5.owner_message == message2, 'invalid owner 5') { }
+  end
+
   def test_enum
     field0 = Protobuf::ASTEnumConstant.new "f1", 1
     field1 = Protobuf::ASTEnumConstant.new "f2", 2