deimos-ruby 2.1.12 → 2.2.0.pre.beta1

data/spec/gen/sample/v1/sample_pb.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: sample/v1/sample.proto
+
+require 'google/protobuf'
+
+require 'google/protobuf/timestamp_pb'
+
+
+descriptor_data = "\n\x16sample/v1/sample.proto\x12\tsample.v1\x1a\x1fgoogle/protobuf/timestamp.proto\"7\n\rNestedMessage\x12\x12\n\nnested_str\x18\x01 \x01(\t\x12\x12\n\nnested_num\x18\x02 \x01(\x05\"\xdb\x02\n\rSampleMessage\x12\x0b\n\x03str\x18\x01 \x01(\t\x12\x0b\n\x03num\x18\x02 \x01(\x05\x12\x0f\n\x07str_arr\x18\x03 \x03(\t\x12\x0c\n\x04\x66lag\x18\x04 \x01(\x08\x12-\n\ttimestamp\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12*\n\x06nested\x18\x06 \x01(\x0b\x32\x18.sample.v1.NestedMessageH\x00\x12\x13\n\tunion_str\x18\x07 \x01(\tH\x00\x12\x32\n\x10non_union_nested\x18\x08 \x01(\x0b\x32\x18.sample.v1.NestedMessage\x12\x35\n\x07str_map\x18\t \x03(\x0b\x32$.sample.v1.SampleMessage.StrMapEntry\x1a-\n\x0bStrMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x07\n\x05unionb\x06proto3"
+
+pool = Google::Protobuf::DescriptorPool.generated_pool
+pool.add_serialized_file(descriptor_data)
+
+module Sample
+  module V1
+    NestedMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("sample.v1.NestedMessage").msgclass
+    SampleMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("sample.v1.SampleMessage").msgclass
+  end
+end

data/spec/generators/schema_class/my_schema_with_complex_types_spec.rb

@@ -67,7 +67,7 @@ RSpec.describe Schemas::MyNamespace::MySchemaWithComplexType do
     let(:schema_fields) do
       %w(test_id test_float test_int_array test_optional_int test_string_array some_integer_map
          some_record some_optional_record some_record_array some_record_map some_enum_array
-         some_optional_enum some_enum_with_default)
+         some_optional_enum some_enum_with_default union_string)
     end
 
     it 'should return the name of the schema and namespace' do
@@ -101,14 +101,15 @@ RSpec.describe Schemas::MyNamespace::MySchemaWithComplexType do
           'record_2' => { 'a_record_field' => 'field 6' }
         },
         'some_enum_array' => %w(sym1 sym2),
-        'some_enum_with_default' => 'sym6'
+        'some_enum_with_default' => 'sym6',
+        'union_string' => ''
       }
 
       expect(klass.as_json).to eq(payload_h)
     end
 
     it 'should return a JSON string of the payload' do
-      s = '{"test_id":"test id","test_float":1.2,"test_string_array":["abc","def"],"test_int_array":[123,456],"test_optional_int":123,"some_integer_map":{"int_1":1,"int_2":2},"some_record":{"a_record_field":"field 1"},"some_optional_record":{"a_record_field":"field 2"},"some_record_array":[{"a_record_field":"field 3"},{"a_record_field":"field 4"}],"some_record_map":{"record_1":{"a_record_field":"field 5"},"record_2":{"a_record_field":"field 6"}},"some_enum_array":["sym1","sym2"],"some_optional_enum":null,"some_enum_with_default":"sym6"}'
+      s = '{"test_id":"test id","union_string":"","test_float":1.2,"test_string_array":["abc","def"],"test_int_array":[123,456],"test_optional_int":123,"some_integer_map":{"int_1":1,"int_2":2},"some_record":{"a_record_field":"field 1"},"some_optional_record":{"a_record_field":"field 2"},"some_record_array":[{"a_record_field":"field 3"},{"a_record_field":"field 4"}],"some_record_map":{"record_1":{"a_record_field":"field 5"},"record_2":{"a_record_field":"field 6"}},"some_enum_array":["sym1","sym2"],"some_optional_enum":null,"some_enum_with_default":"sym6"}'
       expect(klass.to_json).to eq(s)
     end
   end
@@ -209,9 +210,9 @@ RSpec.describe Schemas::MyNamespace::MySchemaWithComplexType do
         expect(klass.some_enum_array.first).
           to eq(described_class::AnEnum.new('new_sym'))
 
