kafkat 0.0.10 → 0.0.11

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    MTY0MTdhYjQ3YjAzNWExNWJjMzMyYThmOTY5ZWQ5NmQyOTcyMGM4Yw==
+    NjhlMGY5MzIyYWFhZjdlZmE3YjlmM2M3MTc3MGU0MjQyYzQ2YjA5Zg==
   data.tar.gz: !binary |-
-    MTkzM2UzODgxNTIzMmJiYTAwNDQ2NmI5ZmVjYWZkYTdiYjNkZDk4Ng==
+    NDNlMjdjNDAxMjc4NTcyZDE2YzliNjI5OTBkNmJkMDA2OGM3Yjc0Zg==
 SHA512:
   metadata.gz: !binary |-
-    N2U0ZTMyMjQ4MmI3M2QzMDkyNTM4ZWFjOWJmMTNhNmIxNjlkM2NjMzBkYTli
-    YmY3N2Y1ZTk5OWIwZjZlZjZhMzNjOGE2MmE4NDg2ZTc4NzFjYTYzZmZlZWNi
-    YzFmNzFjZGU5N2Y5Y2UyMGY5NDhiNmEyZWU5NjdiNzViMjVlNjU=
+    MDdkOWI2ZjBhNjU2ZWM3NmI5YTMwNTFjYTA3MjQ5NjJiNDVjOTFiMGRkMDMw
+    ZmNkNThiNGJhYzYyNjIwYzczOTY0MjNhY2ZlYzA1MTQ2Y2Y4YTQxZDcwNTVm
+    NWU4YjA0NGYzMDQ1Mjc3ZjBiMGUxMTllM2U0N2QwMDdkODYzOGQ=
   data.tar.gz: !binary |-
-    YjRiMzllYmZjNDBiMmYxNjRiNzExMzlkZGM2ZWQwOTA0NmFiMGRkZTQzZjMz
-    ZmRlMGM5OTczMTcwYzI4MmJiYmE1ZGYwMDZlNzhhMTI3ZmQ2YThkN2E4OTdj
-    ZmUzN2QyZDMxNDk5ODE1ZWIwNTdjZmMyNWFkMTYzYWRjOWJmZWY=
+    MzU2ODcyZWFmY2VjMmFiM2NkMDMxNDRjNmQ4NGU5Y2JhOTE5MDczMzYzZGFj
+    MjhmOGE5MjVjNzViZWM3ZDJjMTkzZmZhZGYxMjI1Y2Q1NWEzMDU3ODc5MWUy
+    NzAyMTBlNmExN2E5M2U2MGM5YWZkY2FkNTkxZTAyMjY3NDE2ZTI=

data/.gitignore

@@ -4,3 +4,4 @@
 *.iml
 *.ipr
 
+reports

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--require spec_helper

data/.simplecov

@@ -0,0 +1,5 @@
+if ENV['COVERAGE']
+  SimpleCov.start do
+    coverage_dir("reports/coverage")
+  end
+end

data/.travis.yml

@@ -0,0 +1,10 @@
+language: ruby
+cache: bundler
+sudo: false
+script:
+  - bundle exec rake
+rvm:
+- 1.9.3
+- 2.0.0
+- 2.1.6
+- 2.2.2

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    kafkat (0.0.10)
+    kafkat (0.0.11)
       colored (~> 1.2)
       highline (~> 1.6, >= 1.6.21)
       retryable (~> 1.3, >= 1.3.5)
@@ -11,22 +11,59 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
+    activesupport (4.2.1)
+      i18n (~> 0.7)
+      json (~> 1.7, >= 1.7.7)
+      minitest (~> 5.1)
+      thread_safe (~> 0.3, >= 0.3.4)
+      tzinfo (~> 1.1)
     colored (1.2)
