macks-ruby_protobuf 0.3.2.1

data/test/data/types_source.py

@@ -0,0 +1,22 @@
+import types_pb2
+import sys
+
+types = types_pb2.TestTypes()
+types.type1 = 0.01
+types.type2 = 0.1
+types.type3 = 1
+types.type4 = 10
+types.type5 = 100
+types.type6 = 1000
+types.type7 = -1
+types.type8 = -10
+types.type9 = 10000
+types.type10 = 100000
+types.type11 = False
+types.type12 = 'hello all types'
+# TODO test type13
+#types.type13 = 
+
+f = open('types.bin', 'wb')
+f.write(types.SerializeToString())
+f.close()

data/test/ext_collision.rb

@@ -0,0 +1,25 @@
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/proto/ext_collision.proto>
+# message ExtCollisionTest {
+#   extensions 1 to 10;
+# }
+# 
+# message ExtCollisionTest {
+#   optional string a = 1;
+#   optional string b = 1;
+# }
+
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+class ExtCollisionTest < ::Protobuf::Message
+  defined_in __FILE__
+  extensions 1..10
+end
+class ExtCollisionTest < ::Protobuf::Message
+  defined_in __FILE__
+  optional :string, :a, 1
+  optional :string, :b, 1
+end

data/test/ext_range.rb

@@ -0,0 +1,23 @@
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/proto/ext_range.proto>
+# message ExtCollisionTest {
+#   extensions 1 to 10;
+# }
+# 
+# message ExtCollisionTest {
+#   optional string a = 11;
+# }
+
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+class ExtCollisionTest < ::Protobuf::Message
+  defined_in __FILE__
+  extensions 1..10
+end
+class ExtCollisionTest < ::Protobuf::Message
+  defined_in __FILE__
+  optional :string, :a, 11, :extension => true
+end

data/test/merge.rb

@@ -0,0 +1,40 @@
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/merge.proto>
+# package test;
+# 
+# message MergeMessage {
+#   message InnerMessage1 {
+#     required string name = 1;
+#   }
+#   message InnerMessage2 {
+#     required string name = 1;
+#     repeated InnerMessage1 repeate_message = 2;
+#   }
+# 
+#   required string name = 1;
+#   repeated InnerMessage1 repeate_message = 2;
+#   required InnerMessage2 require_message = 3;
+# }
+
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+module Test
+  class MergeMessage < ::Protobuf::Message
+    defined_in __FILE__
+    class InnerMessage1 < ::Protobuf::Message
+      defined_in __FILE__
+      required :string, :name, 1
+    end
+    class InnerMessage2 < ::Protobuf::Message
+      defined_in __FILE__
+      required :string, :name, 1
+      repeated :InnerMessage1, :repeate_message, 2
+    end
+    required :string, :name, 1
+    repeated :InnerMessage1, :repeate_message, 2
+    required :InnerMessage2, :require_message, 3
+  end
+end

data/test/nested.rb

@@ -0,0 +1,25 @@
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/nested.proto>
+# message Foo {  
+#   message Bar {
+#   } 
+# }
+# message Baaz {  
+#   optional Foo.Bar x = 1;
+# }
+
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+class Foo < ::Protobuf::Message
+  defined_in __FILE__
+  class Bar < ::Protobuf::Message
+    defined_in __FILE__
+  end
+end
+class Baaz < ::Protobuf::Message
+  defined_in __FILE__
+  optional :'Foo::Bar', :x, 1
+end

data/test/proto/addressbook.proto

@@ -0,0 +1,31 @@
+package tutorial;
+
+message Person {
+  required string name = 1;
+  required int32 id = 2;
+  optional string email = 3;
+
+  enum PhoneType {
+    MOBILE = 0;
+    HOME = 1;
+    WORK = 2;
+  }
+
+  message PhoneNumber {
+    required string number = 1;
+    optional PhoneType type = 2 [default = HOME];
+  }
+
+  repeated PhoneNumber phone = 4;
+  optional uint32 age = 5 [default = 20];
+
+  extensions 100 to 200;
+}
+
+extend Person {
+  optional int32 age = 100;
+}
+
+message AddressBook {
+  repeated Person person = 1;
+}