-        klass.some_enum_array.second.an_enum = described_class::AnEnum.
+        klass.some_enum_array.second.value = described_class::AnEnum.
           new('other_sym')
-        expect(klass.some_enum_array.second.an_enum).to eq('other_sym')
+        expect(klass.some_enum_array.second.value).to eq('other_sym')
       end
 
       it 'should modify the value of some_record_map' do

data/spec/protos/sample/v1/sample.proto

@@ -0,0 +1,24 @@
+syntax = "proto3";
+
+package sample.v1;
+
+import "google/protobuf/timestamp.proto";
+
+message NestedMessage {
+  string nested_str = 1;
+  int32 nested_num = 2;
+}
+
+message SampleMessage {
+  string str = 1;
+  int32 num = 2;
+  repeated string str_arr = 3;
+  bool flag = 4;
+  google.protobuf.Timestamp timestamp = 5;
+  oneof union {
+    NestedMessage nested = 6;
+    string union_str = 7;
+  }
+  NestedMessage non_union_nested = 8;
+  map<string, string> str_map = 9;
+}

data/spec/schema_backends/proto_schema_registry_spec.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require 'deimos/schema_backends/proto_schema_registry'
+require_relative "#{__dir__}/../gen/sample/v1/sample_pb"
+
+RSpec.describe Deimos::SchemaBackends::ProtoSchemaRegistry do
+  let(:payload) do
+    Sample::V1::SampleMessage.new(
+      str: "string",
+      num: 123,
+      str_arr: %w[one two],
+      flag: true,
+      timestamp: Time.utc(2017, 1, 1),
+      nested: Sample::V1::NestedMessage.new(nested_str: "string"),
+      non_union_nested: Sample::V1::NestedMessage.new(nested_num: 456),
+      str_map: {'foo' => 'bar'}
+    )
+  end
+
+  let(:backend) { described_class.new(schema: 'sample.v1.SampleMessage') }
+
+  specify('#encode_key') do
+    expect(backend.encode_key(nil, 789)).to eq('789')
+    expect(backend.encode_key(nil, 'string')).to eq('string')
+    expect(backend.encode_key(nil, {foo: 'bar'})).to eq('{"foo":"bar"}')
+    expect(backend.encode_key(:foo, 'bar')).to eq('bar')
+  end
+
+  specify('#decode_key') do
+    expect(backend.decode_key('789', nil)).to eq(789)
+    expect(backend.decode_key('{"foo":"bar"}', :foo)).to eq('bar')
+    expect(backend.decode_key('{"foo":"bar"}', nil)).to eq({"foo" => 'bar'})
+  end
+
+  it 'should encode and decode correctly' do
+    proto_turf = instance_double(ProtoTurf)
+    expect(proto_turf).to receive(:encode).
+      with(payload, subject: 'topic').
+      and_return('encoded-payload')
+    expect(proto_turf).to receive(:decode).
+      with('encoded-payload').
+      and_return(payload)
+    allow(described_class).to receive(:proto_turf).and_return(proto_turf)
+    results = backend.encode(payload, topic: 'topic')
+    expect(results).to eq('encoded-payload')
+    results = backend.decode(results)
+    expect(results).to eq(payload)
+  end
+
+end

data/spec/schemas/com/my-namespace/MySchemaWithComplexTypes.avsc

@@ -9,6 +9,11 @@
       "type": "string",
       "doc": "test string"
     },
