Merge branch 'cli-options'

Implement first CLI options switches and related features:

  * -v enable verbose mode
  * -n enable dry run mode
This commit is contained in:
Thibault Jouan 2014-05-25 13:40:27 +00:00
commit 778816dd68
24 changed files with 399 additions and 38 deletions

View File

@ -0,0 +1,11 @@
Feature: CLI dry run option
Scenario: perfoms a trial run without applying actions
Given a recipe with:
"""
task :say_hello do
echo 'hello'
end
"""
When I successfully execute the recipe with option -n
Then the output must not contain "hello"

View File

@ -5,5 +5,5 @@ Feature: CLI usage
Then the exit status must be 64
And the output must contain exactly:
"""
Usage: producer recipe_file
Usage: producer [-v] [-n] recipe_file
"""

View File

@ -0,0 +1,42 @@
Feature: CLI verbose option
Scenario: prints tasks name
Given a recipe with:
"""
task :say_hello do
end
"""
When I successfully execute the recipe with option -v
Then the output must match /Task:.+say_hello/
Scenario: prints whether condition is met
Given a recipe with:
"""
task :task_ok do
condition { true }
end
task :task_ko do
condition { false }
end
"""
When I successfully execute the recipe with option -v
Then the output must match /task_ok.+ condition: met.*task_ko.* condition: NOT met/
Scenario: prints actions info
Given a recipe with:
"""
task :say_hello do
echo 'hello message'
end
"""
When I successfully execute the recipe with option -v
Then the output must match /say_hello.+ action: echo/
Scenario: formats message with our simple logger
Given a recipe with:
"""
task :say_hello do
end
"""
When I successfully execute the recipe with option -v
Then the output must match /\ATask:.+say_hello.*\n.*condition/

View File

@ -11,6 +11,11 @@ When /^I successfully execute the recipe$/ do
assert_exit_status 0
end
When /^I successfully execute the recipe with option (-\w)$/ do |option|
run_simple "producer #{option} recipe.rb", false
assert_exit_status 0
end
When /^I execute the recipe interactively$/ do
run_interactive 'producer recipe.rb'
end

View File

@ -28,6 +28,7 @@ require 'producer/core/condition'
require 'producer/core/condition/dsl'
require 'producer/core/env'
require 'producer/core/errors'
require 'producer/core/logger_formatter'
require 'producer/core/prompter'
require 'producer/core/recipe'
require 'producer/core/recipe/dsl'

View File

@ -11,6 +11,14 @@ module Producer
@env = env
@arguments = args
end
def name
self.class.name.split('::').last.downcase
end
def to_s
name
end
end
end
end

View File

@ -2,6 +2,10 @@ module Producer
module Core
module Actions
class Echo < Action
def name
'echo'
end
def apply
output.puts arguments.first
end

View File

@ -2,6 +2,10 @@ module Producer
module Core
module Actions
class FileAppend < Action
def name
'file_append'
end
def apply
fs.file_write path, combined_content
end

View File

@ -2,6 +2,10 @@ module Producer
module Core
module Actions
class FileReplaceContent < Action
def name
'file_replace_content'
end
def apply
fs.file_write path, replaced_content
end

View File

@ -2,6 +2,10 @@ module Producer
module Core
module Actions
class FileWriter < Action
def name
'file_write'
end
def apply
case arguments.size
when 2

View File

@ -2,6 +2,10 @@ module Producer
module Core
module Actions
class Mkdir < Action
def name
'mkdir'
end
def apply
case arguments.size
when 1

View File

@ -2,6 +2,10 @@ module Producer
module Core
module Actions
class ShellCommand < Action
def name
'sh'
end
def apply
remote.execute(arguments.first, output)
end

View File

