mycmd 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bf1b833907aa455e71b88d4fd93f821274103a13
-  data.tar.gz: 1dc6c7315cd5d5f157c5bb2b1bfaf5ab6106d9d9
+  metadata.gz: a8c336e5ee888d89abcf88b847b4b014992d5d53
+  data.tar.gz: 0314ddffcabc62fa4de866899f4497be744795b3
 SHA512:
-  metadata.gz: a7eb2b0c883a5207ab4466b251c5e4fd01026126e98a4865ef6ff2069be2ac8b8ea7dd87c8ce4cd4a9c74b79a40e8e368e95ebd275e25493419b4465ddaaf7d4
-  data.tar.gz: 1fdda7ab25c75dcb20b47498328475dc9b1eb0a8ff3e778fa266baf28915cb8409aaab6dd3ee9f276a52b256864e02558ae17f97b82f547e0b5165d7440b6a84
+  metadata.gz: b67247b0b25dcf6b14fab62bf02da1dabdd26e646edf1f42eb812e28605d1740230e47b84cdca6033367d4ca0ee0d9b3783f45ea5d730d013e513b875596c73a
+  data.tar.gz: 33fad2267c9e8a4ab357e397433b5b3fc64ba17397d6df5ff9be3b0e0f28a7fbdf277d836858278431adccd87ce3c52cea3b9f238085a0ed92f5ac7c26bb1008

data/lib/mycmd/cli.rb

@@ -11,7 +11,7 @@ module Mycmd
 
     desc "console", "console will start sql shell."
     def console
-      raise "mysql not found" unless system("which mysql > /dev/null")
+      raise "mysql not found" unless Kernel.system("which mysql > /dev/null")
       conf = Configuration.new
       cmd = conf.to_hash.inject(["mysql"]) do |c,(k,v)|
         case k
@@ -23,7 +23,7 @@ module Mycmd
         end
       end
 
-      system(cmd.join(" "))
+      Kernel.system(cmd.join(" "))
     end
 
     desc 'query "[SQL]"', "query will execute sql."

data/lib/mycmd/clis/config_commands.rb

@@ -14,14 +14,14 @@ module Mycmd
     def cat
       conf = Configuration.config_find
       raise "config not found" if conf.nil?
-      open(conf, "r").each {|line| puts line}
+      File.open(conf, "r").each {|line| puts line}
     end
 
     desc "edit", "edit will edit configuration"
     def edit
       conf = Configuration.config_find
       raise "config not found" if conf.nil?
-      system("#{ENV['EDITOR']} #{conf}")
+      Kernel.system("#{ENV['EDITOR']} #{conf}")
     end
   end
 end

data/lib/mycmd/clis/settings_commands.rb

@@ -4,11 +4,75 @@ module Mycmd
   class SettingsCommands < Thor
     namespace :settings
 
+    GLOBAL_BUFFERS = [
+      :innodb_buffer_pool_size,
+      :innodb_log_buffer_size,
+      :innodb_additional_mem_pool_size,
+      :key_buffer_size,
+      :query_cache_size
+    ].freeze
+
+    THREAD_BUFFERS = [
+      :sort_buffer_size,
+      :read_buffer_size,
+      :read_rnd_buffer_size,
+      :join_buffer_size,
+      :tmp_table_size,
+      :max_heap_table_size,
+      :net_buffer_length,
+      :max_allowed_packet,
+      :thread_stack
+    ].freeze
+
     desc "search innodb_buffer_pool_size", "search will print settings"
     def search(keyword)
       client = Configuration.connect
       printer = Printer.new(client.query("SHOW GLOBAL VARIABLES LIKE \"%#{keyword}%\""))
       printer.print
     end