data/test/proto/addressbook_base.proto

@@ -0,0 +1,26 @@
+package tutorial;
+
+message Person {
+  required string name = 1;
+  required int32 id = 2;
+  optional string email = 3;
+
+  enum PhoneType {
+    MOBILE = 0;
+    HOME = 1;
+    WORK = 2;
+  }
+
+  message PhoneNumber {
+    required string number = 1;
+    optional PhoneType type = 2 [default = HOME];
+  }
+
+  repeated PhoneNumber phone = 4;
+
+  extensions 100 to 200;
+}
+
+message AddressBook {
+  repeated Person person = 1;
+}

data/test/proto/addressbook_ext.proto

@@ -0,0 +1,6 @@
+import "test/addressbook_base.proto";
+package tutorial;
+
+extend Person {
+  optional int32 age = 100;
+}

data/test/proto/collision.proto

@@ -0,0 +1,5 @@
+message CollisionTest {
+  optional string a = 1;
+  optional string b = 1;
+}
+

data/test/proto/ext_collision.proto

@@ -0,0 +1,8 @@
+message ExtCollisionTest {
+  extensions 1 to 10;
+}
+
+message ExtCollisionTest {
+  optional string a = 1;
+  optional string b = 1;
+}

data/test/proto/ext_range.proto

@@ -0,0 +1,7 @@
+message ExtCollisionTest {
+  extensions 1 to 10;
+}
+
+extend ExtCollisionTest {
+  optional string a = 11;
+}

data/test/proto/merge.proto

@@ -0,0 +1,15 @@
+package test;
+
+message MergeMessage {
+  message InnerMessage1 {
+    required string name = 1;
+  }
+  message InnerMessage2 {
+    required string name = 1;
+    repeated InnerMessage1 repeate_message = 2;
+  }
+
+  required string name = 1;
+  repeated InnerMessage1 repeate_message = 2;
+  required InnerMessage2 require_message = 3;
+}

data/test/proto/nested.proto

@@ -0,0 +1,7 @@
+message Foo {  
+  message Bar {
+  } 
+}
+message Baaz {  
+  optional Foo.Bar x = 1;
+}

data/test/proto/rpc.proto

@@ -0,0 +1,6 @@
+package tutorial;
+
+service AddressBookService {
+  rpc Search (Person) returns (AddressBook);
+  rpc Add (Person) returns (Person);
+}

data/test/proto/types.proto

@@ -0,0 +1,17 @@
+package test.types;
+
+message TestTypes {
+  required double type1 = 1;
+  required float type2 = 2;
+  required int32 type3 = 3;
+  required int64 type4 = 4;
+  required uint32 type5 = 5;
+  required uint64 type6 = 6;
+  required sint32 type7 = 7;
+  required sint64 type8 = 8;
+  required fixed32 type9 = 9;
+  required fixed64 type10 = 10;
+  required bool type11 = 11;
+  required string type12 = 12;
+  required bytes type13 = 13;
+}

data/test/test_addressbook.rb

