Class: RSpecSystem::NodeSet::Vagrant
- Defined in:
- lib/rspec-system/node_set/vagrant.rb
Overview
A NodeSet implementation for Vagrant.
Constant Summary collapse
- ENV_TYPE =
'vagrant'
Instance Attribute Summary
Attributes inherited from Base
Instance Method Summary collapse
-
#create_vagrantfile ⇒ Object
private
Create the Vagrantfile for the NodeSet.
-
#initialize(setname, config) ⇒ Vagrant
constructor
Creates a new instance of RSpecSystem::NodeSet::Vagrant.
-
#rcp(opts) ⇒ Boolean
Transfer files to a host in the NodeSet.
-
#run(opts) ⇒ Hash
Run a command on a host in the NodeSet.
-
#setup ⇒ void
Setup the NodeSet by starting all nodes.
-
#ssh_config ⇒ String
private
Here we get vagrant to drop the ssh_config its using so we can monopolize it for transfers and custom stuff.
-
#teardown ⇒ void
Shutdown the NodeSet by shutting down or pausing all nodes.
-
#vagrant(args) ⇒ Object
private
Execute vagrant command in vagrant_path.
Methods included from Util
Methods included from Log
Methods inherited from Base
#default_node, #env_type, #randmac, #random_string, #ssh_exec!, #tmppath
Constructor Details
#initialize(setname, config) ⇒ Vagrant
Creates a new instance of RSpecSystem::NodeSet::Vagrant
17 18 19 20 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 17 def initialize(setname, config) super @vagrant_path = File.(File.join(RSpec.configuration.system_tmp, 'vagrant_projects', setname)) end |
Instance Method Details
#create_vagrantfile ⇒ Object
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Create the Vagrantfile for the NodeSet.
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 122 def create_vagrantfile log.info "[Vagrant#create_vagrantfile] Creating vagrant file here: #{@vagrant_path}" FileUtils.mkdir_p(@vagrant_path) File.open(File.(File.join(@vagrant_path, "Vagrantfile")), 'w') do |f| f.write("Vagrant::Config.run do |c|\n") nodes.each do |k,v| log.debug "Filling in content for #{k}" ps = v.provider_specifics['vagrant'] f.write(<<-EOS) c.vm.define '#{k}' do |v| v.vm.host_name = '#{k}' v.vm.box = '#{ps['box']}' v.vm.box_url = '#{ps['box_url']}' v.vm.base_mac = '#{randmac}' end EOS end f.write('end') end log.debug "[Vagrant#create_vagrantfile] Finished creating vagrant file" nil end |
#rcp(opts) ⇒ Boolean
This is damn ugly, because we ssh in as vagrant, we copy to a temp path then move it later. Its slow and brittle and we need a better solution. Its also very Linux-centrix in its use of temp dirs.
Transfer files to a host in the NodeSet.
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 84 def rcp(opts) #log.debug("[Vagrant@rcp] called with #{opts.inspect}") dest = opts[:d].name source = opts[:sp] dest_path = opts[:dp] # Grab a remote path for temp transfer tmpdest = tmppath # Do the copy and print out results for debugging cmd = "scp -r -F '#{ssh_config}' '#{source}' #{dest}:#{tmpdest}" puts "------------------" puts "localhost$ #{cmd}" r = systemu cmd result = { :exit_code => r[0].exitstatus, :stdout => r[1], :stderr => r[2] } print "#{result[:stdout]}" print "#{result[:stderr]}" puts "Exit code: #{result[:exit_code]}" # Now we move the file into their final destination result = run(:n => opts[:d], :c => "mv #{tmpdest} #{dest_path}") if result[:exit_code] == 0 return true else return false end end |
#run(opts) ⇒ Hash
Run a command on a host in the NodeSet.
65 66 67 68 69 70 71 72 73 74 75 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 65 def run(opts) dest = opts[:n].name cmd = opts[:c] ssh_channels = RSpec.configuration.ssh_channels puts "-----------------" puts "#{dest}$ #{cmd}" result = ssh_exec!(ssh_channels[dest], "cd /tmp && sudo sh -c #{shellescape(cmd)}") puts "-----------------" result end |
#setup ⇒ void
This method returns an undefined value.
Setup the NodeSet by starting all nodes.
25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 25 def setup log.info "[Vagrant#setup] Begin setting up vagrant" create_vagrantfile() teardown() log.info "[Vagrant#setup] Running 'vagrant up'" vagrant("up") # Establish ssh connectivity ssh_channels = {} nodes.each do |k,v| log.info "[Vagrant#setup] establishing Net::SSH channel with #{k}" chan = Net::SSH.start(k, 'vagrant', :config => ssh_config) ssh_channels[k] = chan end RSpec.configuration.ssh_channels = ssh_channels nil end |
#ssh_config ⇒ String
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Here we get vagrant to drop the ssh_config its using so we can monopolize it for transfers and custom stuff. We drop it into a single file, and since its indexed based on our own node names its quite ideal.
153 154 155 156 157 158 159 160 161 162 163 164 165 166 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 153 def ssh_config ssh_config_path = File.(File.join(@vagrant_path, "ssh_config")) begin File.unlink(ssh_config_path) rescue Errno::ENOENT end self.nodes.each do |k,v| Dir.chdir(@vagrant_path) do result = systemu("vagrant ssh-config #{k} >> #{ssh_config_path}") puts result.inspect end end ssh_config_path end |
#teardown ⇒ void
This method returns an undefined value.
Shutdown the NodeSet by shutting down or pausing all nodes.
50 51 52 53 54 55 56 57 58 59 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 50 def teardown log.info "[Vagrant#teardown] closing all ssh channels" RSpec.configuration.ssh_channels.each do |k,v| v.close unless v.closed? end log.info "[Vagrant#teardown] Running 'vagrant destroy'" vagrant("destroy --force") nil end |
#vagrant(args) ⇒ Object
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
This seems a little too specific these days, might want to generalize. It doesn’t use systemu, because we want to see the output immediately, but still - maybe we can make systemu do that.
Execute vagrant command in vagrant_path
175 176 177 178 179 180 |
# File 'lib/rspec-system/node_set/vagrant.rb', line 175 def vagrant(args) Dir.chdir(@vagrant_path) do system("vagrant #{args}") end nil end |