+
+    desc "memories", "memories will print memory settings"
+    def memories
+      variables = Configuration.get_variables
+      global, gtotal = create_result_variables(GLOBAL_BUFFERS, variables)
+      thread, ttotal = create_result_variables(THREAD_BUFFERS, variables)
+      expected = expected_memory_used(gtotal, variables)
+      Printer.print_title("GLOBAL BUFFERS")
+      Printer.new(global).print
+      Printer.print_title("THREAD BUFFERS", true)
+      Printer.new(thread).print
+      Printer.print_title("EXPECTED MEMORY USED", true)
+      Printer.new(expected).print
+    end
+
+    private
+    def create_result_variables(keys, variables)
+      rows = []
+      total = keys.inject(0) do |t,key|
+        bytes = variables[key].to_i
+        rows << [key.to_s ,"#{bytes} bytes (#{bytes.to_f/1024/1024} MB)"]
+        t + bytes
+      end
+      return rows, total
+    end
+
+    def expected_memory_used(global, variables)
+      max_connections = variables[:max_connections].to_i
+      threads = expected_threads(variables, max_connections)
+      total = global + threads
+      [
+        ["max_connections", "#{max_connections} connections"],
+        ["expected memory use of global", "#{global} bytes (#{global.to_f/1024/1024} MB)"],
+        ["expected memory use of threads", "#{threads} bytes (#{threads.to_f/1024/1024} MB)"],
+        ["expected total", "#{total} bytes (#{total.to_f/1024/1024} MB)"]
+      ]
+    end
+
+    def expected_threads(variables, max_connections)
+      base = variables[:net_buffer_length].to_i + variables[:thread_stack].to_i
+      tmp = variables[:tmp_table_size].to_i * 0.1
+      buffer = THREAD_BUFFERS[0..3].inject(0){|b,k| b + (variables[k].to_i * 0.5)}
+      ((base + tmp + buffer) * max_connections)
+    end
   end
 end

data/lib/mycmd/configuration.rb

@@ -67,6 +67,16 @@ module Mycmd
         conf.connect
       end
 
+      def get_variables
+        sql = "SELECT * FROM INFORMATION_SCHEMA.GLOBAL_VARIABLES"
+        client = self.connect
+        variables = {}
+        client.query(sql).each do |row|
+          variables.store(row["VARIABLE_NAME"].downcase.to_sym, row["VARIABLE_VALUE"])
+        end
+        variables
+      end
+
       def config_find(path = File.expand_path("."))
         file = File.join(path, CONFIG_FILE)
         if File.exists?(file)

data/lib/mycmd/printer.rb

@@ -2,25 +2,36 @@
 
 module Mycmd
   class Printer
+    BORDER = "*" * 30
+    attr_accessor :result, :header, :width
+    
     def initialize(result, header=false)
-      @result = result
-      @header = header
+      @result = result.is_a?(Mysql2::Result) ? result_to_array(result) : result
+      @header = result.is_a?(Mysql2::Result) ? result.fields : header
     end
 
     def print
       if @result.respond_to? :each
         set_width
-        print_line(@result.fields) if @header
-        @result.each(as: :array) do |row|
+        print_line(@header) if @header
+        @result.each do |row|
           print_line(row)
         end
       end
     end
 
     private
+    def result_to_array(result)
+      result_array = []
+      result.each(as: :array) do |row|
+        result_array << row
+      end
+      result_array
+    end
+
     def set_width
-      @width = @result.fields.map{|f| f.size}
-      @result.each(as: :array) do |row|
+      @width = @header ? @header.map{|f| f.size} : Array.new(@result.first.size){0}
+      @result.each do |row|
         row.each_with_index do |v,i|
           @width[i] = v.to_s.size if @width[i] < v.to_s.size
         end
@@ -30,5 +41,12 @@ module Mycmd
     def print_line(line_array)
       puts line_array.map.with_index {|f,i| f.to_s.ljust(@width[i], " ")}.join("\t")
     end
+
+    class << self
+      def print_title(title, empty_line=false)
+        puts if empty_line
+        puts "#{BORDER}\n#{title}\n#{BORDER}"
+      end
+    end
   end
 end

data/lib/mycmd/version.rb

@@ -1,3 +1,3 @@
 module Mycmd
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/mycmd/cli_spec.rb

@@ -1,4 +1,89 @@
 require "spec_helper"
 