-    highline (1.6.21)
-    little-plugger (1.1.3)
-    logging (1.8.2)
-      little-plugger (>= 1.1.3)
-      multi_json (>= 1.8.4)
-    multi_json (1.10.1)
-    retryable (1.3.5)
-    trollop (2.0)
-    zk (1.9.4)
-      logging (~> 1.8.2)
+    diff-lcs (1.2.5)
+    docile (1.1.5)
+    factory_girl (4.5.0)
+      activesupport (>= 3.0.0)
+    highline (1.7.2)
+    i18n (0.7.0)
+    json (1.8.3)
+    minitest (5.7.0)
+    multi_json (1.11.1)
+    rake (10.4.2)
+    retryable (1.3.6)
+    rspec (3.2.0)
+      rspec-core (~> 3.2.0)
+      rspec-expectations (~> 3.2.0)
+      rspec-mocks (~> 3.2.0)
+    rspec-collection_matchers (1.1.2)
+      rspec-expectations (>= 2.99.0.beta1)
+    rspec-core (3.2.3)
+      rspec-support (~> 3.2.0)
+    rspec-expectations (3.2.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.2.0)
+    rspec-mocks (3.2.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.2.0)
+    rspec-support (3.2.2)
+    simplecov (0.9.0)
+      docile (~> 1.1.0)
+      multi_json
+      simplecov-html (~> 0.8.0)
+    simplecov-html (0.8.0)
+    thread_safe (0.3.5)
+    trollop (2.1.2)
+    tzinfo (1.2.2)
+      thread_safe (~> 0.1)
+    zk (1.9.6)
       zookeeper (~> 1.4.0)
-    zookeeper (1.4.9)
+    zookeeper (1.4.11)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
+  factory_girl (~> 4.5.0)
   kafkat!
+  rake
+  rspec (~> 3.2.0)
+  rspec-collection_matchers (~> 1.1.0)
+  simplecov

data/README.md

