kumogata 0.1.0

data/spec/kumogata_create_spec.rb

@@ -0,0 +1,183 @@
+describe 'Kumogata::Client#create' do
+  it 'create a stack from Ruby template' do
+    template = <<-EOS
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType "t1.micro"
+    end
+  end
+end
+
+Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+end
+    EOS
+
+    run_client(:create, :template => template) do |client, cf|
+      json = eval_template(template, :update_deletion_policy => true).to_json
+      stacks = double('status')
+
+      output = make_double('output') do |obj|
+        obj.should_receive(:key) { 'AZ' }
+        obj.should_receive(:value) { 'ap-northeast-1b' }
+      end
+
+      resource_summary = make_double('resource_summary') do |obj|
+        obj.should_receive(:[]).with(:logical_resource_id) { 'myEC2Instance' }
+        obj.should_receive(:[]).with(:physical_resource_id) { 'i-XXXXXXXX' }
+        obj.should_receive(:[]).with(:resource_type) { 'AWS::EC2::Instance' }
+        obj.should_receive(:[]).with(:resource_status) { 'CREATE_COMPLETE' }
+        obj.should_receive(:[]).with(:resource_status_reason) { nil }
+        obj.should_receive(:[]).with(:last_updated_timestamp) { '2014-03-02 04:35:12 UTC' }
+      end
+
+      stack = make_double('stack') do |obj|
+        obj.should_receive(:status).and_return(
+            'CREATE_COMPLETE', 'CREATE_COMPLETE',
+            'DELETE_COMPLETE', 'DELETE_COMPLETE', 'DELETE_COMPLETE')
+        obj.should_receive(:outputs) { [output] }
+        obj.should_receive(:resource_summaries) { [resource_summary] }
+        obj.should_receive(:delete)
+      end
+
+      stacks = make_double('stacks') do |obj|
+        obj.should_receive(:create)
+           .with('kumogata-XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX', json, {}) { stack }
+        obj.should_receive(:[])
+           .with('kumogata-XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX') { stack }
+      end
+
+      cf.should_receive(:stacks).twice { stacks }
+    end
+  end
+
+  it 'create a stack from Ruby template with parameters' do
+    template = <<-EOS
+Parameters do
+  InstanceType do
+    Default "t1.micro"
+    Description "Instance Type"
+    Type "String"
+  end
+end
+
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType { Ref "InstanceType" }
+    end
+  end
+end
+
+Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+end
+    EOS
+
+    run_client(:create, :template => template, :options => {:parameters => ['InstanceType=m1.large']}) do |client, cf|
+      json = eval_template(template, :update_deletion_policy => true).to_json
+      stacks = double('stacks')
+
+      output = make_double('output') do |obj|
+        obj.should_receive(:key) { 'AZ' }
+        obj.should_receive(:value) { 'ap-northeast-1b' }
+      end
+
+      resource_summary = make_double('resource_summary') do |obj|
+        obj.should_receive(:[]).with(:logical_resource_id) { 'myEC2Instance' }
+        obj.should_receive(:[]).with(:physical_resource_id) { 'i-XXXXXXXX' }
+        obj.should_receive(:[]).with(:resource_type) { 'AWS::EC2::Instance' }
+        obj.should_receive(:[]).with(:resource_status) { 'CREATE_COMPLETE' }
+        obj.should_receive(:[]).with(:resource_status_reason) { nil }
+        obj.should_receive(:[]).with(:last_updated_timestamp) { '2014-03-02 04:35:12 UTC' }
+      end
+
+      stack = make_double('stack') do |obj|
+        obj.should_receive(:status).and_return(
+            'CREATE_COMPLETE', 'CREATE_COMPLETE',
+            'DELETE_COMPLETE', 'DELETE_COMPLETE', 'DELETE_COMPLETE')
+        obj.should_receive(:outputs) { [output] }
+        obj.should_receive(:resource_summaries) { [resource_summary] }
+        obj.should_receive(:delete)
+      end
+
+      stacks = make_double('stacks') do |obj|
+        obj.should_receive(:create)
+           .with('kumogata-XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX', json, {:parameters=>{"InstanceType"=>"m1.large"}}) { stack }
+        obj.should_receive(:[])
+           .with('kumogata-XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX') { stack }
+      end
+
+      cf.should_receive(:stacks).twice { stacks }
+    end
+  end
+
+  it 'create a stack from Ruby template with stack name' do
+    template = <<-EOS
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType "t1.micro"
+    end
+  end
+end
+
+Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+end
+    EOS
+
+    run_client(:create, :arguments => ['MyStack'], :template => template) do |client, cf|
+      json = eval_template(template).to_json
+      stacks = double('status')
+
+      output = make_double('output') do |obj|
+        obj.should_receive(:key) { 'AZ' }
+        obj.should_receive(:value) { 'ap-northeast-1b' }
+      end
+
+      resource_summary = make_double('resource_summary') do |obj|
+        obj.should_receive(:[]).with(:logical_resource_id) { 'myEC2Instance' }
+        obj.should_receive(:[]).with(:physical_resource_id) { 'i-XXXXXXXX' }
+        obj.should_receive(:[]).with(:resource_type) { 'AWS::EC2::Instance' }
+        obj.should_receive(:[]).with(:resource_status) { 'CREATE_COMPLETE' }
+        obj.should_receive(:[]).with(:resource_status_reason) { nil }
+        obj.should_receive(:[]).with(:last_updated_timestamp) { '2014-03-02 04:35:12 UTC' }
+      end
+
+      stack = make_double('stack') do |obj|
+        obj.should_receive(:status).and_return(
+            'CREATE_COMPLETE',
+            'CREATE_COMPLETE')
+        obj.should_receive(:outputs) { [output] }
+        obj.should_receive(:resource_summaries) { [resource_summary] }
+      end
+
+      stacks = make_double('status') do |obj|
+        obj.should_receive(:create)
+           .with('MyStack', json, {}) { stack }
+      end
+
+      cf.should_receive(:stacks) { stacks }
+    end
+  end
+end