@@ -0,0 +1,43 @@
+require 'test/unit'
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'test/addressbook'
+
+class AddressbookTest < Test::Unit::TestCase
+  def test_enum
+    phone_number = Tutorial::Person::PhoneNumber.new
+    assert_equal Tutorial::Person::PhoneType::HOME, phone_number.type
+    phone_number.type = Tutorial::Person::PhoneType::MOBILE
+    assert_equal 0, phone_number.type
+    phone_number.type = Tutorial::Person::PhoneType::HOME
+    assert_equal 1, phone_number.type
+    phone_number.type = Tutorial::Person::PhoneType::WORK
+    assert_equal 2, phone_number.type
+    assert_raise TypeError do
+      phone_number.type = 3
+    end
+  end
+
+  def test_initial_value
+    person = Tutorial::Person.new
+    assert_nil person.name
+    assert_nil person.id
+    assert_equal [], person.phone
+    assert_equal '', person.email
+  end
+
+  def test_repeatable
+    address_book = Tutorial::AddressBook.new
+    assert_equal [], address_book.person
+    assert_instance_of Protobuf::Field::FieldArray, address_book.person
+    address_book.person << Tutorial::Person.new
+    assert_equal 1, address_book.person.size
+    assert_raise TypeError do
+      address_book.person << 1
+    end
+    assert_equal 1, address_book.person.size
+    address_book.person << Tutorial::Person.new
+    assert_equal 2, address_book.person.size
+  end
+end
+

data/test/test_compiler.rb