-describe Mycmd::CLI do 
+describe Mycmd::CLI do
+  let(:conn_mock) {double("connection mock").as_null_object}
+  let(:printer_mock) {double("printer mock").as_null_object}
+
+  describe "#console" do
+    let(:args) {["console"]}
+
+    before do
+      Mycmd::Configuration.stub(:config_find).and_return(nil)
+    end
+
+    it "should call Kernel.system" do
+      conf = Mycmd::Configuration.new
+      conf.password = "secret"
+      conf.database = "test"
+      Mycmd::Configuration.should_receive(:new).and_return(conf)
+      Kernel.should_receive(:system).exactly(2).and_return(true)
+      expect {
+        Mycmd::CLI.start(args)
+      }.not_to raise_error
+    end
+
+    it "should generate exception if mysql command not found" do
+      Kernel.should_receive(:system).and_return(false)
+      expect {
+        Mycmd::CLI.start(args)
+      }.to raise_error
+    end
+  end
+
+  describe "#query" do
+    before do
+      Mycmd::Configuration.stub(:connect).and_return(conn_mock)
+      Mycmd::Printer.stub(:new).and_return(printer_mock)
+    end
+
+    after do
+      expect {
+        Mycmd::CLI.start(["query", "some sql"])
+      }.not_to raise_error
+    end
+
+    it "should call Configuration.connect" do
+      Mycmd::Configuration.should_receive(:connect).and_return(conn_mock)
+    end
+
+    it "should create Printer object" do
+      Mycmd::Printer.should_receive(:new).with(conn_mock, true).and_return(printer_mock)
+    end
+
+    it "should call Printer#print" do
+      printer_mock.should_receive(:print)
+    end
+  end
+
+  describe "#tasks" do
+    let(:config_mock) {double("configuration mock").as_null_object}
+
+    before do
+      config_mock.stub(:connect).and_return(conn_mock)
+      Mycmd::Configuration.stub(:new).and_return(config_mock)
+      Mycmd::Printer.stub(:new).and_return(printer_mock)
+    end
+
+    after do
+      expect {
+        Mycmd::CLI.start(["tasks", "some task"])
+      }.not_to raise_error
+    end
+
+    it "should create Configuration object" do
+      Mycmd::Configuration.should_receive(:new).and_return(config_mock)
+    end
+
+    it "should call Configuration#connect" do
+      config_mock.should_receive(:connect).and_return(conn_mock)
+    end
+
+    it "should create Printer object" do
+      Mycmd::Printer.should_receive(:new).with(conn_mock, true).and_return(printer_mock)
+    end
+
+    it "should call Printer#print" do
+      printer_mock.should_receive(:print)
+    end
+  end
 end

data/spec/mycmd/clis/config_commands_spec.rb

@@ -0,0 +1,75 @@
+require "spec_helper"
+
+describe Mycmd::ConfigCommands do
+  describe "#which" do
+    let(:args) {["which"]}
+
+    it "should call Configuration.config_find" do
+      Mycmd::Configuration.should_receive(:config_find).and_return(nil)
+      expect(
+        capture(:stdout){
+          Mycmd::ConfigCommands.start(args)
+        }
+      ).not_to be_nil
+    end
+
+    it "should print not found message if file not found" do
+      Mycmd::Configuration.stub(:config_find).and_return(nil)
+      expect(
+        capture(:stdout){
+          Mycmd::ConfigCommands.start(args)
+        }
+      ).to eq("config not found\n")
+    end
+
+    it "should print config path" do
+      config = "/Users/i2bskn/.mycmd.yml"
+      Mycmd::Configuration.stub(:config_find).and_return(config)
+      expect(
+        capture(:stdout){
+          Mycmd::ConfigCommands.start(args)
+        }
+      ).to eq("#{config}\n")
+    end
+  end
+
+  describe "#edit" do
+    let(:args) {["edit"]}
+
+    it "should execute edit command" do
+      Mycmd::Configuration.stub(:config_find).and_return(".mycmd.yml")
+      Kernel.should_receive(:system).with("#{ENV['EDITOR']} .mycmd.yml")
+      Mycmd::ConfigCommands.start(args)
+    end
+
+    it "should generate exception if file not found" do
+      Mycmd::Configuration.stub(:config_find).and_return(nil)
+      expect {
+        Mycmd::ConfigCommands.start(args)
+      }.to raise_error(RuntimeError)
+    end
+
+    it "should call Configuration.config_find" do
+      Mycmd::Configuration.should_receive(:config_find).and_return(".mycmd.yml")
+      Kernel.stub(:system)
+      Mycmd::ConfigCommands.start(args)
+    end
+  end
+
+  describe "#cat" do
+    let(:args) {["cat"]}
+
+    it "should call Configuration.config_find" do
+      Mycmd::Configuration.should_receive(:config_find).and_return(".mycmd.yml")
+      File.stub(:open).and_return([])
+      Mycmd::ConfigCommands.start(args)
+    end
+
+    it "should generate exception if file not found" do
+      Mycmd::Configuration.stub(:config_find).and_return(nil)
+      expect {
+        Mycmd::ConfigCommands.start(args)
+      }.to raise_error(RuntimeError)
+    end
+  end
+end

data/spec/mycmd/clis/settings_commands_spec.rb