data/spec/kumogata_delete_spec.rb

@@ -0,0 +1,20 @@
+describe 'Kumogata::Client#delete' do
+  it 'update a stack from Ruby template' do
+    run_client(:delete, :arguments => ['MyStack'], :options => {:force => true}) do |client, cf|
+      stacks = double('status')
+
+      stack = make_double('stack') do |obj|
+        obj.should_receive(:delete).with(no_args())
+        obj.should_receive(:status).and_return(
+            'DELETE_COMPLETE', 'DELETE_COMPLETE', 'DELETE_COMPLETE')
+      end
+
+      stacks = make_double('stacks') do |obj|
+        obj.should_receive(:[])
+           .with('MyStack') { stack }
+      end
+
+      cf.should_receive(:stacks) { stacks }
+    end
+  end
+end

data/spec/kumogata_update_spec.rb

@@ -0,0 +1,121 @@
+describe 'Kumogata::Client#update' do
+  it 'update a stack from Ruby template' do
+    template = <<-EOS
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType "t1.micro"
+    end
+  end
+end
+
+Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+end
+    EOS
+
+    run_client(:update, :arguments => ['MyStack'], :template => template) do |client, cf|
+      json = eval_template(template).to_json
+      stacks = double('status')
+
+      output = make_double('output') do |obj|
+        obj.should_receive(:key) { 'AZ' }
+        obj.should_receive(:value) { 'ap-northeast-1b' }
+      end
+
+      resource_summary = make_double('resource_summary') do |obj|
+        obj.should_receive(:[]).with(:logical_resource_id) { 'myEC2Instance' }
+        obj.should_receive(:[]).with(:physical_resource_id) { 'i-XXXXXXXX' }
+        obj.should_receive(:[]).with(:resource_type) { 'AWS::EC2::Instance' }
+        obj.should_receive(:[]).with(:resource_status) { 'UPDATE_COMPLETE' }
+        obj.should_receive(:[]).with(:resource_status_reason) { nil }
+        obj.should_receive(:[]).with(:last_updated_timestamp) { '2014-03-02 04:35:12 UTC' }
+      end
+
+      stack = make_double('stack') do |obj|
+        obj.should_receive(:update).with(:template => json)
+        obj.should_receive(:status).and_return(
+            'UPDATE_COMPLETE', 'UPDATE_COMPLETE', 'UPDATE_COMPLETE')
+        obj.should_receive(:outputs) { [output] }
+        obj.should_receive(:resource_summaries) { [resource_summary] }
+      end
+
+      stacks = make_double('stacks') do |obj|
+        obj.should_receive(:[])
+           .with('MyStack') { stack }
+      end
+
+      cf.should_receive(:stacks) { stacks }
+    end
+  end
+
+  it 'update a stack from Ruby template with parameters' do
+    template = <<-EOS
+Parameters do
+  InstanceType do
+    Default "t1.micro"
+    Description "Instance Type"
+    Type "String"
+  end
+end
+
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType { Ref "InstanceType" }
+    end
+  end
+end
+
+Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+end
+    EOS
+
+    run_client(:update, :arguments => ['MyStack'], :template => template, :options => {:parameters => ['InstanceType=m1.large']}) do |client, cf|
+      json = eval_template(template).to_json
+      stacks = double('status')
+
+      output = make_double('output') do |obj|
+        obj.should_receive(:key) { 'AZ' }
+        obj.should_receive(:value) { 'ap-northeast-1b' }
+      end
+
+      resource_summary = make_double('resource_summary') do |obj|
+        obj.should_receive(:[]).with(:logical_resource_id) { 'myEC2Instance' }
+        obj.should_receive(:[]).with(:physical_resource_id) { 'i-XXXXXXXX' }
+        obj.should_receive(:[]).with(:resource_type) { 'AWS::EC2::Instance' }
+        obj.should_receive(:[]).with(:resource_status) { 'UPDATE_COMPLETE' }
+        obj.should_receive(:[]).with(:resource_status_reason) { nil }
+        obj.should_receive(:[]).with(:last_updated_timestamp) { '2014-03-02 04:35:12 UTC' }
+      end
+
+      stack = make_double('stack') do |obj|
+        obj.should_receive(:update).with(:template => json, :parameters=>{"InstanceType"=>"m1.large"})
+        obj.should_receive(:status).and_return(
+            'UPDATE_COMPLETE', 'UPDATE_COMPLETE', 'UPDATE_COMPLETE')
+        obj.should_receive(:outputs) { [output] }
+        obj.should_receive(:resource_summaries) { [resource_summary] }
+      end
+
+      stacks = make_double('stacks') do |obj|
+        obj.should_receive(:[])
+           .with('MyStack') { stack }
+      end
+
+      cf.should_receive(:stacks) { stacks }
+    end
+  end
+end

