cd73cea850
Add two new hooks pre-app-boot and post-app-boot. They are analagous to the pre/post proxy reboot hooks. If the boot strategy deploys in groups, then the hooks are called once per group of hosts and `KAMAL_HOSTS` contains a comma delimited list of the hosts in that group. If all hosts are deployed to at once, then they are called once with `KAMAL_HOSTS` containing all the hosts. It is possible to have pauses between groups of hosts in the boot config, where this is the case the pause happens after the post-app-boot hook is called.
100 lines
2.5 KiB
Ruby
100 lines
2.5 KiB
Ruby
require "bundler/setup"
|
|
require "active_support/test_case"
|
|
require "active_support/testing/autorun"
|
|
require "active_support/testing/stream"
|
|
require "rails/test_unit/line_filtering"
|
|
require "debug"
|
|
require "mocha/minitest" # using #stubs that can alter returns
|
|
require "minitest/autorun" # using #stub that take args
|
|
require "sshkit"
|
|
require "kamal"
|
|
|
|
ActiveSupport::LogSubscriber.logger = ActiveSupport::Logger.new(STDOUT) if ENV["VERBOSE"]
|
|
|
|
# Applies to remote commands only.
|
|
SSHKit.config.backend = SSHKit::Backend::Printer
|
|
|
|
class SSHKit::Backend::Printer
|
|
def upload!(local, location, **kwargs)
|
|
local = local.string.inspect if local.respond_to?(:string)
|
|
puts "Uploading #{local} to #{location} on #{host}"
|
|
end
|
|
end
|
|
|
|
# Ensure local commands use the printer backend too.
|
|
# See https://github.com/capistrano/sshkit/blob/master/lib/sshkit/dsl.rb#L9
|
|
module SSHKit
|
|
module DSL
|
|
def run_locally(&block)
|
|
SSHKit::Backend::Printer.new(SSHKit::Host.new(:local), &block).run
|
|
end
|
|
end
|
|
end
|
|
|
|
class ActiveSupport::TestCase
|
|
include ActiveSupport::Testing::Stream
|
|
extend Rails::LineFiltering
|
|
|
|
private
|
|
setup do
|
|
SSHKit::Backend::Netssh.pool.close_connections
|
|
end
|
|
|
|
def stdouted
|
|
capture(:stdout) { yield }.strip
|
|
end
|
|
|
|
def stderred
|
|
capture(:stderr) { yield }.strip
|
|
end
|
|
|
|
def with_test_secrets(**files)
|
|
setup_test_secrets(**files)
|
|
yield
|
|
ensure
|
|
teardown_test_secrets
|
|
end
|
|
|
|
def setup_test_secrets(**files)
|
|
@original_pwd = Dir.pwd
|
|
@secrets_tmpdir = Dir.mktmpdir
|
|
fixtures_dup = File.join(@secrets_tmpdir, "test")
|
|
FileUtils.mkdir_p(fixtures_dup)
|
|
FileUtils.cp_r("test/fixtures/", fixtures_dup)
|
|
|
|
Dir.chdir(@secrets_tmpdir)
|
|
FileUtils.mkdir_p(".kamal")
|
|
Dir.chdir(".kamal") do
|
|
files.each do |filename, contents|
|
|
File.binwrite(filename.to_s, contents)
|
|
end
|
|
end
|
|
end
|
|
|
|
def teardown_test_secrets
|
|
Dir.chdir(@original_pwd)
|
|
FileUtils.rm_rf(@secrets_tmpdir)
|
|
end
|
|
end
|
|
|
|
class SecretAdapterTestCase < ActiveSupport::TestCase
|
|
setup do
|
|
`true` # Ensure $? is 0
|
|
end
|
|
|
|
private
|
|
def stub_ticks
|
|
Kamal::Secrets::Adapters::Base.any_instance.stubs(:`)
|
|
end
|
|
|
|
def stub_ticks_with(command, succeed: true)
|
|
# Sneakily run `false`/`true` after a match to set $? to 1/0
|
|
stub_ticks.with { |c| c == command && (succeed ? `true` : `false`) }
|
|
Kamal::Secrets::Adapters::Base.any_instance.stubs(:`)
|
|
end
|
|
|
|
def shellunescape(string)
|
|
"\"#{string}\"".undump.gsub(/\\([{}])/, "\\1")
|
|
end
|
|
end
|