@@ -0,0 +1,47 @@
+require "spec_helper"
+
+describe Mycmd::SettingsCommands do
+  describe "#search" do
+    let(:args) {["search", "param_name"]}
+    let(:conn_mock) do
+      mock = double("connection mock")
+      mock.stub(:query).and_return(create_result)
+      mock
+    end
+
+    before {Mycmd::Configuration.stub(:connect).and_return(conn_mock)}
+
+    after do
+      expect(
+        capture(:stdout) {
+          Mycmd::SettingsCommands.start(args)
+        }
+      ).not_to be_nil
+    end
+
+    it "should call Configuration.connect" do
+      Mycmd::Configuration.should_receive(:connect).and_return(conn_mock)
+    end
+
+    it "should create Printer object" do
+      Mycmd::Printer.should_receive(:new).and_return(double.as_null_object)
+    end
+
+    it "should call Printer#print" do
+      Mycmd::Printer.any_instance.should_receive(:print)
+    end
+  end
+
+  describe "#memories" do
+    let(:args){["memories"]}
+
+    it "returns expected threads" do
+      Mycmd::Configuration.stub(:get_variables).and_return(create_variables)
+      expect(
+        capture(:stdout) {
+          Mycmd::SettingsCommands.start(args)
+        }
+      ).not_to be_nil
+    end
+  end
+end

data/spec/mycmd/configuration_spec.rb

@@ -53,21 +53,21 @@ describe Mycmd::Configuration do
   end
 
   describe "#reset" do
+    after do
+      expect {
+        Mycmd::Configuration.new
+      }.not_to raise_error
+    end
+
     it "should load yaml file" do
       Mycmd::Configuration.should_receive(:config_find).and_return(conf_file)
       Mycmd::Configuration.any_instance.should_receive(:merge)
       YAML.should_receive(:load_file).with(conf_file)
-      expect {
-        Mycmd::Configuration.new
-      }.not_to raise_error
     end
 
     it "should call #default if config not found" do
       Mycmd::Configuration.should_receive(:config_find).and_return(nil)
       Mycmd::Configuration.any_instance.should_receive(:default)
-      expect {
-        Mycmd::Configuration.new
-      }.not_to raise_error
     end
   end
 
@@ -87,7 +87,7 @@ describe Mycmd::Configuration do
     end
   end
 
-  describe ".connect" do
+  describe ".#connect" do
     it "should call #connect" do
       mock = double("configuration mock")
       mock.should_receive(:connect)
@@ -98,7 +98,34 @@ describe Mycmd::Configuration do
     end
   end
 
-  describe ".config_find" do
+  describe ".#get_variables" do
+    let(:conn_mock) {double("connection mock")}
+
+    before do
+      conn_mock.stub(:query).and_return([{"VARIABLE_NAME" => "innodb_buffer_pool_size", "VARIABLE_VALUE" => "268435456"}])
+      Mycmd::Configuration.stub(:connect).and_return(conn_mock)
+    end
+
+    after do
+      expect {
+        Mycmd::Configuration.get_variables
+      }.not_to raise_error
+    end
+
+    it "should call .#connect" do
+      Mycmd::Configuration.should_receive(:connect).and_return(conn_mock)
+    end
+
+    it "should call Mysql2::Client#query" do
+      conn_mock.should_receive(:query).and_return([])
+    end
+
+    it "should return variables" do
+      expect(Mycmd::Configuration.get_variables).to eq({innodb_buffer_pool_size: "268435456"})
+    end
+  end
+
+  describe ".#config_find" do
     let(:path) {Mycmd::Configuration.config_find(home_dir)}
 
     it "returns config path" do

data/spec/mycmd/printer_spec.rb

