rory-deploy 1.8.4.1

data/spec/dogestry_spec.rb

@@ -0,0 +1,73 @@
+require 'spec_helper'
+require 'centurion/dogestry'
+
+describe Centurion::Dogestry do
+  let(:dogestry_options) {
+    {
+      aws_access_key_id: "abc",
+      aws_secret_key: "xyz",
+      s3_bucket: "s3-registry-test"
+    }
+  }
+  let(:registry) { Centurion::Dogestry.new(dogestry_options) }
+  let(:repo) { 'google/golang' }
+  let(:pull_hosts) {
+    [
+      'tcp://host-1:2375',
+      'tcp://host-2:2375'
+    ]
+  }
+  let(:flags) { "-pullhosts #{pull_hosts.join(',')}"}
+
+  describe '#aws_access_key_id' do
+    it 'returns correct value' do
+      expect(registry.aws_access_key_id).to eq(dogestry_options[:aws_access_key_id])
+    end
+  end
+
+  describe '#aws_secret_key' do
+    it 'returns correct value' do
+      expect(registry.aws_secret_key).to eq(dogestry_options[:aws_secret_key])
+    end
+  end
+
+  describe '#s3_bucket' do
+    it 'returns correct value' do
+      expect(registry.s3_bucket).to eq(dogestry_options[:s3_bucket])
+    end
+  end
+
+  describe '#s3_region' do
+    it 'returns correct default value' do
+      expect(registry.s3_region).to eq("us-east-1")
+    end
+  end
+
+  describe '#s3_url' do
+    it 'returns correct value' do
+      expect(registry.s3_url).to eq("s3://#{registry.s3_bucket}/?region=#{registry.s3_region}")
+    end
+  end
+
+  describe '#exec_command' do
+    it 'returns correct value' do
+      expect(registry.exec_command('pull', repo)).to start_with('dogestry')
+    end
+  end
+
+ describe '#pull' do
+   it 'returns correct value' do
+    if registry.which('dogestry')
+      expect(registry).to receive(:echo).with("dogestry #{flags} pull #{registry.s3_url} #{repo}")
+      registry.pull(repo, pull_hosts)
+    end
+   end
+ end
+
+  describe '#which' do
+    it 'finds dogestry command line' do
+      allow(File).to receive(:executable?).and_return(true)
+      expect(registry.which('dogestry')).to_not be_nil
+    end
+  end
+end

data/spec/logging_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+require 'centurion/logging'
+
+class TestLogging
+  extend Centurion::Logging
+  def self.logger
+    log
+  end
+end
+
+describe Centurion::Logging do
+  let(:message) { %w{ something something_else } }
+
+  context '#info' do
+    it 'passes through to Logger' do
+      expect(TestLogging.logger).to receive(:info).with(message.join(' '))
+      TestLogging.info(*message)
+    end
+  end
+
+  context '#warn' do
+    it 'passes through to Logger' do
+      expect(TestLogging.logger).to receive(:warn).with(message.join(' '))
+      TestLogging.warn(*message)
+    end
+  end
+
+  context '#debug' do
+    it 'passes through to Logger' do
+      expect(TestLogging.logger).to receive(:debug).with(message.join(' '))
+      TestLogging.debug(*message)
+    end
+  end
+
+  context '#error' do
+    it 'passes through to Logger' do
+      expect(TestLogging.logger).to receive(:error).with(message.join(' '))
+      TestLogging.error(*message)
+    end
+  end
+end

data/spec/service_spec.rb