@ -3,7 +3,7 @@ module Producer
class CLI
ArgumentError = Class.new(::ArgumentError)
USAGE = "Usage: #{File.basename $0} recipe_file"
USAGE = "Usage: #{File.basename $0} [-v] [-n] recipe_file"
EX_USAGE = 64
@ -11,6 +11,7 @@ module Producer
def run!(arguments, output: $stderr)
begin
cli = new(arguments)
cli.parse_arguments!
rescue ArgumentError
output.puts USAGE
exit EX_USAGE
@ -19,21 +20,41 @@ module Producer
end
end
attr_reader :arguments, :stdout, :recipe
attr_reader :arguments, :stdout, :env, :recipe
def initialize(arguments, stdout: $stdout)
raise ArgumentError unless arguments.any?
@arguments = arguments
def initialize(args, stdout: $stdout)
@arguments = args
@stdout = stdout
@env = Env.new(output: stdout)
end
def run(worker: Worker.new)
def parse_arguments!
@arguments = arguments.inject([]) do |m, e|
case e
when '-v'
env.log_level = Logger::INFO
when '-n'
env.dry_run = true
else
m << e
end
m
end
raise ArgumentError unless arguments.any?
end
def run(worker: build_worker)
load_recipe
worker.process recipe.tasks
end
def load_recipe
@recipe = Recipe.evaluate_from_file(@arguments.first, Env.new)
@recipe = Recipe.evaluate_from_file(@arguments.first, env)
end
def build_worker
Worker.new(env)
end
end
end

View File

@ -1,15 +1,15 @@
module Producer
module Core
class Env
attr_reader :input, :output, :registry
attr_accessor :target
attr_reader :input, :output, :registry, :logger
attr_accessor :target, :dry_run
def initialize(input: $stdin, output: $stdout, remote: nil, registry: {})
@input = input
@output = output
@registry = registry
@remote = remote
@target = nil
@dry_run = false
end
def remote
@ -23,6 +23,31 @@ module Producer
def []=(key, value)
@registry[key] = value
end
def logger
@logger ||= begin
logger = Logger.new(output)
logger.level = Logger::ERROR
logger.formatter = LoggerFormatter.new
logger
end
end
def log(message)
logger.info message
end
def log_level
logger.level
end
def log_level=(level)
logger.level = level
end
def dry_run?
@dry_run
end
end
end
end

View File

@ -0,0 +1,9 @@
module Producer
module Core
class LoggerFormatter < Logger::Formatter
def call(severity, datetime, progname, message)
message + "\n"
end
end
end
end

View File

@ -17,6 +17,10 @@ module Producer
@condition = condition
end
def to_s
name.to_s
end
def condition_met?
!!@condition
end

View File

@ -1,12 +1,27 @@
module Producer
module Core
class Worker
attr_accessor :env
def initialize(env)
@env = env
end
def process(tasks)
tasks.each { |t| process_task t }
end
def process_task(task)
task.actions.each(&:apply) if task.condition_met?
env.log "Task: #{task} applying"
if task.condition_met?
env.log ' condition: met'
task.actions.each do |e|
env.log " action: #{e} applying"
e.apply unless env.dry_run?
end
else
env.log ' condition: NOT met'
end
end
end
end

View File

@ -3,5 +3,13 @@ require 'spec_helper'
module Producer::Core
describe Action do
it_behaves_like 'action'
describe '#name' do
subject(:action) { described_class.new(double 'env') }
it 'infers action name from class name' do
expect(action.name).to eq 'action'
end
end
end
end

View File

