kafka-consumer 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,48 @@
1
+ require 'minitest/autorun'
2
+ require 'kafka/consumer'
3
+ require 'kazoo'
4
+ require 'mocha/mini_test'
5
+ require 'pp'
6
+
7
+ module MockCluster
8
+ def mock_cluster
9
+ cluster = Kazoo::Cluster.new('example.com:2181')
10
+ cluster.stubs(:zk).returns(mock)
11
+
12
+ cluster.stubs(:brokers).returns(
13
+ 1 => Kazoo::Broker.new(cluster, 1, 'example.com', 9091),
14
+ 2 => Kazoo::Broker.new(cluster, 2, 'example.com', 9092),
15
+ 3 => Kazoo::Broker.new(cluster, 3, 'example.com', 9093),
16
+ )
17
+
18
+ cluster.stubs(:topics).returns(
19
+ 'test.1' => topic_1 = Kazoo::Topic.new(cluster, 'test.1'),
20
+ 'test.4' => topic_4 = Kazoo::Topic.new(cluster, 'test.4')
21
+ )
22
+
23
+ topic_1.stubs(:partitions).returns([
24
+ Kazoo::Partition.new(topic_1, 0, replicas: [cluster.brokers[1], cluster.brokers[2]]),
25
+ ])
26
+
27
+ topic_4.stubs(:partitions).returns([
28
+ Kazoo::Partition.new(topic_4, 0, replicas: [cluster.brokers[2], cluster.brokers[1]]),
29
+ Kazoo::Partition.new(topic_4, 1, replicas: [cluster.brokers[2], cluster.brokers[3]]),
30
+ Kazoo::Partition.new(topic_4, 2, replicas: [cluster.brokers[1], cluster.brokers[3]]),
31
+ Kazoo::Partition.new(topic_4, 3, replicas: [cluster.brokers[3], cluster.brokers[2]]),
32
+ ])
33
+
34
+ topic_1.partitions[0].stubs(:isr).returns([cluster.brokers[1], cluster.brokers[2]])
35
+ topic_4.partitions[0].stubs(:isr).returns([cluster.brokers[2], cluster.brokers[1]])
36
+ topic_4.partitions[1].stubs(:isr).returns([cluster.brokers[2], cluster.brokers[3]])
37
+ topic_4.partitions[2].stubs(:isr).returns([cluster.brokers[1], cluster.brokers[3]])
38
+ topic_4.partitions[3].stubs(:isr).returns([cluster.brokers[3], cluster.brokers[2]])
39
+
40
+ topic_1.partitions[0].stubs(:leader).returns(cluster.brokers[1])
41
+ topic_4.partitions[0].stubs(:leader).returns(cluster.brokers[2])
42
+ topic_4.partitions[1].stubs(:leader).returns(cluster.brokers[2])
43
+ topic_4.partitions[2].stubs(:leader).returns(cluster.brokers[1])
44
+ topic_4.partitions[3].stubs(:leader).returns(cluster.brokers[3])
45
+
46
+ return cluster
47
+ end
48
+ end
@@ -0,0 +1,40 @@
1
+ require 'test_helper'
2
+
3
+ class TopicTest < Minitest::Test
4
+ include MockCluster
5
+
6
+ def setup
7
+ @cluster = mock_cluster
8
+ end
9
+
10
+ def test_from_json
11
+ json_payload = '{"version":1,"partitions":{"2":[1,2,3],"1":[3,1,2],"3":[2,3,1],"0":[3,2,1]}}'
12
+ topic = Kazoo::Topic.from_json(@cluster, 'test.4', JSON.parse(json_payload))
13
+
14
+ assert_equal 4, topic.partitions.length
15
+ assert_equal [3,2,1], topic.partitions[0].replicas.map(&:id)
16
+ assert_equal [3,1,2], topic.partitions[1].replicas.map(&:id)
17
+ assert_equal [1,2,3], topic.partitions[2].replicas.map(&:id)
18
+ assert_equal [2,3,1], topic.partitions[3].replicas.map(&:id)
19
+ end
20
+
21
+ def test_replication_factor
22
+ json_payload = '{"version":1,"partitions":{"2":[1,2,3],"1":[3,1,2],"3":[2,3,1],"0":[3,2,1]}}'
23
+ topic = Kazoo::Topic.from_json(@cluster, 'test.4', JSON.parse(json_payload))
24
+ assert_equal 3, topic.replication_factor
25
+
26
+ json_payload = '{"version":1,"partitions":{"2":[2,3],"1":[2],"3":[2,3,1],"0":[3,2,1]}}'
27
+ topic = Kazoo::Topic.from_json(@cluster, 'test.4', JSON.parse(json_payload))
28
+ assert_equal 1, topic.replication_factor
29
+ end
30
+
31
+ def tets_topic_under_replicated?
32
+ refute @cluster.topics['test.1'].under_replicated?
33
+ refute @cluster.topics['test.1'].partitions[0].under_replicated?
34
+
35
+ @cluster.topics['test.1'].partitions[0].expects(:isr).returns([@cluster.brokers[1]])
36
+
37
+ assert @cluster.topics['test.1'].partitions[0].under_replicated?
38
+ assert @cluster.topics['test.1'].under_replicated?
39
+ end
40
+ end
metadata ADDED
@@ -0,0 +1,161 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: kafka-consumer
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.0.1
5
+ platform: ruby
6
+ authors:
7
+ - Willem van Bergen
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2015-08-09 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: bundler
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - ~>
18
+ - !ruby/object:Gem::Version
19
+ version: '1.7'
20
+ type: :development
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - ~>
25
+ - !ruby/object:Gem::Version
26
+ version: '1.7'
27
+ - !ruby/object:Gem::Dependency
28
+ name: rake
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - ~>
32
+ - !ruby/object:Gem::Version
33
+ version: '10.0'
34
+ type: :development
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - ~>
39
+ - !ruby/object:Gem::Version
40
+ version: '10.0'
41
+ - !ruby/object:Gem::Dependency
42
+ name: minitest
43
+ requirement: !ruby/object:Gem::Requirement
44
+ requirements:
45
+ - - ~>
46
+ - !ruby/object:Gem::Version
47
+ version: '5.0'
48
+ type: :development
49
+ prerelease: false
50
+ version_requirements: !ruby/object:Gem::Requirement
51
+ requirements:
52
+ - - ~>
53
+ - !ruby/object:Gem::Version
54
+ version: '5.0'
55
+ - !ruby/object:Gem::Dependency
56
+ name: mocha
57
+ requirement: !ruby/object:Gem::Requirement
58
+ requirements:
59
+ - - ~>
60
+ - !ruby/object:Gem::Version
61
+ version: '1.0'
62
+ type: :development
63
+ prerelease: false
64
+ version_requirements: !ruby/object:Gem::Requirement
65
+ requirements:
66
+ - - ~>
67
+ - !ruby/object:Gem::Version
68
+ version: '1.0'
69
+ - !ruby/object:Gem::Dependency
70
+ name: poseidon
71
+ requirement: !ruby/object:Gem::Requirement
72
+ requirements:
73
+ - - ~>
74
+ - !ruby/object:Gem::Version
75
+ version: 0.0.5
76
+ type: :runtime
77
+ prerelease: false
78
+ version_requirements: !ruby/object:Gem::Requirement
79
+ requirements:
80
+ - - ~>
81
+ - !ruby/object:Gem::Version
82
+ version: 0.0.5
83
+ - !ruby/object:Gem::Dependency
84
+ name: zookeeper
85
+ requirement: !ruby/object:Gem::Requirement
86
+ requirements:
87
+ - - ~>
88
+ - !ruby/object:Gem::Version
89
+ version: '1.4'
90
+ type: :runtime
91
+ prerelease: false
92
+ version_requirements: !ruby/object:Gem::Requirement
93
+ requirements:
94
+ - - ~>
95
+ - !ruby/object:Gem::Version
96
+ version: '1.4'
97
+ description: High-level consumer for Kafka. Implements the Zookeeper-backed consumer
98
+ implementation that offers offset management, load balancing and automatic failovers.
99
+ email:
100
+ - willem@vanbergen.org
101
+ executables:
102
+ - kazoo
103
+ extensions: []
104
+ extra_rdoc_files: []
105
+ files:
106
+ - .gitignore
107
+ - Gemfile
108
+ - LICENSE.txt
109
+ - README.md
110
+ - Rakefile
111
+ - bin/kazoo
112
+ - kafka-consumer.gemspec
113
+ - lib/kafka/consumer.rb
114
+ - lib/kafka/consumer/message.rb
115
+ - lib/kafka/consumer/partition_consumer.rb
116
+ - lib/kafka/consumer/version.rb
117
+ - lib/kazoo.rb
118
+ - lib/kazoo/broker.rb
119
+ - lib/kazoo/cli.rb
120
+ - lib/kazoo/cluster.rb
121
+ - lib/kazoo/consumergroup.rb
122
+ - lib/kazoo/partition.rb
123
+ - lib/kazoo/topic.rb
124
+ - lib/kazoo/version.rb
125
+ - test/broker_test.rb
126
+ - test/cluster_test.rb
127
+ - test/partition_distribution_test.rb
128
+ - test/partition_test.rb
129
+ - test/test_helper.rb
130
+ - test/topic_test.rb
131
+ homepage: https://github.com/wvanbergen/kafka-consumer
132
+ licenses:
133
+ - MIT
134
+ metadata: {}
135
+ post_install_message:
136
+ rdoc_options: []
137
+ require_paths:
138
+ - lib
139
+ required_ruby_version: !ruby/object:Gem::Requirement
140
+ requirements:
141
+ - - '>='
142
+ - !ruby/object:Gem::Version
143
+ version: '0'
144
+ required_rubygems_version: !ruby/object:Gem::Requirement
145
+ requirements:
146
+ - - '>='
147
+ - !ruby/object:Gem::Version
148
+ version: '0'
149
+ requirements: []
150
+ rubyforge_project:
151
+ rubygems_version: 2.0.14
152
+ signing_key:
153
+ specification_version: 4
154
+ summary: High-level consumer for Kafka
155
+ test_files:
156
+ - test/broker_test.rb
157
+ - test/cluster_test.rb
158
+ - test/partition_distribution_test.rb
159
+ - test/partition_test.rb
160
+ - test/test_helper.rb
161
+ - test/topic_test.rb