1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
|
require 'tempfile'
require 'minitest/autorun'
require 'minitest/reporters'
require 'mocha/minitest'
require 'json'
$LOAD_PATH.unshift(File.dirname(__FILE__))
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
require 'sshkit'
Dir[File.expand_path('test/support/*.rb')].each { |file| require file }
class UnitTest < Minitest::Test
def setup
SSHKit.reset_configuration!
end
SSHKit::Backend::ConnectionPool.class_eval do
alias_method :old_flush_connections, :flush_connections
def flush_connections
Thread.current[:sshkit_pool] = {}
end
end
end
class FunctionalTest < Minitest::Test
def setup
unless VagrantWrapper.running?
warn "Vagrant VMs are not running. Please, start it manually with `vagrant up`"
end
end
private
def create_user_with_key(username, password = :secret)
username, password = username.to_s, password.to_s
keys = VagrantWrapper.hosts.collect do |_name, host|
Net::SSH.start(host.hostname, host.user, port: host.port, password: host.password) do |ssh|
# Remove the user, make it again, force-generate a key for him
# short keys save us a few microseconds
ssh.exec!("sudo userdel -rf #{username}; true") # The `rescue nil` of the shell world
ssh.exec!("sudo useradd -m #{username}")
ssh.exec!("sudo echo y | ssh-keygen -b 1024 -f #{username} -N ''")
ssh.exec!("sudo chown vagrant:vagrant #{username}*")
ssh.exec!("sudo echo #{username}:#{password} | chpasswd")
# Make the .ssh directory, change the ownership and the
ssh.exec!("sudo mkdir -p /home/#{username}/.ssh")
ssh.exec!("sudo chown #{username}:#{username} /home/#{username}/.ssh")
ssh.exec!("sudo chmod 700 /home/#{username}/.ssh")
# Move the key to authorized keys and chown and chmod it
ssh.exec!("sudo cat #{username}.pub > /home/#{username}/.ssh/authorized_keys")
ssh.exec!("sudo chown #{username}:#{username} /home/#{username}/.ssh/authorized_keys")
ssh.exec!("sudo chmod 600 /home/#{username}/.ssh/authorized_keys")
key = ssh.exec!("cat /home/vagrant/#{username}")
# Clean Up Files
ssh.exec!("sudo rm #{username} #{username}.pub")
key
end
end
Hash[VagrantWrapper.hosts.collect { |n, _h| n.to_sym }.zip(keys)]
end
end
#
# Force colours in Autotest
#
Minitest::Reporters.use! Minitest::Reporters::SpecReporter.new
|