data/spec/kumogata_validate_spec.rb

@@ -0,0 +1,135 @@
+describe 'Kumogata::Client#validate' do
+  it 'validate Ruby template (without error)' do
+    template = <<-EOS
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType "t1.micro"
+    end
+  end
+end
+
+Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+end
+    EOS
+
+    run_client(:validate, :template => template) do |client, cf|
+      json = eval_template(template).to_json
+
+      cf.should_receive(:validate_template).with(json) {
+        {}
+      }
+    end
+  end
+
+  it 'validate Ruby template (with error)' do
+    template = <<-EOS
+Resources do
+  myEC2Instance do
+    Type "AWS::EC2::Instance"
+    Properties do
+      ImageId "ami-XXXXXXXX"
+      InstanceType "t1.micro"
+    end
+  end
+end
+
+#Outputs do
+  AZ do
+    Value do
+      Fn__GetAtt "myEC2Instance", "AvailabilityZone"
+    end
+  end
+#end
+    EOS
+
+    expect {
+      run_client(:validate, :template => template) do |client, cf|
+        json = eval_template(template).to_json
+
+        cf.should_receive(:validate_template).with(json) {
+          {
+            :code => 'CODE',
+            :message => 'MESSAGE'
+          }
+        }
+      end
+    }.to raise_error('CODE: MESSAGE')
+  end
+
+  it 'validate JSON template (without error)' do
+    template = <<-EOS
+{
+  "Resources": {
+    "myEC2Instance": {
+      "Type": "AWS::EC2::Instance",
+      "Properties": {
+        "ImageId": "ami-XXXXXXXX",
+        "InstanceType": "t1.micro"
+      }
+    }
+  },
+  "AZ": {
+    "Value": {
+      "Fn::GetAtt": [
+        "myEC2Instance",
+        "AvailabilityZone"
+      ]
+    }
+  }
+}
+    EOS
+
+    run_client(:validate, :template => template, :template_ext => '.template') do |client, cf|
+      json = JSON.parse(template).to_json
+
+      cf.should_receive(:validate_template).with(json) {
+        {}
+      }
+    end
+  end
+
+  it 'validate JSON template (with error)' do
+    template = <<-EOS
+{
+  "Resources": {
+    "myEC2Instance": {
+      "Type": "AWS::EC2::Instance",
+      "Properties": {
+        "ImageId": "ami-XXXXXXXX",
+        "InstanceType": "t1.micro"
+      }
+    }
+  },
+  "AZ": {
+    "Value": {
+      "Fn::GetAtt": [
+        "myEC2Instance",
+        "AvailabilityZone"
+      ]
+    }
+  }
+}
+    EOS
+
+    expect {
+      run_client(:validate, :template => template, :template_ext => '.template') do |client, cf|
+        json = JSON.parse(template).to_json
+
+        cf.should_receive(:validate_template).with(json) {
+          {
+            :code => 'CODE',
+            :message => 'MESSAGE'
+          }
+        }
+      end
+    }.to raise_error('CODE: MESSAGE')
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,68 @@
+require 'kumogata'
+require 'tempfile'
+require 'uuidtools'
+
+class UUIDTools::UUID
+  def self.timestamp_create; 'XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX'; end
+end
+
+def tempfile(content, template_ext)
+  basename = "#{File.basename __FILE__}.#{$$}"
+  basename = [basename, template_ext]
+
+  Tempfile.open(basename) do |f|
+    f << content
+    f.flush
+    f.rewind
+    yield(f)
+  end
+end
+
+def run_client(command, options = {})
+  $stdout = open('/dev/null', 'w')
+
+  kumogata_template = options[:template]
+  kumogata_arguments = options[:arguments] || []
+  kumogata_options = Kumogata::ArgumentParser::DEFAULT_OPTIONS.merge(options[:options] || {})
+  template_ext = options[:template_ext] || '.rb'
+
+  client = Kumogata::Client.new(kumogata_options)
+  cloud_formation = client.instance_variable_get(:@cloud_formation)
+  yield(client, cloud_formation) if block_given?
+
+  if kumogata_template
+    tempfile(kumogata_template, template_ext) do |f|
+      kumogata_arguments.unshift(f.path)
+      client.send(command, *kumogata_arguments)
+    end
+  else
+    client.send(command, *kumogata_arguments)
+  end
+end
+
+def eval_template(template, options = {})
+  kumogata_options = Kumogata::ArgumentParser::DEFAULT_OPTIONS.merge(options[:options] || {})
+  template_ext = options[:template_ext] || '.rb'
+
+  template = tempfile(template, template_ext) do |f|
+    Kumogata::Client.new(kumogata_options).send(:evaluate_template, f)
+  end
+
+  if options[:update_deletion_policy]
+    update_deletion_policy(template)
+  end
+
+  return template
+end
+
+def update_deletion_policy(template)
+  template['Resources'].each do |k, v|
+    v['DeletionPolicy'] = 'Retain'
+  end
+end
+
+def make_double(name)
+  obj = double(name)
+  yield(obj)
+  return obj
+end