@@ -0,0 +1,288 @@
+require 'spec_helper'
+require 'centurion/service'
+
+describe Centurion::Service do
+
+  let(:service)  { Centurion::Service.new(:redis) }
+  let(:hostname) { 'shakespeare' }
+  let(:image)    { 'redis' }
+
+  it 'creates a service from the environment' do
+    extend Capistrano::DSL
+    set_current_environment(:test)
+    set(:name, 'mycontainer')
+    set(:image, image)
+    set(:hostname, hostname)
+    set(:binds, [ Centurion::Service::Volume.new('/foo', '/foo/bar') ])
+    set(:port_bindings, [ Centurion::Service::PortBinding.new(12340, 80, 'tcp') ])
+
+    svc = Centurion::Service.from_env
+    expect(svc.name).to eq('mycontainer')
+    expect(svc.image).to eq(image)
+    expect(svc.dns).to be_nil
+    expect(svc.volumes.size).to eq(1)
+    expect(svc.volumes.first.host_volume).to eq('/foo')
+    expect(svc.port_bindings.size).to eq(1)
+    expect(svc.port_bindings.first.container_port).to eq(80)
+  end
+
+  it 'starts with a command' do
+    service.command = ['redis-server']
+    expect(service.command).to eq(['redis-server'])
+  end
+
+  it 'has memory bounds' do
+    service.memory = 1024
+    expect(service.memory).to eq(1024)
+  end
+
+  it 'rejects non-numeric memory bounds' do
+    expect(-> { service.memory = 'all' }).to raise_error
+  end
+
+  it 'has cpu shares bounds' do
+    service.cpu_shares = 512
+    expect(service.cpu_shares).to eq(512)
+  end
+
+  it 'rejects non-numeric cpu shares' do
+    expect(-> { service.cpu_shares = 'all' }).to raise_error
+  end
+
+  it 'has a custom dns association' do
+    service.dns = 'redis.example.com'
+    expect(service.dns).to eq('redis.example.com')
+  end
+
+  it 'boots from a docker image' do
+    service.image = 'registry.hub.docker.com/library/redis'
+    expect(service.image).to eq('registry.hub.docker.com/library/redis')
+  end
+
+  it 'has env vars' do
+    service.add_env_vars(SLAVE_OF: '127.0.0.1')
+    service.add_env_vars(USE_AOF: '1')
+    expect(service.env_vars).to eq(SLAVE_OF: '127.0.0.1', USE_AOF: '1')
+  end
+
+  it 'has volume bindings' do
+    service.add_volume('/volumes/redis/data', '/data')
+    service.add_volume('/volumes/redis/config', '/config')
+    expect(service.volumes).to eq([Centurion::Service::Volume.new('/volumes/redis/data', '/data'),
+                                   Centurion::Service::Volume.new('/volumes/redis/config', '/config')])
+  end
+
+  it 'has port mappings' do
+    service.add_port_bindings(8000, 6379, 'tcp', '127.0.0.1')
+    service.add_port_bindings(18000, 16379, 'tcp', '127.0.0.1')
+    expect(service.port_bindings).to eq([Centurion::Service::PortBinding.new(8000, 6379, 'tcp', '127.0.0.1'),
+                                         Centurion::Service::PortBinding.new(18000, 16379, 'tcp', '127.0.0.1')])
+  end
+
+  it 'builds a list of public ports for the service' do
+    service.add_port_bindings(8000, 6379, 'tcp', '127.0.0.1')
+    service.add_port_bindings(18000, 16379, 'tcp', '127.0.0.1')
+    expect(service.public_ports).to eq([8000, 18000])
+  end
+
+  context 'building a container configuration' do
+      service = Centurion::Service.new(:redis)
+      service.image = 'http://registry.hub.docker.com/library/redis'
+      service.command = ['redis-server', '--appendonly', 'yes']
+      service.memory = 1024
+      service.cpu_shares = 512
+      service.add_env_vars(SLAVE_OF: '127.0.0.2')
+      service.add_port_bindings(8000, 6379, 'tcp', '10.0.0.1')
+      service.network_mode = 'host'
+      service.add_volume('/volumes/redis.8000', '/data')
+
+    it 'builds a valid docker container configuration' do
+      expect(service.build_config('example.com')).to eq({
+        'Image' => 'http://registry.hub.docker.com/library/redis',
+        'Cmd' => ['redis-server', '--appendonly', 'yes'],
+        'Memory' => 1024,
+        'CpuShares' => 512,
+        'ExposedPorts' => {'6379/tcp' => {}},
+        'Env' => ['SLAVE_OF=127.0.0.2'],
+        'Volumes' => {'/data' => {}},
+
+        # TODO: Ignoring this for now because Docker 1.6
+        # https://github.com/newrelic/centurion/issues/117
+        # 'VolumesFrom' => 'parent'
+      })
+    end
+
+    it 'overrides the default hostname when passed a block' do
+      expect(service.build_config('example.com') { |s| "host.#{s}" }).to eq({
+        'Image' => 'http://registry.hub.docker.com/library/redis',
+        'Hostname' => 'host.example.com',
+        'Cmd' => ['redis-server', '--appendonly', 'yes'],
+        'Memory' => 1024,
+        'CpuShares' => 512,
+        'ExposedPorts' => {'6379/tcp' => {}},
+        'Env' => ['SLAVE_OF=127.0.0.2'],
+        'Volumes' => {'/data' => {}},
+
+        # TODO: Ignoring this for now because Docker 1.6
+        # https://github.com/newrelic/centurion/issues/117
+        # 'VolumesFrom' => 'parent'
+      })
+    end
+  end
+
+  it 'interpolates hostname into env variables' do
+    allow(Socket).to receive(:getaddrinfo).and_return([["AF_INET", 0, "93.184.216.34", "93.184.216.34", 2, 1, 6]])
+    service = Centurion::Service.new(:redis)
+    service.add_env_vars(HOST: '%DOCKER_HOSTNAME%')
+
+    expect(service.build_config('example.com')['Env']).to eq(['HOST=example.com'])
+  end
+
+  it 'interpolates host ip into env variables' do
+    allow(Socket).to receive(:getaddrinfo).and_return([["AF_INET", 0, "93.184.216.34", "93.184.216.34", 2, 1, 6]])
+    service = Centurion::Service.new(:redis)
+    service.add_env_vars(HOST: '%DOCKER_HOST_IP%')
+
+    expect(service.build_config('example.com')['Env']).to eq(['HOST=93.184.216.34'])
+  end
+
+  it 'does not blow up on non-string values' do
+    expect { service.add_env_vars(SOMETHING: true) }.not_to raise_error
+  end
+
+  it 'builds a valid docker host configuration' do
+    service = Centurion::Service.new(:redis)
+    service.dns = 'example.com'
+    service.add_port_bindings(8000, 6379)
+    service.cap_adds = ['IPC_BIND', 'NET_RAW']
+    service.cap_drops = ['DAC_OVERRIDE']
+    service.add_volume('/volumes/redis.8000', '/data')
+
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('on-failure', 10))).to eq({
+      'Binds' => ['/volumes/redis.8000:/data'],
+      'CapAdd' => ['IPC_BIND', 'NET_RAW'],
+      'CapDrop' => ['DAC_OVERRIDE'],
+      'PortBindings' => {
+        '6379/tcp' => [{'HostPort' => '8000'}]
+      },
+      'NetworkMode' => 'bridge',
+      'Dns' => 'example.com',
+      'RestartPolicy' => {
+        'Name' => 'on-failure',
+        'MaximumRetryCount' => 10
+      }
+    })
+  end
+
+  it 'ignores garbage restart policy' do
+    service = Centurion::Service.new(:redis)
+
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('garbage'))).to eq({
+       'Binds' => [],
+       'CapAdd' => [],
+       'CapDrop' => [],
+       'PortBindings' => {},
+       'NetworkMode' => 'bridge',
+       'RestartPolicy' => {
+         'Name' => 'on-failure',
+         'MaximumRetryCount' => 10
+       }
+     })
+  end
+
+  it 'accepts "no" restart policy' do
+    service = Centurion::Service.new(:redis)
+
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('no'))).to eq({
+      'Binds' => [],
+      'CapAdd' => [],
+      'CapDrop' => [],
+      'PortBindings' => {},
+      'NetworkMode' => 'bridge',
+       'RestartPolicy' => {
+         'Name' => 'no',
+       }
+     })
+  end
+
+  it 'accepts "always" restart policy' do
+    service = Centurion::Service.new(:redis)
+
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('always'))).to eq({
+      'Binds' => [],
+      'CapAdd' => [],
+      'CapDrop' => [],
+      'PortBindings' => {},
+      'NetworkMode' => 'bridge',
+       'RestartPolicy' => {
+         'Name' => 'always',
+       }
+     })
+  end
+
+  it 'accepts "on-failure" restart policy with retry count' do
+    service = Centurion::Service.new(:redis)
+
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('on-failure', 50))).to eq({
+      'Binds' => [],
+      'CapAdd' => [],
+      'CapDrop' => [],
+      'NetworkMode' => 'bridge',
+      'PortBindings' => {},
+       'RestartPolicy' => {
+         'Name' => 'on-failure',
+         'MaximumRetryCount' => 50
+       }
+     })
+  end
+
+  it 'builds docker configuration for volume binds' do
+    service.add_volume('/volumes/redis/data', '/data')
+    expect(service.volume_binds_config).to eq(['/volumes/redis/data:/data'])
+  end
+
+  it 'builds docker configuration for port bindings' do
+    service.add_port_bindings(8000, 6379, 'tcp', '127.0.0.1')
+    expect(service.port_bindings_config).to eq({
+      '6379/tcp' => [{'HostPort' => '8000', 'HostIp' => '127.0.0.1'}]
+    })
+  end
+
+  it 'builds docker configuration for port bindings without host ip' do
+    service.add_port_bindings(8000, 6379, 'tcp')
+    expect(service.port_bindings_config).to eq({
+      '6379/tcp' => [{'HostPort' => '8000'}]
+    })
+  end
+
+  it 'builds docker configuration for container-linked networking' do
+    service.network_mode = 'container:a2e8937b'
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('on-failure', 50))).to eq({
+      'Binds' => [],
+      'CapAdd' => [],
+      'CapDrop' => [],
+      'NetworkMode' => 'container:a2e8937b',
+      'PortBindings' => {},
+       'RestartPolicy' => {
+         'Name' => 'on-failure',
+         'MaximumRetryCount' => 50
+      }
+   })
+  end
+
+  it 'builds docker configuration for host networking' do
+    service.network_mode = 'host'
+    expect(service.build_host_config(Centurion::Service::RestartPolicy.new('on-failure', 50))).to eq({
+      'Binds' => [],
+      'CapAdd' => [],
+      'CapDrop' => [],
+      'NetworkMode' => 'host',
+      'PortBindings' => {},
+       'RestartPolicy' => {
+         'Name' => 'on-failure',
+         'MaximumRetryCount' => 50
+      }
+   })
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'simplecov'
+SimpleCov.start do
+  add_filter '/spec'
+end
+
+$: << File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib'))
+Dir[File.dirname(__FILE__) + '/support/**/*.rb'].each {|f| require f}

