deimos-ruby 1.24.3 → 2.0.0.pre.alpha1

data/spec/karafka_config/karafka_spec.rb

@@ -0,0 +1,97 @@
+RSpec.describe 'Karafka configs' do
+  before(:each) do
+    KarafkaApp.routes.clear
+    $found_stuff = nil
+  end
+
+  let(:consumer_class) do
+    Class.new(Deimos::Consumer) do
+      def consume_message(message)
+        $found_stuff = message.payload
+      end
+    end
+  end
+
+  let(:producer_class) do
+    Class.new(Deimos::Producer) do
+    end
+  end
+
+  describe 'producers' do
+    before(:each) do
+      stub_const('MyProducer', producer_class)
+    end
+
+    it 'should work with key none' do
+      KarafkaApp.routes.draw do
+        topic 'MyTopic' do
+          producer_class MyProducer
+          schema 'MySchema'
+          namespace 'com.my-namespace'
+          key_config(none: true)
+        end
+      end
+      producer_class.publish({test_id: "id1", some_int: 5})
+      expect('MyTopic').to have_sent({test_id: "id1", some_int: 5})
+    end
+
+    it 'should work with key plain' do
+      KarafkaApp.routes.draw do
+        topic 'MyTopic' do
+          producer_class MyProducer
+          schema 'MySchema'
+          namespace 'com.my-namespace'
+          key_config({plain: true})
+        end
+      end
+      producer_class.publish({test_id: "id1", some_int: 5, payload_key: 'key'})
+      expect('MyTopic').to have_sent({test_id: "id1", some_int: 5}, 'key')
+    end
+
+    it 'should work with key field' do
+      KarafkaApp.routes.draw do
+        topic 'MyTopic' do
+          producer_class MyProducer
+          schema 'MySchema'
+          namespace 'com.my-namespace'
+          key_config({field: :test_id})
+        end
+      end
+      producer_class.publish({test_id: "id1", some_int: 5})
+      expect('MyTopic').to have_sent({test_id: "id1", some_int: 5}, 'id1')
+    end
+
+    it 'should work with key schema' do
+      KarafkaApp.routes.draw do
+        topic 'MyTopic' do
+          producer_class MyProducer
+          schema 'MySchema'
+          namespace 'com.my-namespace'
+          key_config({schema: 'MySchema_key'})
+        end
+      end
+      producer_class.publish({test_id: "id1", some_int: 5, payload_key: {test_id: 'id3'}})
+      expect('MyTopic').to have_sent({test_id: "id1", some_int: 5}, { test_id: 'id3'})
+    end
+
+  end
+
+  it 'should be able to pick up a consumer' do
+    stub_const('MyConsumer', consumer_class)
+    KarafkaApp.routes.draw do
+      topic 'MyTopic' do
+        consumer MyConsumer
+        schema 'MySchema'
+        namespace 'com.my-namespace'
+        key_config({field: :test_id})
+      end
+    end
+
+    test_consume_message('MyTopic', {test_id: "id1", some_int: 5}, key: "id1")
+    expect($found_stuff).to eq({'test_id' => "id1", 'some_int' => 5})
+    $found_stuff = nil
+    test_consume_message(MyConsumer, {test_id: "id1", some_int: 5}, key: "id1")
+    expect($found_stuff).to eq({'test_id' => "id1", 'some_int' => 5})
+  end
+
+end

data/spec/logging_spec.rb

@@ -0,0 +1,25 @@
+RSpec.describe Deimos::Logging do
+  include_context 'with publish_backend'
+  describe '#messages_log_text' do
+    it 'should return whole payload (default behavior)' do
+      log_message = described_class.messages_log_text(:payloads, messages)
+      expect(log_message[:payloads].count).to eq(3)
+      expect(log_message[:payloads].first[:payload]).to eq({ some_int: 1, test_id: 'foo1' })
+      expect(log_message[:payloads].first[:key]).to eq('foo1')
+    end
+
+    it 'should return only keys of messages' do
+      log_message = described_class.messages_log_text(:keys, messages)
+      expect(log_message[:payload_keys].count).to eq(3)
+      expect(log_message[:payload_keys]).to be_a(Array)
+      expect(log_message[:payload_keys].first).to eq('foo1')
+    end
+
+    it 'should return only messages count' do
+      log_message = described_class.messages_log_text(:count, messages)
+      expect(log_message[:payloads_count]).to be_a(Integer)
+      expect(log_message[:payloads_count]).to eq(3)
+    end
+  end
+
+end

data/spec/message_spec.rb

@@ -2,37 +2,37 @@
 
 RSpec.describe(Deimos::Message) do
   it 'should detect tombstones' do
-    expect(described_class.new(nil, nil, key: 'key1')).
+    expect(described_class.new(nil, key: 'key1')).
       to be_tombstone
-    expect(described_class.new({ v: 'val1' }, nil, key: 'key1')).
+    expect(described_class.new({ v: 'val1' }, key: 'key1')).
       not_to be_tombstone
-    expect(described_class.new({ v: '' }, nil, key: 'key1')).
+    expect(described_class.new({ v: '' }, key: 'key1')).
       not_to be_tombstone
-    expect(described_class.new({ v: 'val1' }, nil, key: nil)).
+    expect(described_class.new({ v: 'val1' }, key: nil)).
       not_to be_tombstone
   end
 
   it 'can support complex keys/values' do
-    expect { described_class.new({ a: 1, b: 2 }, nil, key: { c: 3, d: 4 }) }.
+    expect { described_class.new({ a: 1, b: 2 }, key: { c: 3, d: 4 }) }.
       not_to raise_exception
   end
 
   describe 'headers' do
     it 'returns nil when not set' do
-      expect(described_class.new({ v: 'val1' }, nil, key: 'key1')).
+      expect(described_class.new({ v: 'val1' }, key: 'key1')).
         to have_attributes(headers: nil)
     end
 
     it 'can set and get headers' do
-      expect(described_class.new({ v: 'val1' }, nil, key: 'key1', headers: { a: 1 })).
+      expect(described_class.new({ v: 'val1' }, key: 'key1', headers: { a: 1 })).
         to have_attributes(headers: { a: 1 })
     end
 
     it 'includes headers when converting to Hash' do
-      expect(described_class.new({ v: 'val1' }, nil, key: 'key1', headers: { a: 1 }).to_h).
+      expect(described_class.new({ v: 'val1' }, key: 'key1', headers: { a: 1 }).to_h).
         to include(headers: { a: 1 })
 
-      expect(described_class.new({ v: 'val1' }, nil, key: 'key1', headers: { a: 1 }).encoded_hash).
+      expect(described_class.new({ v: 'val1' }, key: 'key1', headers: { a: 1 }).encoded_hash).
         to include(headers: { a: 1 })
     end
   end