@@ -1,3 +1,5 @@
+[![Build Status](https://travis-ci.org/airbnb/kafkat.png?branch=master)](https://travis-ci.org/airbnb/kafkat)
+
 kafkat
 ======
 
@@ -52,6 +54,7 @@ Here's a list of supported commands:
   set-replication-factor [topic] [--newrf <n>] [--brokers id[,id]]    Set the replication factor of
   shutdown <broker id>                                                Gracefully remove leaderships from a broker (requires JMX).
   topics                                                              Print all topics.
+  drain <broker id> [--topic <t>] [--brokers <ids>]                   Reassign partitions from a specific broker to other brokers.
   
 ```
 

data/Rakefile

@@ -1,8 +1,25 @@
-require "bundler/gem_tasks"
+require 'bundler/gem_tasks'
 require 'rspec/core/rake_task'
-task :default => :spec
+require 'rake/clean'
 
-RSpec::Core::RakeTask.new(:spec) do |spec|
-  spec.pattern = 'spec/**/*_spec.rb'
-  spec.rspec_opts = ['-cfs --backtrace']
+
+CLEAN << 'reports'
+CLOBBER << FileList['kafkat*.gem']
+
+desc 'Run all specs'
+RSpec::Core::RakeTask.new :spec do |t|
+  t.pattern = ['spec/**/*_spec.rb']
+  t.rspec_opts = %w{--order random --format documentation --color --format html --out reports/specs.html}
+  t.rspec_opts << "--backtrace" if ENV['backtrace']
+  t.verbose = true
+end
+
+
+desc 'Run all specs and generate spec and coverage reports'
+task :coverage do
+  ENV['COVERAGE'] = 'true'
+  Rake::Task['spec'].invoke
 end
+
+task :test => :spec
+task :default => :coverage

data/_kafkat

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+
+# Run kafkat from source, specifically for testing stuff without doing a rake install.
+
+require 'rubygems'
+$LOAD_PATH << File.join(File.dirname(__FILE__), 'lib') << File.join(File.dirname(__FILE__), 'addon')
+
+require 'kafkat/version'
+spec = Gem::Specification.load(File.join(File.dirname(__FILE__), 'kafkat.gemspec'))
+
+# To avoid a warning about the version_requirements deprecation, we use this method inline.
+def version_required(gem_def)
+  return Gem::Dependency.instance_methods.map(&:to_sym).include?(:requirement) ? gem_def.requirement : gem_def.version_requirements
+end
+spec.dependencies.each do |dep|
+  gem dep.name, version_required(dep) if dep.type == :runtime
+end
+Gem.loaded_specs['kafkat'] = spec # Prevents RubyGem from loading files from installed kafkat gems
+
+require 'kafkat'
+Kafkat::CLI.run!

data/kafkat.gemspec

@@ -23,4 +23,10 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency 'highline', '~> 1.6', '>= 1.6.21'
   s.add_runtime_dependency 'retryable', '~> 1.3', '>= 1.3.5'
   s.add_runtime_dependency 'colored', '~> 1.2'
+
+  s.add_development_dependency 'rake'
+  s.add_development_dependency 'simplecov'
+  s.add_development_dependency 'rspec', '~> 3.2.0'
+  s.add_development_dependency 'rspec-collection_matchers', '~> 1.1.0'
+  s.add_development_dependency 'factory_girl', '~> 4.5.0'
 end

data/lib/kafkat/command.rb

@@ -14,6 +14,7 @@ module Kafkat
 
     class Base
       include Formatting
+      include CommandIO
 
       attr_reader :config
 

data/lib/kafkat/command/drain.rb

@@ -0,0 +1,94 @@
+module Kafkat
+  module Command
+    class Drain < Base
+      register_as 'drain'
+
+      usage 'drain <broker id> [--topic <t>] [--brokers <ids>]',
+            'Reassign partitions from a specific broker to destination brokers.'
+
+      # For each partition (of speicified topic) on the source broker, the command is to
+      # assign the partition to one of the destination brokers that does not already have
+      # this partition, along with existing brokers to achieve minimal movement of data.
+      # To help distribute data evenly, if there are more than one destination brokers
+      # meet the requirement, the command will always choose the brokers with the lowest
+      # number of partitions of the involving topic.
+      #
+      # In order to find out the broker with lowest number of partitions, the command maintain
+      # a hash table with broker id as key and number of partitions as value. The hash table
+      # will be updated along with assignment.
+      def run
+        source_broker = ARGV[0] && ARGV.shift.to_i
+        if source_broker.nil?
+          puts "You must specify a broker ID."
+          exit 1
+        end
+
+        opts = Trollop.options do
+          opt :brokers, "destination broker IDs", type: :string
+          opt :topic,   "topic name to reassign", type: :string
+        end
+
+        topic_name = opts[:topic]
+        topics = topic_name && zookeeper.get_topics([topic_name])
+        topics ||= zookeeper.get_topics
+
+        destination_brokers = opts[:brokers] && opts[:brokers].split(',').map(&:to_i)
+        destination_brokers ||= zookeeper.get_brokers.values.map(&:id)
+        destination_brokers.delete(source_broker)
+
+        active_brokers = zookeeper.get_brokers.values.map(&:id)
+
+        unless (inactive_brokers = destination_brokers - active_brokers).empty?
+          print "ERROR: Broker #{inactive_brokers} are not currently active.\n"
+          exit 1
+        end
+
+        assignments =
+          generate_assignments(source_broker, topics, destination_brokers)
+        prompt_and_execute_assignments(assignments)
+      end
+
+      def generate_assignments(source_broker, topics, destination_brokers)
+
+        assignments = []
+        topics.each do |_, t|
+          partitions_by_broker = build_partitions_by_broker(t, destination_brokers)
+
+          t.partitions.each do |p|
+            if p.replicas.include? source_broker
+              replicas = p.replicas - [source_broker]
+              potential_broker_ids = destination_brokers - replicas
+              if potential_broker_ids.empty?
+                print "ERROR: Not enough destination brokers to reassign topic \"#{t.name}\".\n"
+                exit 1
+              end
+
+              num_partitions_on_potential_broker =
+                partitions_by_broker.select { |id, _| potential_broker_ids.include? id }
+              assigned_broker_id = num_partitions_on_potential_broker.min_by{ |id, num| num }[0]
+              replicas << assigned_broker_id
+              partitions_by_broker[assigned_broker_id] += 1
+
+              assignments << Assignment.new(t.name, p.id, replicas)
+            end
+          end
+        end
+
+        assignments
+      end
+
+      # Build a hash map from broker id to number of partitions on it to facilitate
+      # finding the broker with lowest number of partitions to help balance brokers.
+      def build_partitions_by_broker(topic, destination_brokers)
+        partitions_by_broker = Hash.new(0)
+        destination_brokers.each { |id| partitions_by_broker[id] = 0 }
+        topic.partitions.each do |p|
+          p.replicas.each do |r|
+            partitions_by_broker[r] += 1
+          end
+        end
+        partitions_by_broker
+      end
+    end
+  end
+end

data/lib/kafkat/command/reassign.rb

@@ -68,21 +68,7 @@ module Kafkat
 
         # ****************
 
-        print "This operation executes the following assignments:\n\n"
-        print_assignment_header
-        assignments.each { |a| print_assignment(a) }
-        print "\n"
-
-        return unless agree("Proceed (y/n)?")
-
-        result = nil
-        begin
-          print "\nBeginning.\n"
-          result = admin.reassign!(assignments)
-          print "Started.\n"
-        rescue Interface::Admin::ExecutionFailedError
-          print result
-        end
+        prompt_and_execute_assignments(assignments)
       end
     end
   end

data/lib/kafkat/utility.rb

@@ -1 +1,2 @@
 require 'kafkat/utility/formatting'
+require 'kafkat/utility/command_io'

data/lib/kafkat/utility/command_io.rb

@@ -0,0 +1,21 @@
+module Kafkat
+  module CommandIO
+    def prompt_and_execute_assignments(assignments)
+      print "This operation executes the following assignments:\n\n"
+      print_assignment_header
+      assignments.each { |a| print_assignment(a) }
+      print "\n"
+
+      return unless agree("Proceed (y/n)?")
+
+      result = nil
+      begin
+        print "\nBeginning.\n"
+        result = admin.reassign!(assignments)
+        print "Started.\n"
+      rescue Interface::Admin::ExecutionFailedError
+        print result
+      end
+    end
+  end
+end

data/lib/kafkat/version.rb

@@ -1,3 +1,3 @@
 module Kafkat
-  VERSION = '0.0.10'
+  VERSION = '0.0.11'
 end

data/spec/factories/topic.rb

@@ -0,0 +1,39 @@
+module Kafkat
+  FactoryGirl.define do
+    factory :topic, class:Topic do
+      name "topic_name"
+
+      factory :topic_with_one_empty_broker do
+        partitions {[Partition.new(name, 0, [0], 0, 0),
+                     Partition.new(name, 1, [1], 1, 1),
+                     Partition.new(name, 2, [0], 0, 0),
+                     Partition.new(name, 3, [0], 0, 0),
+                     Partition.new(name, 4, [1], 1, 1)]}
+      end
+
+      factory :topic_rep_factor_one do
+        partitions {[Partition.new(name, 0, [0], 0, 0),
+                     Partition.new(name, 1, [1], 1, 1),
+                     Partition.new(name, 2, [2], 2, 2),
+                     Partition.new(name, 3, [0], 0, 0),
+                     Partition.new(name, 4, [1], 1, 1)]}
+      end
+
+      factory :topic_rep_factor_two do
+        partitions {[Partition.new(name, 0, [0, 1], 0, 0),
+                     Partition.new(name, 1, [0, 2], 2, 2),
+                     Partition.new(name, 2, [1, 2], 1, 1),
+                     Partition.new(name, 3, [0, 1], 0, 0),
+                     Partition.new(name, 4, [0, 2], 2, 2)]}
+      end
+
+      factory :topic_rep_factor_three do
+        partitions {[Partition.new(name, 0, [0, 1, 2], 0, 0),
+                     Partition.new(name, 1, [0, 1, 2], 1, 1),
+                     Partition.new(name, 2, [0, 1, 2], 2, 2),
+                     Partition.new(name, 3, [0, 1, 2], 0, 0),
+                     Partition.new(name, 4, [0, 1, 2], 1, 1)]}
+      end
+    end
+  end
+end

data/spec/lib/kafkat/command/drain_spec.rb

@@ -0,0 +1,59 @@
+require 'spec_helper'
+
+module Kafkat
+  RSpec.describe Command::Drain do
+    let(:drain) { Command::Drain.new({}) }
+    let(:broker_id) { 0 }
+    let(:destination_broker_ids) { [1, 2] }
+
+    context 'three nodes with replication factor 1' do
+      let(:topic_rep_factor_one) { FactoryGirl.build(:topic_rep_factor_one) }
+
+      it 'should put replicas to broker with lowest number of replicas' do
+        assignments = drain.generate_assignments(broker_id,
+                                                 {"topic_name" => topic_rep_factor_one},
+                                                 destination_broker_ids)
+        expect(assignments).to have_exactly(2).Partition
+        expect(assignments[0].replicas).to eq([2])
+        expect(assignments[1].replicas).to eq([1])
+      end
+    end
+
+    context 'three nodes with replication factor 2' do
+      let(:topic_rep_factor_two) { FactoryGirl.build(:topic_rep_factor_two) }
+      it 'should put replicas to broker with lowest number of replicas' do
+        assignments = drain.generate_assignments(broker_id,
+                                                 {"topic_name" => topic_rep_factor_two},
+                                                 destination_broker_ids)
+        expect(assignments).to have_exactly(4).Partition
+        expect(assignments[0].replicas).to eq([1, 2])
+        expect(assignments[1].replicas).to eq([2, 1])
+        expect(assignments[2].replicas).to eq([1, 2])
+        expect(assignments[3].replicas).to eq([2, 1])
+      end
+    end
+
+    context 'not enough brokers to keep all replicas' do
+      let(:topic_rep_factor_three) { FactoryGirl.build(:topic_rep_factor_three) }
+
+      it 'should raise SystemExit' do
+        expect do
+          drain.generate_assignments(broker_id,
+                                     {"topic_name" => topic_rep_factor_three},
+                                     destination_broker_ids)
+        end.to raise_error(SystemExit)
+      end
+    end
+
+    context 'one destination broker is empty' do
+      let(:topic_with_one_empty_broker) { FactoryGirl.build(:topic_with_one_empty_broker) }
+      it 'should not raise exception' do
+        expect do
+          drain.generate_assignments(broker_id,
+                                     {"topic_name" => topic_with_one_empty_broker},
+                                     destination_broker_ids)
+        end.not_to raise_error
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,102 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# The generated `.rspec` file contains `--require spec_helper` which will cause
+# this file to always be loaded, without a need to explicitly require it in any
+# files.
+#
+# Given that it is always loaded, you are encouraged to keep this file as
+# light-weight as possible. Requiring heavyweight dependencies from this file
+# will add to the boot time of your test suite on EVERY test run, even for an
+# individual file that may not need all of that loaded. Instead, consider making
+# a separate helper file that requires the additional dependencies and performs
+# the additional setup, and require it from the spec files that actually need
+# it.
+#
+# The `.rspec` file also contains a few flags that are not defaults but that
+# users commonly want.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+if ENV['COVERAGE']
+    require 'simplecov'
+end
+require 'kafkat'
+require 'factory_girl'
+require 'rspec/collection_matchers'
+
+require_relative '../spec/factories/topic'
+
+RSpec.configure do |config|
+  # rspec-expectations config goes here. You can use an alternate
+  # assertion/expectation library such as wrong or the stdlib/minitest
+  # assertions if you prefer.
+  config.expect_with :rspec do |expectations|
+    # This option will default to `true` in RSpec 4. It makes the `description`
+    # and `failure_message` of custom matchers include text for helper methods
+    # defined using `chain`, e.g.:
+    #     be_bigger_than(2).and_smaller_than(4).description
+    #     # => "be bigger than 2 and smaller than 4"
+    # ...rather than:
+    #     # => "be bigger than 2"
+    expectations.include_chain_clauses_in_custom_matcher_descriptions = true
+  end
+
+  # rspec-mocks config goes here. You can use an alternate test double
+  # library (such as bogus or mocha) by changing the `mock_with` option here.
+  config.mock_with :rspec do |mocks|
+    # Prevents you from mocking or stubbing a method that does not exist on
+    # a real object. This is generally recommended, and will default to
+    # `true` in RSpec 4.
+    mocks.verify_partial_doubles = true
+  end
+
+  config.include FactoryGirl::Syntax::Methods
+
+# The settings below are suggested to provide a good initial experience
+# with RSpec, but feel free to customize to your heart's content.
+=begin
+  # These two settings work together to allow you to limit a spec run
+  # to individual examples or groups you care about by tagging them with
+  # `:focus` metadata. When nothing is tagged with `:focus`, all examples
+  # get run.
+  config.filter_run :focus
+  config.run_all_when_everything_filtered = true
+
+  # Limits the available syntax to the non-monkey patched syntax that is
+  # recommended. For more details, see:
+  #   - http://myronmars.to/n/dev-blog/2012/06/rspecs-new-expectation-syntax
+  #   - http://teaisaweso.me/blog/2013/05/27/rspecs-new-message-expectation-syntax/
+  #   - http://myronmars.to/n/dev-blog/2014/05/notable-changes-in-rspec-3#new__config_option_to_disable_rspeccore_monkey_patching
+  config.disable_monkey_patching!
+
+  # This setting enables warnings. It's recommended, but in some cases may
+  # be too noisy due to issues in dependencies.
+  config.warnings = true
+
+  # Many RSpec users commonly either run the entire suite or an individual
+  # file, and it's useful to allow more verbose output when running an
+  # individual spec file.
+  if config.files_to_run.one?
+    # Use the documentation formatter for detailed output,
+    # unless a formatter has already been configured
+    # (e.g. via a command-line flag).
+    config.default_formatter = 'doc'
+  end
+
+  # Print the 10 slowest examples and example groups at the
+  # end of the spec run, to help surface which specs are running
+  # particularly slow.
+  config.profile_examples = 10
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = :random
+
+  # Seed global randomization in this process using the `--seed` CLI option.
+  # Setting this allows you to use `--seed` to deterministically reproduce
+  # test failures related to randomization by passing the same `--seed` value
+  # as the one that triggered the failure.
+  Kernel.srand config.seed
+=end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kafkat
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.11
 platform: ruby
 authors:
 - Nelson Gauthier
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-02-01 00:00:00.000000000 Z
+date: 2016-01-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zk
@@ -98,6 +98,76 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+- !ruby/object:Gem::Dependency
+  name: rspec-collection_matchers
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: factory_girl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.5.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.5.0
 description: Simplified command-line administration for Kafka brokers
 email:
 - nelson@airbnb.com
@@ -107,11 +177,15 @@ extensions: []
 extra_rdoc_files: []
 files:
 - .gitignore
+- .rspec
+- .simplecov
+- .travis.yml
 - Gemfile
 - Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
+- _kafkat
 - bin/kafkat
 - kafkat.gemspec
 - lib/kafkat.rb
@@ -125,6 +199,7 @@ files:
 - lib/kafkat/command/brokers.rb
 - lib/kafkat/command/clean-indexes.rb
 - lib/kafkat/command/controller.rb
+- lib/kafkat/command/drain.rb
 - lib/kafkat/command/elect-leaders.rb
 - lib/kafkat/command/partitions.rb
 - lib/kafkat/command/reassign.rb
@@ -138,8 +213,12 @@ files:
 - lib/kafkat/interface/kafka_logs.rb
 - lib/kafkat/interface/zookeeper.rb
 - lib/kafkat/utility.rb
+- lib/kafkat/utility/command_io.rb
 - lib/kafkat/utility/formatting.rb
 - lib/kafkat/version.rb
+- spec/factories/topic.rb
+- spec/lib/kafkat/command/drain_spec.rb
+- spec/spec_helper.rb
 homepage: https://github.com/airbnb/kafkat
 licenses:
 - Apache-v2
@@ -164,4 +243,7 @@ rubygems_version: 2.4.1
 signing_key: 
 specification_version: 4
 summary: Simplified command-line administration for Kafka brokers
-test_files: []
+test_files:
+- spec/factories/topic.rb
+- spec/lib/kafkat/command/drain_spec.rb
+- spec/spec_helper.rb