data/spec/support/matchers/capistrano_dsl_matchers.rb

@@ -0,0 +1,13 @@
+RSpec::Matchers.define :have_key_and_value do |expected_key, expected_value|
+  match do |actual|
+    actual.env[actual.current_environment].has_key?(expected_key.to_sym) && (actual.fetch(expected_key.to_sym) == expected_value)
+  end
+
+  failure_message do |actual|
+    "expected that #{actual.env[actual.current_environment].keys.inspect} would include #{expected_key.inspect} with value #{expected_value.inspect}"
+  end
+
+  failure_message_when_negated do |actual|
+    "expected that #{actual.env[actual.current_environment].keys.join(', ')} would not include #{expected_key.inspect} with value #{expected_value.inspect}"
+  end
+end

data/spec/support/matchers/exit_code_matches.rb

@@ -0,0 +1,38 @@
+# https://gist.github.com/mmasashi/58bd7e2668836a387856
+RSpec::Matchers.define :terminate do |code|
+  actual = nil
+
+  def supports_block_expectations?
+    true
+  end
+
+  match do |block|
+    begin
+      block.call
+    rescue SystemExit => e
+      actual = e.status
+    end
+    actual and actual == status_code
+  end
+
+  chain :with_code do |status_code|
+    @status_code = status_code
+  end
+
+  failure_message do |block|
+    "expected block to call exit(#{status_code}) but exit" +
+      (actual.nil? ? " not called" : "(#{actual}) was called")
+  end
+
+  failure_message_when_negated do |block|
+    "expected block not to call exit(#{status_code})"
+  end
+
+  description do
+    "expect block to call exit(#{status_code})"
+  end
+
+  def status_code
+    @status_code ||= 0
+  end
+end