@@ -0,0 +1,89 @@
+require "spec_helper"
+
+describe Mycmd::Printer do
+  let(:result) {create_result}
+  let(:printer) {Mycmd::Printer.new(result)}
+
+  describe "#initialize" do
+    it "should set specified result" do
+      expect(printer.result).to eq(result)
+    end
+
+    it "should set default header flug" do
+      expect(printer.header).to be_false
+    end
+
+    it "should set specified header flug" do
+      printer = Mycmd::Printer.new(result, true)
+      expect(printer.header).to be_true
+    end
+
+    it "should generate exception if not specified arguments" do
+      expect {
+        Mycmd::Printer.new
+      }.to raise_error
+    end
+  end
+
+  describe "#print" do
+    after do
+      expect {
+        printer.print
+      }.not_to raise_error
+    end
+
+    it "should call #set_width" do
+      printer.send(:set_width)
+      Mycmd::Printer.any_instance.should_receive(:set_width)
+      Mycmd::Printer.any_instance.stub(:print_line).and_return(nil)
+    end
+
+    it "should call #print_line" do
+      Mycmd::Printer.any_instance.should_receive(:print_line).exactly(3)
+    end
+
+    it "should not call #print_line if each method not found" do
+      Mycmd::Printer.any_instance.should_not_receive(:print_line)
+      result.should_receive(:respond_to?).and_return(false)
+    end
+  end
+
+  describe "#result_to_array" do
+    it "should convert to array from result" do
+      expect(printer.send(:result_to_array, create_result)).to eq([
+        ["first", "one"],
+        ["second", "two"],
+        ["third", "three"]
+      ])
+    end
+  end
+
+  describe "#set_width" do
+    it "should set max length" do
+      printer.send(:set_width)
+      expect(printer.width).to eq([6,5])
+    end
+
+    it "should hoge" do
+      printer.header = result.fields
+      printer.send(:set_width)
+      expect(printer.width).to eq([6,6])
+    end
+  end
+
+  describe "#print_line" do
+    it "should print line" do
+      expect(capture(:stdout){printer.print}).not_to be_nil
+    end
+  end
+
+  describe "#print_title" do
+    it "should print title" do
+      expect(capture(:stdout){Mycmd::Printer.print_title("title")}).not_to be_nil
+    end
+
+    it "should print empty line" do
+      expect(capture(:stdout){Mycmd::Printer.print_title("title", true)}).to match(/^\n/)
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -14,6 +14,10 @@ end
 
 require "mycmd"
 
+support_files = File.join(File.expand_path("..", __FILE__), "support/*.rb")
+Dir[support_files].each {|f| require f}
+
 RSpec.configure do |config|
   config.order = "random"
+  config.include(SpecUtils)
 end

data/spec/support/spec_utils.rb

@@ -0,0 +1,61 @@
+require "stringio"
+
+module SpecUtils
+  def capture(stream)
+    begin
+      stream = stream.to_s
+      eval "$#{stream} = StringIO.new"
+      yield
+      result = eval("$#{stream}").string
+    ensure
+      eval("$#{stream} = #{stream.upcase}")
+    end
+    result
+  end
+
+  def create_variables
+    {
+      innodb_buffer_pool_size: "268435456",
+      innodb_log_buffer_size: "8388608",
+      innodb_additional_mem_pool_size: "16777216",
+      key_buffer_size: "16777216",
+      query_cache_size: "0",
+      sort_buffer_size: "524288",
+      read_buffer_size: "262144",
+      read_rnd_buffer_size: "524288",
+      join_buffer_size: "262144",
+      tmp_table_size: "16777216",
+      max_heap_table_size: "16777216",
+      net_buffer_length: "16384",
+      max_allowed_packet: "1048576",
+      thread_stack: "262144"
+    }
+  end
+
+  def create_result
+    Result.new
+  end
+
+  class Result
+    attr_reader :fields
+
+    def initialize
+      @data = [
+        ["first", "one"],
+        ["second", "two"],
+        ["third", "three"]
+      ]
+      @fields = ["order", "number"]
+    end
+
+    def first
+      @data.first
+    end
+
+    def each(params={})
+      @data.each do |d|
+        yield(d)
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mycmd
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - i2bskn
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-21 00:00:00.000000000 Z
+date: 2013-08-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -105,8 +105,12 @@ files:
 - lib/mycmd/version.rb
 - mycmd.gemspec
 - spec/mycmd/cli_spec.rb
+- spec/mycmd/clis/config_commands_spec.rb
+- spec/mycmd/clis/settings_commands_spec.rb
 - spec/mycmd/configuration_spec.rb
+- spec/mycmd/printer_spec.rb
 - spec/spec_helper.rb
+- spec/support/spec_utils.rb
 homepage: https://github.com/i2bskn/mycmd
 licenses:
 - MIT
@@ -133,5 +137,9 @@ specification_version: 4
 summary: MySQL command line tool
 test_files:
 - spec/mycmd/cli_spec.rb
+- spec/mycmd/clis/config_commands_spec.rb
+- spec/mycmd/clis/settings_commands_spec.rb
 - spec/mycmd/configuration_spec.rb
+- spec/mycmd/printer_spec.rb
 - spec/spec_helper.rb
+- spec/support/spec_utils.rb