@ -6,29 +6,41 @@ module Producer::Core
include FixturesHelpers
let(:recipe_file) { fixture_path_for 'recipes/some_recipe.rb' }
let(:arguments) { [recipe_file] }
let(:options) { [] }
let(:arguments) { [*options, recipe_file] }
let(:stdout) { StringIO.new }
subject(:cli) { CLI.new(arguments, stdout: stdout) }
describe '.run!' do
let(:cli) { double('cli').as_null_object }
let(:output) { StringIO.new }
subject(:run) { described_class.run! arguments, output: output }
it 'builds a new CLI with given arguments' do
expect(CLI).to receive(:new).with(arguments).and_call_original
expect(described_class)
.to receive(:new).with(arguments).and_call_original
run
end
it 'runs the CLI' do
cli = double 'cli'
allow(CLI).to receive(:new) { cli }
allow(described_class).to receive(:new) { cli }
expect(cli).to receive :run
run
end
context 'when recipe argument is missing' do
let(:arguments) { [] }
it 'parses CLI arguments' do
allow(described_class).to receive(:new) { cli }
expect(cli).to receive :parse_arguments!
run
end
context 'when an argument error is raised' do
before do
allow(described_class).to receive(:new) { cli }
allow(cli).to receive(:parse_arguments!)
.and_raise described_class::ArgumentError
end
it 'exits with a return status of 64' do
expect { run }.to raise_error(SystemExit) { |e|
@ -44,17 +56,15 @@ module Producer::Core
end
describe '#initialize' do
subject(:cli) { CLI.new(arguments) }
it 'assigns $stdout as the default standard output' do
expect(cli.stdout).to be $stdout
it 'assigns the env with CLI output' do
expect(cli.env.output).to be stdout
end
context 'without arguments' do
let(:arguments) { [] }
context 'without options' do
subject(:cli) { described_class.new(arguments) }
it 'raises our ArgumentError exception' do
expect { cli }.to raise_error described_class::ArgumentError
it 'assigns $stdout as the default standard output' do
expect(cli.stdout).to be $stdout
end
end
end
@ -71,6 +81,40 @@ module Producer::Core
end
end
describe '#parse_arguments!' do
context 'with options' do
let(:options) { %w[-v] }
before { cli.parse_arguments! }
it 'removes options from arguments' do
expect(cli.arguments).to eq [recipe_file]
end
context 'verbose' do
it 'sets env logger level to INFO' do
expect(cli.env.log_level).to eq Logger::INFO
end
end
context 'dry run' do
let(:options) { %w[-n] }
it 'enables env dry run' do
expect(cli.env).to be_dry_run
end
end
end
context 'without arguments' do
let(:arguments) { [] }
it 'raises the argument error exception' do
expect { cli.parse_arguments! }.to raise_error described_class::ArgumentError
end
end
end
describe '#run' do
it 'loads the recipe' do
cli.run
@ -84,10 +128,16 @@ module Producer::Core
end
end
describe '#env' do
it 'returns an env' do
expect(cli.env).to be_an Env
end
end
describe '#load_recipe' do
it 'evaluates the recipe file with an env' do
it 'evaluates the recipe file with the CLI env' do
expect(Recipe)
.to receive(:evaluate_from_file).with(recipe_file, kind_of(Env))
.to receive(:evaluate_from_file).with(recipe_file, cli.env)
cli.load_recipe
end
@ -96,5 +146,15 @@ module Producer::Core
expect(cli.recipe.tasks.count).to be 2
end
end
describe '#build_worker' do
it 'returns a worker' do
expect(cli.build_worker).to be_a Worker
end
it 'assigns the CLI env' do
expect(cli.build_worker.env).to eq cli.env
end
end
end
end

View File

@ -2,17 +2,14 @@ require 'spec_helper'
module Producer::Core
describe Env do
subject(:env) { Env.new }
let(:output) { StringIO.new }
subject(:env) { Env.new(output: output) }
describe '#initialize' do
it 'assigns $stdin as the default output' do
expect(env.input).to be $stdin
end
it 'assigns $stdout as the default output' do
expect(env.output).to be $stdout
end
it 'assigns no default target' do
expect(env.target).not_to be
end
@ -21,6 +18,18 @@ module Producer::Core
expect(env.registry).to be_empty
end
it 'assigns dry run as false' do
expect(env.dry_run).to be false
end
context 'when output is not given as argument' do
subject(:env) { Env.new }
it 'assigns $stdout as the default output' do
expect(env.output).to be $stdout
end
end
context 'when input is given as argument' do
let(:input) { double 'input' }
subject(:env) { described_class.new(input: input) }
@ -31,7 +40,6 @@ module Producer::Core
end
context 'when output is given as argument' do
let(:output) { double 'output' }
subject(:env) { described_class.new(output: output) }
it 'assigns the given output' do
@ -58,6 +66,25 @@ module Producer::Core
end
end
describe '#logger' do
it 'returns a logger' do
expect(env.logger).to be_a Logger
end
it 'uses env output' do
env.logger.error 'some message'
expect(output.string).to include 'some message'
end
it 'has a log level of ERROR' do
expect(env.log_level).to eq Logger::ERROR
end
it 'uses our formatter' do
expect(env.logger.formatter).to be_a LoggerFormatter
end
end
describe '#remote' do
it 'builds a Remote with the current target' do
env.target = 'some_hostname.example'
@ -90,5 +117,33 @@ module Producer::Core
expect(env[:some_key]).to eq :some_value
end
end
describe '#log' do
it 'logs an info message through the assigned logger' do
expect(env.logger).to receive(:info).with 'message'
env.log 'message'
end
end
describe '#log_level' do
it 'returns the logger level' do
expect(env.log_level).to eq env.logger.level
end
end
describe '#log_level=' do
it 'sets the logger level' do
env.log_level = Logger::DEBUG
expect(env.logger.level).to eq Logger::DEBUG
end
end
describe '#dry_run?' do
before { env.dry_run = true }
it 'returns true when dry run is enabled' do
expect(env.dry_run?).to be true
end
end
end
end

View File

@ -0,0 +1,18 @@
require 'spec_helper'
module Producer::Core
describe LoggerFormatter do
describe '#call' do
let(:severity) { double 'severity' }
let(:datetime) { double 'datetime' }
let(:progname) { double 'progname' }
let(:message) { 'some message' }
subject { described_class.new.call(severity, datetime, progname, message) }
it 'returns the given message with a line separator' do
expect(subject).to eq "#{message}\n"
end
end
end
end

View File

@ -61,6 +61,12 @@ module Producer::Core
end
end
describe '#to_s' do
it 'includes the task name' do
expect(task.to_s).to include name.to_s
end
end
describe '#condition_met?' do
context 'when condition is truthy' do
let(:condition) { Condition.new([], true) }

View File

@ -2,7 +2,8 @@ require 'spec_helper'
module Producer::Core
describe Worker do
subject(:worker) { described_class.new }
let(:env) { double 'env', log: nil, dry_run?: false }
subject(:worker) { described_class.new(env) }
describe '#process' do
it 'processes each task' do
@ -12,23 +13,59 @@ module Producer::Core
end
describe '#process_task' do
let(:action) { double 'action' }
let(:task) { double('task', actions: [action]).as_null_object }
let(:action) { double('action', to_s: 'echo').as_null_object }
let(:task_name) { 'some_task' }
let(:task) { Task.new(task_name, [action]) }
it 'logs task info' do
expect(env).to receive(:log).with /\ATask: #{task_name}/
worker.process_task task
end
context 'when task condition is met' do
it 'applies the actions' do
expect(action).to receive :apply
worker.process_task task
end
it 'logs condition info' do
expect(env).to receive(:log).with(' condition: met')
worker.process_task task
end
it 'logs action info' do
expect(env).to receive(:log).with /\A action: echo/
worker.process_task task
end
context 'when dry run is enabled' do
before { allow(env).to receive(:dry_run?) { true } }
it 'does not apply the actions' do
expect(action).not_to receive :apply
worker.process_task task
end
end
end
context 'when task condition is not met' do
before { allow(task).to receive(:condition_met?) { false } }
let(:task) { Task.new(task_name, [action], false) }
it 'does not apply the actions' do
expect(action).not_to receive :apply
worker.process_task task
end
it 'logs condition info' do
expect(env).to receive(:log).with(' condition: NOT met')
worker.process_task task
end
end
end
describe '#env' do
it 'returns the assigned env' do
expect(worker.env).to be env
end
end
end

View File

@ -40,5 +40,17 @@ module Producer::Core
expect(action.fs).to be env.remote.fs
end
end
describe '#name' do
it 'returns a word' do
expect(action.name).to match /\A\w+\z/
end
end
describe '#to_s' do
it 'returns a word' do
expect(action.to_s).to eq action.name
end
end
end
end