+    {
+      "name": "union_string",
+      "type": ["string", "null"],
+      "default": ""
+    },
     {
       "name": "test_float",
       "type": "float",

data/spec/schemas/my_namespace/generated.rb

@@ -41,13 +41,20 @@ module Schemas; module MyNamespace
 
     # Autogenerated Schema for Enum at com.my-namespace.AnEnum
     class AnEnum < Deimos::SchemaClass::Enum
-      # @return ['sym1', 'sym2']
-      attr_accessor :an_enum
-  
       # @override
       def symbols
         %w(sym1 sym2)
       end
+  
+      
+      def sym1?
+        @value == 'sym1'
+      end
+      
+      def sym2?
+        @value == 'sym2'
+      end
+      
     end
 
 

data/spec/schemas/my_namespace/my_schema_with_complex_type.rb

@@ -41,35 +41,56 @@ module Schemas; module MyNamespace
 
     # Autogenerated Schema for Enum at com.my-namespace.AnEnum
     class AnEnum < Deimos::SchemaClass::Enum
-      # @return ['sym1', 'sym2']
-      attr_accessor :an_enum
-  
       # @override
       def symbols
         %w(sym1 sym2)
       end
+  
+      
+      def sym1?
+        @value == 'sym1'
+      end
+      
+      def sym2?
+        @value == 'sym2'
+      end
+      
     end
 
     # Autogenerated Schema for Enum at com.my-namespace.AnotherEnum
     class AnotherEnum < Deimos::SchemaClass::Enum
-      # @return ['sym3', 'sym4']
-      attr_accessor :another_enum
-  
       # @override
       def symbols
         %w(sym3 sym4)
       end
+  
+      
+      def sym3?
+        @value == 'sym3'
+      end
+      
+      def sym4?
+        @value == 'sym4'
+      end
+      
     end
 
     # Autogenerated Schema for Enum at com.my-namespace.YetAnotherEnum
     class YetAnotherEnum < Deimos::SchemaClass::Enum
-      # @return ['sym5', 'sym6']
-      attr_accessor :yet_another_enum
-  
       # @override
       def symbols
         %w(sym5 sym6)
       end
+  
+      
+      def sym5?
+        @value == 'sym5'
+      end
+      
+      def sym6?
+        @value == 'sym6'
+      end
+      
     end
 
 
@@ -92,6 +113,8 @@ module Schemas; module MyNamespace
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -147,6 +170,7 @@ module Schemas; module MyNamespace
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -162,6 +186,7 @@ module Schemas; module MyNamespace
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -197,6 +222,7 @@ module Schemas; module MyNamespace
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers-no-nest.snap

@@ -749,6 +749,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -804,6 +806,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -819,6 +822,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -847,6 +851,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers.snap

@@ -825,6 +825,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -880,6 +882,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -895,6 +898,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -923,6 +927,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_and_producers-no-nest.snap

@@ -749,6 +749,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -804,6 +806,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -819,6 +822,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -847,6 +851,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_and_producers.snap

@@ -825,6 +825,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -880,6 +882,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -895,6 +898,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -923,6 +927,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_circular-no-nest.snap

@@ -749,6 +749,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -804,6 +806,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -819,6 +822,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -847,6 +851,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_circular.snap

@@ -825,6 +825,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -880,6 +882,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -895,6 +898,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -923,6 +927,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_complex_types-no-nest.snap

@@ -749,6 +749,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -804,6 +806,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -819,6 +822,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -847,6 +851,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_complex_types.snap

@@ -825,6 +825,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -880,6 +882,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -895,6 +898,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -923,6 +927,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_nested-no-nest.snap

@@ -749,6 +749,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -804,6 +806,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -819,6 +822,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -847,6 +851,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/consumers_nested.snap

@@ -825,6 +825,8 @@ module Schemas
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -880,6 +882,7 @@ module Schemas
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -895,6 +898,7 @@ module Schemas
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -923,6 +927,7 @@ module Schemas
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/namespace_folders.snap

@@ -700,6 +700,8 @@ module Schemas; module Com; module MyNamespace
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -755,6 +757,7 @@ module Schemas; module Com; module MyNamespace
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -770,6 +773,7 @@ module Schemas; module Com; module MyNamespace
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -798,6 +802,7 @@ module Schemas; module Com; module MyNamespace
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,

data/spec/snapshots/namespace_map.snap

@@ -745,6 +745,8 @@ module Schemas; module MyNamespace
     ### Attribute Accessors ###
     # @return [String]
     attr_accessor :test_id
+    # @return [String, nil]
+    attr_accessor :union_string
     # @return [Float]
     attr_accessor :test_float
     # @return [Array<String>]
@@ -800,6 +802,7 @@ module Schemas; module MyNamespace
 
     # @override
     def initialize(_from_message: false, test_id: nil,
+                   union_string: "",
                    test_float: nil,
                    test_string_array: ["test"],
                    test_int_array: [123],
@@ -815,6 +818,7 @@ module Schemas; module MyNamespace
       @_from_message = _from_message
       super
       self.test_id = test_id
+      self.union_string = union_string
       self.test_float = test_float
       self.test_string_array = test_string_array
       self.test_int_array = test_int_array
@@ -843,6 +847,7 @@ module Schemas; module MyNamespace
     def as_json(_opts={})
       {
         'test_id' => @test_id,
+        'union_string' => @union_string,
         'test_float' => @test_float,
         'test_string_array' => @test_string_array,
         'test_int_array' => @test_int_array,