@@ -0,0 +1,313 @@
+require 'test/unit'
+#require 'protobuf/compiler/compiler_old'
+require 'protobuf/compiler/compiler'
+
+class CompilerTest < Test::Unit::TestCase
+  # Issue 12:    Parse error on float default value
+  def test_compile_float_default
+    assert_compile_proto <<-eos, 'test/proto/float_default.proto'
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/proto/float_default.proto>
+# message M {  optional float f = 1 [default = 4.2];  }
+# 
+
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+class M < ::Protobuf::Message
+  defined_in __FILE__
+  optional :float, :f, 1, :default => 4.2
+end
+    eos
+  end
+
+  def test_create_message
+    assert_compile_proto <<-eos, 'test/proto/addressbook.proto'
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/proto/addressbook.proto>
+# package tutorial;
+# 
+# message Person {
+#   required string name = 1;
+#   required int32 id = 2;
+#   optional string email = 3;
+# 
+#   enum PhoneType {
+#     MOBILE = 0;
+#     HOME = 1;
+#     WORK = 2;
+#   }
+# 
+#   message PhoneNumber {
+#     required string number = 1;
+#     optional PhoneType type = 2 [default = HOME];
+#   }
+# 
+#   repeated PhoneNumber phone = 4;
+#   optional uint32 age = 5 [default = 20];
+# 
+#   extensions 100 to 200;
+# }
+# 
+# extend Person {
+#   optional int32 age = 100;
+# }
+# 
+# message AddressBook {
+#   repeated Person person = 1;
+# }
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+module Tutorial
+  
+  class Person < ::Protobuf::Message
+    defined_in __FILE__
+    required :string, :name, 1
+    required :int32, :id, 2
+    optional :string, :email, 3
+    
+    class PhoneType < ::Protobuf::Enum
+      defined_in __FILE__
+      MOBILE = 0
+      HOME = 1
+      WORK = 2
+    end
+    
+    class PhoneNumber < ::Protobuf::Message
+      defined_in __FILE__
+      required :string, :number, 1
+      optional :PhoneType, :type, 2, :default => :HOME
+    end
+    
+    repeated :PhoneNumber, :phone, 4
+    optional :uint32, :age, 5, :default => 20
+    
+    extensions 100..200
+  end
+  
+  class Person < ::Protobuf::Message
+    defined_in __FILE__
+    optional :int32, :age, 100, :extension => true
+  end
+  
+  class AddressBook < ::Protobuf::Message
+    defined_in __FILE__
+    repeated :Person, :person, 1
+  end
+end
+    eos
+  end
+
+  def test_create_nested_message
+    assert_compile_proto <<-eos, 'test/proto/nested.proto'
+### Generated by rprotoc. DO NOT EDIT!
+### <proto file: test/proto/nested.proto>
+# message Foo {  
+#   message Bar {
+#   } 
+# }
+# message Baaz {  
+#   optional Foo.Bar x = 1;
+# }
+require 'protobuf/message/message'
+require 'protobuf/message/enum'
+require 'protobuf/message/service'
+require 'protobuf/message/extend'
+
+class Foo < ::Protobuf::Message
+  defined_in __FILE__
+  class Bar < ::Protobuf::Message
+    defined_in __FILE__
+  end
+end
+
+class Baaz < ::Protobuf::Message
+  defined_in __FILE__
+  optional :'Foo::Bar', :x, 1
+end
+    eos
+  end
+
+  def test_nested_message
+    Protobuf::Visitor::CreateMessageVisitor.instance_eval {remove_const :Baaz}
+    file_contents = Protobuf::Compiler.new.create_message('test/proto/nested.proto', '.', '.', false)
+    assert_nothing_raised {eval file_contents}
+    assert_raise(TypeError) {Baaz.new.x = 1}
+    assert_nothing_raised {Baaz.new.x = Foo::Bar.new}
+  end
+
+  def test_create_rpc
+    file_contents = Protobuf::Compiler.new.create_rpc('test/proto/rpc.proto', '.', '.', false)
+
+    assert_source <<-eos, file_contents['./test/proto/address_book_service.rb']
+require 'protobuf/rpc/server'
+require 'protobuf/rpc/handler'
+require 'test/proto/rpc.pb'
+
+class Tutorial::SearchHandler < Protobuf::Rpc::Handler
+  request Tutorial::Person
+  response Tutorial::AddressBook
+
+  def self.process_request(request, response)
+    # TODO: edit this method
+  end
+end
+
+class Tutorial::AddHandler < Protobuf::Rpc::Handler
+  request Tutorial::Person
+  response Tutorial::Person
+
+  def self.process_request(request, response)
+    # TODO: edit this method
+  end
+end
+
+class Tutorial::AddressBookService < Protobuf::Rpc::Server
+  def setup_handlers
+    @handlers = {
+      :search => Tutorial::SearchHandler,
+      :add => Tutorial::AddHandler,
+    }
+  end
+end
+    eos
+
+    assert_source <<-eos, file_contents['./test/proto/start_address_book_service']
+#!/usr/bin/env ruby
+require 'address_book_service'
+
+Tutorial::AddressBookService.new(:port => 9999).start
+    eos
+
+    assert_source <<-eos, file_contents['./test/proto/client_search.rb']
+#!/usr/bin/env ruby
+require 'protobuf/rpc/client'
+require 'test/proto/rpc.pb'
+
+# build request
+request = Tutorial::Person.new
+# TODO: setup a request
+raise StandardError.new('setup a request')
+
+# create blunk response
+response = Tutorial::AddressBook.new
+
+# execute rpc
+Protobuf::Rpc::Client.new('localhost', 9999).call :search, request, response
+
+# show response
+puts response
+    eos
+
+    assert_source <<-eos, file_contents['./test/proto/client_add.rb']
+#!/usr/bin/env ruby
+require 'protobuf/rpc/client'
+require 'test/proto/rpc.pb'
+
+# build request
+request = Tutorial::Person.new
+# TODO: setup a request
+raise StandardError.new('setup a request')
+
+# create blunk response
+response = Tutorial::Person.new
+
+# execute rpc
+Protobuf::Rpc::Client.new('localhost', 9999).call :add, request, response
+
+# show response
+puts response
+    eos
+  end
+
+  def test_create_descriptor
+    proto_path = 'test/proto/addressbook.proto'
+    visitor = Protobuf::Visitor::CreateDescriptorVisitor.new proto_path
+    File.open proto_path, 'r' do |file|
+      visitor.visit Protobuf::ProtoParser.new.parse(file)
+    end
+    file_descriptor = visitor.file_descriptor
+    assert_equal proto_path, file_descriptor.name
+    assert_equal 'tutorial', file_descriptor.package
+ 
+    person_descriptor = file_descriptor.message_type[0]
+    assert_equal 'Person', person_descriptor.name
+    assert_equal [:name, :id, :email, :phone, :age].size, person_descriptor.field.size
+
+    name_field_descriptor = person_descriptor.field.find {|d| d.name == 'name'}
+    assert_equal 1, name_field_descriptor.number
+    assert_equal Google::Protobuf::FieldDescriptorProto::Type::TYPE_STRING, name_field_descriptor.type
+    assert_equal Google::Protobuf::FieldDescriptorProto::Label::LABEL_REQUIRED, name_field_descriptor.label
+    assert_equal 'string', name_field_descriptor.type_name
+
+    phone_field_descriptor = person_descriptor.field.find {|d| d.name == 'phone'}
+    assert_equal 4, phone_field_descriptor.number
+    assert_equal 0, phone_field_descriptor.type #TODO: is this right?
+    assert_equal Google::Protobuf::FieldDescriptorProto::Label::LABEL_REPEATED, phone_field_descriptor.label
+    assert_equal 'PhoneNumber', phone_field_descriptor.type_name
+
+    age_field_descriptor = person_descriptor.field.find {|d| d.name == 'age'}
+    assert_equal 5, age_field_descriptor.number
+    assert_equal Google::Protobuf::FieldDescriptorProto::Type::TYPE_UINT32, age_field_descriptor.type
+    assert_equal Google::Protobuf::FieldDescriptorProto::Label::LABEL_OPTIONAL, age_field_descriptor.label
+    assert_equal 'uint32', age_field_descriptor.type_name
+    assert_equal '20', age_field_descriptor.default_value
+
+    phone_type_descriptor = person_descriptor.enum_type.first
+    assert_equal 'PhoneType', phone_type_descriptor.name
+    assert_equal 3, phone_type_descriptor.value.size
+
+    phone_type_home_descriptor = phone_type_descriptor.value.find {|d| d.name == 'HOME'}
+    assert_equal 'HOME', phone_type_home_descriptor.name
+    assert_equal 1, phone_type_home_descriptor.number
+
+    extensions_descriptor = person_descriptor.extension_range.first
+    assert_equal 100, extensions_descriptor.start
+    assert_equal 200, extensions_descriptor.end
+
+    phone_number_descriptor = person_descriptor.nested_type.first
+    assert_equal 'PhoneNumber', phone_number_descriptor.name
+    assert_equal [:number, :type].size, phone_number_descriptor.field.size
+
+    #TODO: test extend
+    #extend_person_descriptor = ??
+    #assert_equal extend_person_descriptor
+
+    addressbook_descriptor = file_descriptor.message_type[1]
+    assert_equal 'AddressBook', addressbook_descriptor.name
+  end
+
+  def test_collision
+    assert_raise Protobuf::Message::TagCollisionError do require 'test/collision' end
+    assert_raise Protobuf::Message::TagCollisionError do 
+      Protobuf::Compiler.new.create_message('test/proto/collision.proto', '.', '.', false) 
+    end
+  end
+
+  def test_ext_collision
+    assert_raise Protobuf::Message::TagCollisionError do require 'test/ext_collision' end
+    assert_raise Protobuf::Message::TagCollisionError do 
+      Protobuf::Compiler.new.create_message('test/proto/ext_collision.proto', '.', '.', false) 
+    end
+  end
+
+  def test_ext_range
+    assert_raise RangeError do require 'test/ext_range' end
+    assert_raise RangeError do 
+      Protobuf::Compiler.new.create_message('test/proto/ext_range.proto', '.', '.', false) 
+    end
+  end
+
+  def assert_compile_proto(ideal, filename)
+    assert_equal ideal.gsub(/^\s*\n/, '').strip, Protobuf::Compiler.new.create_message(filename, '.', '.', false).gsub(/^\s*\n/, '').strip
+  end
+
+  def assert_source(ideal, real)
+    assert_equal ideal.strip.gsub(/^\s*\n/, '').gsub(/\s+\n/, "\n"), real.strip.gsub(/^\s*\n/, '').gsub(/\s+\n/, "\n")
+  end
+end