kontena-cli 1.0.6 → 1.1.0.pre1

data/lib/kontena/stacks_cache.rb

@@ -68,7 +68,7 @@ module Kontena
       end
 
       def dputs(msg)
-        ENV["DEBUG"] && puts(msg)
+        ENV["DEBUG"] && STDERR.puts(msg)
       end
 
       def cache(stack, version = nil)

data/lib/kontena_cli.rb

@@ -7,16 +7,16 @@ module Kontena
   # @param [String] command_line
   # @return [Fixnum] exit_code
   def self.run(cmdline = "", returning: :status)
-    ENV["DEBUG"] && puts("Running Kontena.run(#{cmdline.inspect}, returning: #{returning}")
+    ENV["DEBUG"] && STDERR.puts("Running Kontena.run(#{cmdline.inspect}, returning: #{returning}")
     result = Kontena::MainCommand.new(File.basename(__FILE__)).run(cmdline.shellsplit)
-    ENV["DEBUG"] && puts("Command completed, result: #{result.inspect} status: 0")
+    ENV["DEBUG"] && STDERR.puts("Command completed, result: #{result.inspect} status: 0")
     return 0 if returning == :status
     return result if returning == :result
   rescue SystemExit
-    ENV["DEBUG"] && puts("Command completed with failure, result: #{result.inspect} status: #{$!.status}")
+    ENV["DEBUG"] && STDERR.puts("Command completed with failure, result: #{result.inspect} status: #{$!.status}")
     returning == :status ? $!.status : nil
   rescue
-    ENV["DEBUG"] && puts("Command raised #{$!} with message: #{$!.message}\n  #{$!.backtrace.join("  \n")}")
+    ENV["DEBUG"] && STDERR.puts("Command raised #{$!} with message: #{$!.message}\n  #{$!.backtrace.join("  \n")}")
     returning == :status ? 1 : nil
   end
 
@@ -25,12 +25,29 @@ module Kontena
     "kontena-cli/#{Kontena::Cli::VERSION}"
   end
 
+  def self.on_windows?
+    ENV['OS'] == 'Windows_NT' && RUBY_PLATFORM !~ /cygwin/
+  end
+
+  def self.simple_terminal?
+    ENV['KONTENA_SIMPLE_TERM'] || !$stdout.tty?
+  end
+
   def self.pastel
-    @pastel ||= Pastel.new(enabled: $stdout.tty?)
+    @pastel ||= Pastel.new(enabled: !simple_terminal?)
   end
 
   def self.prompt
-    @prompt ||= TTY::Prompt.new(
+    return @prompt if @prompt
+    if simple_terminal?
+      require_relative 'kontena/light_prompt'
+      klass = Kontena::LightPrompt
+    else
+      require 'tty-prompt'
+      klass = TTY::Prompt
+    end
+
+    @prompt = klass.new(
       active_color: :cyan,
       help_color: :white,
       error_color: :red,

data/spec/kontena/cli/grids/health_command_spec.rb

@@ -0,0 +1,390 @@
+require 'kontena/cli/grids/health_command'
+
+describe Kontena::Cli::Grids::HealthCommand do
+  include ClientHelpers
+  include OutputHelpers
+
+  before do
+    allow(subject).to receive(:health_icon) {|health| health.inspect }
+  end
+
+  context "For a initial_size=1 grid" do
+    let :grid do
+      {
+        "id" => "test",
+        "name" => "test",
+        "initial_size" => 1,
+        "node_count" => 1,
+      }
+    end
+
+    context "without any nodes" do
+      let :grid_nodes do
+        { "nodes" => [ ] }
+      end
+
+      describe '#grid_health' do
+        it "returns error" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:error)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns false" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output false, [
+            ":error Grid does not have any initial nodes, and requires at least 1 of 1 initial nodes for operation",
+          ]
+        end
+      end
+    end
+
+    context "with a single offline node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => false,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns error" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:error)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns false" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output false, [
+            ":error Grid does not have any initial nodes online, and requires at least 1 of 1 initial nodes for operation",
+            ":error Initial node node-1 is offline",
+          ]
+        end
+      end
+    end
+
+    context "with a single online node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns ok" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:ok)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns true" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output true, [
+            ":warning Grid only has 1 initial node, and is not high-availability"
+          ]
+        end
+      end
+    end
+  end
+
+  context "For a initial_size=2 grid" do
+    let :grid do
+      {
+        "id" => "test",
+        "name" => "test",
+        "initial_size" => 2,
+        "node_count" => 1,
+      }
+    end
+
+    context "with a single online node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns error" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:error)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns false" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output false, [
+            ":error Grid only has 1 initial nodes, and requires at least 2 of 2 initial nodes for operation",
+          ]
+        end
+      end
+    end
+
+    context "with two online nodes" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-2",
+            "node_number" => 2,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns ok" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:ok)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns true" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output true, [
+            ":warning Grid only has 2 initial nodes, and is not high-availability",
+          ]
+        end
+      end
+    end
+  end
+
+  context "For a initial_size=3 grid" do
+    let :grid do
+      {
+        "id" => "test",
+        "name" => "test",
+        "initial_size" => 3,
+        "node_count" => 1,
+      }
+    end
+
+    context "with a single node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns error" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:error)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns false" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output false, [
+            ":error Grid only has 1 initial nodes, and requires at least 2 of 3 initial nodes for operation",
+          ]
+        end
+      end
+    end
+
+    context "with a single online node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+          {
+            "connected" => false,
+            "name" => "node-2",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns error" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:error)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns false" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output false, [
+            ":error Grid only has 1 initial nodes online, and requires at least 2 of 3 initial nodes for operation",
+            ":error Initial node node-2 is offline",
+          ]
+        end
+      end
+    end
+
+    context "with two online nodes" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-2",
+            "node_number" => 2,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns warning" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:warning)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns true" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output true, [
+            ":warning Grid only has 2 initial nodes of 3 required for high-availability",
+          ]
+        end
+      end
+    end
+
+    context "with two online nodes and one offline node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-2",
+            "node_number" => 2,
+            "initial_member" => true,
+          },
+          {
+            "connected" => false,
+            "name" => "node-3",
+            "node_number" => 3,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns warning" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:warning)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns true" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output true, [
+            ":warning Grid only has 2 initial nodes online of 3 required for high-availability",
+            ":warning Initial node node-3 is offline",
+          ]
+        end
+      end
+    end
+
+    context "with two online initial nodes and one non-initial node" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-2",
+            "node_number" => 2,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-4",
+            "node_number" => 4,
+            "initial_member" => false,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns warning" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:warning)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns true" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output true, [
+            ":warning Grid only has 2 initial nodes of 3 required for high-availability",
+          ]
+        end
+      end
+    end
+
+    context "with three online initial nodes" do
+      let :grid_nodes do
+        { "nodes" => [
+          {
+            "connected" => true,
+            "name" => "node-1",
+            "node_number" => 1,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-2",
+            "node_number" => 2,
+            "initial_member" => true,
+          },
+          {
+            "connected" => true,
+            "name" => "node-3",
+            "node_number" => 3,
+            "initial_member" => true,
+          },
+        ] }
+      end
+
+      describe '#grid_health' do
+        it "returns ok" do
+          expect(subject.grid_health(grid, grid_nodes['nodes'])).to eq(:ok)
+        end
+      end
+
+      describe '#show_grid_health' do
+        it "returns true" do
+          expect{subject.show_grid_health(grid, grid_nodes['nodes'])}.to return_and_output true, [
+            ":ok Grid has all 3 of 3 initial nodes online",
+          ]
+        end
+      end
+    end
+  end
+end