zaws 0.0.5 → 0.1.1

data/spec/zaws/command/ec2/compute/images_spec.rb

@@ -0,0 +1,69 @@
+require 'spec_helper'
+
+describe ZAWS::Services::EC2::Compute do
+
+  before(:each) {
+
+    @var_security_group_id="sg-abcd1234"
+    @var_output_json="json"
+    @var_output_table="table"
+    @var_region="us-west-1"
+    @var_vpc_id="my_vpc_id"
+    @var_sec_group_name="my_security_group_name"
+
+    options_json = {:region => @var_region,
+                    :verbose => false,
+                    :check => false,
+                    :undofile => false,
+                    :viewtype => 'json',
+                    :owner => 'me'
+    }
+
+    options_table = {:region => @var_region,
+                     :verbose => false,
+                     :check => false,
+                     :undofile => false,
+                     :viewtype => 'table',
+                    :owner => 'self'
+    }
+
+
+    @textout=double('outout')
+    @shellout=double('ZAWS::Helper::Shell')
+    @undofile=double('ZAWS::Helper::ZFile')
+    @aws=ZAWS::AWS.new(@shellout, ZAWS::AWSCLI.new(@shellout, true), @undofile)
+    @command_image = ZAWS::Command::Compute.new([], options_table, {})
+    @command_image.aws=@aws
+    @command_image.out=@textout
+    @command_image.print_exit_code = true
+    @command_image_json = ZAWS::Command::Compute.new([], options_json, {})
+    @command_image_json.aws=@aws
+    @command_image_json.out=@textout
+    @command_image_json.print_exit_code = true
+
+  }
+
+  describe "#view" do
+
+    it "Get compute images in a human readable table." do
+      desc_image = ZAWS::External::AWSCLI::Commands::EC2::DescribeImages.new
+      desc_image.owner("self")
+      desc_image.aws.output(@var_output_table).region(@var_region)
+      expect(@shellout).to receive(:cli).with(desc_image.aws.get_command, nil).ordered.and_return('test output')
+      expect(@textout).to receive(:puts).with('test output').ordered
+      @command_image.view_images
+    end
+
+    it "Get compute images in JSON form for me specifically" do
+      desc_image = ZAWS::External::AWSCLI::Commands::EC2::DescribeImages.new
+      desc_image.owner("me")
+      desc_image.aws.output(@var_output_json).region(@var_region)
+      expect(@shellout).to receive(:cli).with(desc_image.aws.get_command, nil).ordered.and_return('test output')
+      expect(@textout).to receive(:puts).with('test output').ordered
+      @command_image_json.view_images
+    end
+
+  end
+end
+
+

data/spec/zaws/command/ec2/compute/management_spec.rb

@@ -0,0 +1,364 @@
+require 'spec_helper'
+
+describe ZAWS::Services::EC2::Compute do
+
+  let(:vap_region) { "us-west-1" }
+  let(:vap_role) { "my_role" }
+  let(:vap_policy) { "my_policy" }
+
+  let(:vap_my_instance_id1) { "i-abcdefg1" }
+  let(:vap_my_instance_name1) { "my-name1" }
+  let(:vap_my_instance_id2) { "i-abcdefg2" }
+  let(:vap_my_instance_name2) { "my-name2" }
+  let(:vap_my_instance_id3) { "i-abcdefg3" }
+  let(:vap_my_instance_name3) { "my-name3" }
+
+  let(:vap_policy_arn) { "arn:aws:iam::123456789abc:policy/#{vap_policy}" }
+  let(:vap_policy_version) { "v2" }
+
+  let(:var_policy_doc) { <<-eos
+    {
+		"Version": "2012-10-17",
+		"Statement": [
+			{
+				"Action": [
+					"ec2:StartInstances",
+					"ec2:StopInstances",
+					"ec2:CreateTags"
+				],
+				"Resource": [
+					"arn:aws:ec2:us-east-1:123456789abc:instance/#{vap_my_instance_id1}",
+					"arn:aws:ec2:us-east-1:123456789abc:instance/#{vap_my_instance_id2}"
+				],
+				"Effect": "Allow"
+			},
+			{
+				"Action": [
+					"iam:GetPolicy",
+					"iam:GetPolicyVersion"
+				],
+				"Resource": [
+					"*"
+				],
+				"Effect": "Allow"
+			}
+		]
+	}
+  eos
+  }
+
+  let(:var_policy_version_doc) { <<-eos
+  {
+    "PolicyVersion": {
+        "CreateDate": "2015-09-08T06:14:59Z",
+        "VersionId": "v2",
+        "Document": #{var_policy_doc},
+        "IsDefaultVersion": true
+    }
+  }
+  eos
+  }
+
+  let(:var_policy_meta_data) { <<-eos
+	{
+		"Policy": {
+			"PolicyName": "#{vap_policy}",
+			"CreateDate": "2015-09-08T05:21:54Z",
+			"AttachmentCount": 1,
+			"IsAttachable": true,
+			"PolicyId": "123456789012345678901",
+			"DefaultVersionId": "#{vap_policy_version}",
+			"Path": "/",
+			"Arn": "#{vap_policy_arn}",
+			"UpdateDate": "2015-09-08T06:14:59Z"
+		}
+	} 
+  eos
+  }
+
+
+  let(:options) { {:region => vap_region, :viewtype => 'json'} }
+  let(:options_policy_arn) { {:region => vap_region,
+                              :viewtype => 'json',
+                              :policy_arn => vap_policy_arn,
+                              :verbose => nil} }
+
+  let(:options_start_name) { {:region => vap_region,
+                              :name => vap_my_instance_name1,
+                              :verbose => nil,
+                              :skipruncheck => true} }
+
+  let(:options_stop_name) { {:region => vap_region,
+                             :name => vap_my_instance_name2,
+                             :verbose => nil,
+                             :skipruncheck => 0} }
+
+  let(:options_set_interval) { {:region => vap_region,
+                                :name => vap_my_instance_name2,
+                                :viewtype => 'json',
+                                :policy_arn => vap_policy_arn,
+                                :verbose => nil,
+                                :overridebasetime => 0} }
+
+  let(:options_interval_cron) { {:region => vap_region,
+                                 :viewtype => 'json',
+                                 :policy_arn => vap_policy_arn,
+                                 :verbose => nil,
+                                 :overridebasetime => 3601} }
+
+  let(:var_policy) { <<-eos
+	  {
+		  "RoleName": "testStartStop",
+		  "PolicyDocument": {
+			  "Version": "2012-10-17",
+			  "Statement": [
+				  {
+					  "Action": [
+						  "ec2:StartInstances",
+						  "ec2:StopInstances"
+					  ],
+					  "Resource": [
+						  "arn:aws:ec2:us-east-1:123456789012:instance/#{vap_my_instance_id1}",
+					      "arn:aws:ec2:us-east-1:123456789012:instance/#{vap_my_instance_id2}"
+					  ],
+					  "Effect": "Allow"
+				  }
+			  ]
+		  },
+		  "PolicyName": "testStopStart"
+	  }
+  eos
+  }
+
+  let(:var_describe_instance) { <<-eos
+		{
+			"Reservations": [
+				{
+					"OwnerId": "123456789012",
+					"ReservationId": "r-88ef5d66",
+					"Groups": [],
+					"Instances": [
+						{
+							"State": {
+								"Code": 80,
+								"Name": "stopped"
+							},
+							"InstanceId": "#{vap_my_instance_id1}",
+							"Tags": [
+								{
+									"Value": "#{vap_my_instance_name1}",
+									"Key": "Name"
+								},
+								{
+									"Value": "0:7200:test@test.com",
+									"Key": "interval"
+								}
+
+							],
+							"AmiLaunchIndex": 0
+						}
+					]
+				},
+				{
+					"OwnerId": "123456789012",
+					"ReservationId": "r-88ef5d66",
+					"Groups": [],
+					"Instances": [
+						{
+							"State": {
+								"Code": 80,
+								"Name": "running"
+							},
+							"InstanceId": "#{vap_my_instance_id2}",
+							"Tags": [
+								{
+									"Value": "#{vap_my_instance_name2}",
+									"Key": "Name"
+								},
+								{
+									"Value": "0:3600:test@test.com",
+									"Key": "interval"
+								}
+							],
+							"AmiLaunchIndex": 0
+						}
+					]
+				},
+				{
+					"OwnerId": "123456789012",
+					"ReservationId": "r-88ef5d66",
+					"Groups": [],
+					"Instances": [
+						{
+							"State": {
+								"Code": 80,
+								"Name": "stopped"
+							},
+							"InstanceId": "#{vap_my_instance_id3}",
+							"Tags": [
+								{
+									"Value": "#{vap_my_instance_name3}",
+									"Key": "Name"
+								}
+							],
+							"AmiLaunchIndex": 0
+						}
+					]
+				}
+
+
+			]
+		}	
+  eos
+  }
+
+  let(:vap_list_instance_ids) { "i-abcdefg1\ni-abcdefg2" }
+  let(:var_list_instance_names) { "my-name1\nmy-name2" }
+  let(:interval_val) { "0:3600:test@test.com" }
+
+  before(:each) {
+    @textout=double('outout')
+    @shellout=double('ZAWS::Helper::Shell')
+    @command_compute = ZAWS::Command::Compute.new([], options, {});
+    @aws=ZAWS::AWS.new(@shellout, ZAWS::AWSCLI.new(@shellout,true))
+    @command_compute.aws=@aws
+    @command_compute.out=@textout
+    @command_compute.print_exit_code = true
+  }
+
+  describe "#instance_id_by_external_id" do
+    it "provides an instance id when you give it a external id" do
+
+      tags = ZAWS::External::AWSCLI::Generators::Result::EC2::Tags.new
+      tags = tags.add("externalid", "my_instance")
+      instances = ZAWS::External::AWSCLI::Generators::Result::EC2::Instances.new
+      instances = instances.instance_id(0, "i-XXXXXXX")
+      instances = instances.tags(0, tags)
+      compute_instances=instances.get_json
+
+      desc_instances = ZAWS::External::AWSCLI::Commands::EC2::DescribeInstances.new
+      desc_instances.filter.vpc_id("my_vpc_id").tags(tags)
+      desc_instances.aws.output("json").region("us-west-1")
+
+
+      expect(@shellout).to receive(:cli).with(desc_instances.aws.get_command, nil).and_return(compute_instances)
+      instanceid = @aws.ec2.compute.instance_id_by_external_id('us-west-1', 'my_instance', 'my_vpc_id', nil, nil)
+      expect(instanceid).to eq("i-XXXXXXX")
+    end
+  end
+
+  describe "#tag_resource" do
+    it "tags an instance when created" do
+      tag_created = <<-eos
+		{ "return":"true" }
+      eos
+
+      tags = ZAWS::External::AWSCLI::Generators::Result::EC2::Tags.new
+      tags = tags.add("externalid", "extername")
+
+      create_tags = ZAWS::External::AWSCLI::Commands::EC2::CreateTags.new
+      aws_command = ZAWS::External::AWSCLI::Commands::AWS.new
+      create_tags = create_tags.resource("id-X").tags(tags)
+      aws_command = aws_command.output("json").region("us-west-1").subcommand(create_tags)
+
+      expect(@shellout).to receive(:cli).with(aws_command.get_command, nil).and_return(tag_created)
+
+      tags = ZAWS::External::AWSCLI::Generators::Result::EC2::Tags.new
+      tags = tags.add("Name", "extername")
+
+      create_tags = ZAWS::External::AWSCLI::Commands::EC2::CreateTags.new
+      aws_command = ZAWS::External::AWSCLI::Commands::AWS.new
+      create_tags = create_tags.resource("id-X").tags(tags)
+      aws_command = aws_command.output("json").region("us-west-1").subcommand(create_tags)
+
+      expect(@shellout).to receive(:cli).with(aws_command.get_command, nil).and_return(tag_created)
+
+      @aws.ec2.compute.tag_resource('us-west-1', 'id-X', 'extername')
+    end
+
+  end
+
+  describe "#interval_eligible" do
+    context "role policy exists" do
+      it "lists the names of the instances" do
+        @command_compute = ZAWS::Command::Compute.new([], options_policy_arn, {});
+        @command_compute.aws=@aws
+        @command_compute.out=@textout
+        @command_compute.print_exit_code = true
+        expect(@shellout).to receive(:cli).with("aws --output json iam get-policy --policy-arn #{vap_policy_arn}", nil).ordered.and_return(var_policy_meta_data)
+        expect(@shellout).to receive(:cli).with("aws --output json iam get-policy-version --policy-arn #{vap_policy_arn} --version-id #{vap_policy_version}", nil).ordered.and_return(var_policy_version_doc)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 describe-instances", nil).and_return(var_describe_instance)
+        expect(@textout).to receive(:puts).with("my-name1\nmy-name2")
+        @command_compute.interval_eligible()
+      end
+    end
+  end
+
+  describe "#start" do
+    context "the instance is stopped, the name as been provided and the name exists" do
+      it "will start the instance" do
+        @command_compute = ZAWS::Command::Compute.new([], options_start_name, {});
+        @command_compute.aws=@aws
+        @command_compute.out=@textout
+        @command_compute.print_exit_code = true
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 describe-instances", nil).and_return(var_describe_instance)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 start-instances --instance-ids #{vap_my_instance_id1}", nil).and_return(var_describe_instance)
+        expect(@textout).to receive(:puts).with("Instance #{vap_my_instance_name1} started.")
+        @command_compute.start()
+      end
+    end
+  end
+
+  describe "#stop" do
+    context "the instance is running, the name as been provided and the name exists" do
+      it "will stop the instance" do
+        @command_compute = ZAWS::Command::Compute.new([], options_stop_name, {});
+        @command_compute.aws=@aws
+        @command_compute.out=@textout
+        @command_compute.print_exit_code = true
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 describe-instances", nil).and_return(var_describe_instance)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 stop-instances --instance-ids #{vap_my_instance_id2}", nil).and_return(var_describe_instance)
+        expect(@textout).to receive(:puts).with("Instance #{vap_my_instance_name2} stopped.")
+        @command_compute.stop()
+      end
+    end
+  end
+
+  describe "#set_interval" do
+    context "the instance exists" do
+      it "will tag instance with interval" do
+        @command_compute = ZAWS::Command::Compute.new([], options_set_interval, {});
+        @command_compute.aws=@aws
+        @command_compute.out=@textout
+        @command_compute.print_exit_code = true
+        expect(@shellout).to receive(:cli).with("aws --output json iam get-policy --policy-arn #{vap_policy_arn}", nil).ordered.and_return(var_policy_meta_data)
+        expect(@shellout).to receive(:cli).with("aws --output json iam get-policy-version --policy-arn #{vap_policy_arn} --version-id #{vap_policy_version}", nil).ordered.and_return(var_policy_version_doc)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 describe-instances", nil).and_return(var_describe_instance)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 create-tags --resources #{vap_my_instance_id2} --tags \"Key=interval,Value=#{interval_val}\"", nil).and_return(var_describe_instance)
+        expect(@textout).to receive(:puts).with("Instance #{vap_my_instance_name2} tagged: Key=interval,Value=#{interval_val}")
+        @command_compute.set_interval(1, "test@test.com")
+      end
+    end
+  end
+
+  describe "#interval_cron" do
+    context "one instance set to start that is stopped, one vice versa, one other" do
+      it "will start one, stop one, and nothing one" do
+        @command_compute = ZAWS::Command::Compute.new([], options_interval_cron, {});
+        @command_compute.aws=@aws
+        @command_compute.out=@textout
+        @command_compute.print_exit_code = true
+        expect(@shellout).to receive(:cli).with("aws --output json iam get-policy --policy-arn #{vap_policy_arn}", nil).ordered.and_return(var_policy_meta_data)
+        expect(@shellout).to receive(:cli).with("aws --output json iam get-policy-version --policy-arn #{vap_policy_arn} --version-id #{vap_policy_version}", nil).ordered.and_return(var_policy_version_doc)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 describe-instances", nil).and_return(var_describe_instance)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 start-instances --instance-ids #{vap_my_instance_id1}", nil).and_return(var_describe_instance)
+        expect(@shellout).to receive(:cli).with("aws --output json --region #{vap_region} ec2 stop-instances --instance-ids #{vap_my_instance_id2}", nil).and_return(var_describe_instance)
+        expect(@textout).to receive(:puts).with("Instance #{vap_my_instance_name1} started.")
+        expect(@textout).to receive(:puts).with("Instance #{vap_my_instance_name2} stopped.")
+        @command_compute.interval_cron()
+      end
+    end
+  end
+
+
+end

data/spec/zaws/command/ec2/compute/network_spec.rb

@@ -0,0 +1,93 @@
+require 'spec_helper'
+
+describe ZAWS::Services::EC2::Compute do
+
+  before(:each) {
+    @textout=double('outout')
+    @shellout=double('ZAWS::Helper::Shell')
+    @aws=ZAWS::AWS.new(@shellout, ZAWS::AWSCLI.new(@shellout, true))
+  }
+  describe "#instance_ping?" do
+    it "determines an instance is reachable over the network with ping" do
+
+      comline ='ping -q -c 2 0.0.0.0'
+      times_called = 0
+      @shellout.stub(:cli).with(comline, nil).and_return do
+        times_called += 1
+        raise Mixlib::ShellOut::ShellCommandFailed if times_called == 2
+      end
+      @aws.ec2.compute.instance_ping?('0.0.0.0', 10, 1)
+
+    end
+
+    it "determines an instance is not reachable over the network with ping" do
+
+      comline ='ping -q -c 2 0.0.0.0'
+      times_called = 0
+      @shellout.stub(:cli).with(comline, nil).and_return do
+        times_called += 1
+        raise Mixlib::ShellOut::ShellCommandFailed if times_called < 4
+      end
+      expect { @aws.ec2.compute.instance_ping?('0.0.0.0', 2, 1)
+             }.to raise_error(StandardError, 'Timeout before instance responded to ping.')
+    end
+  end
+
+  describe "#nosdcheck" do
+    it "sets no source/destination check for instances intended to be NAT instances" do
+      nosd_check_result = '{ "return":"true" }'
+
+      mod_inst_attribute = ZAWS::External::AWSCLI::Commands::EC2::ModifyInstanceAttribute.new
+      aws_command = ZAWS::External::AWSCLI::Commands::AWS.new
+      mod_inst_attribute = mod_inst_attribute.instance_id("id-X").no_source_dest_check
+      aws_command = aws_command.output("json").region("us-west-1").subcommand(mod_inst_attribute)
+
+      expect(@shellout).to receive(:cli).with(aws_command.get_command, nil).and_return(nosd_check_result)
+      @aws.ec2.compute.nosdcheck('us-west-1', 'id-X')
+    end
+  end
+
+  describe "#network_interface_json" do
+    it "provides a network interface structure" do
+
+      subnets = ZAWS::External::AWSCLI::Generators::Result::EC2::Subnets.new
+      count=0
+      [["10.0.1.0/24", "subnet-XXXXXX"], ["10.0.0.0/24", "subnet-YYYYYY"]].each do |x|
+        subnets = subnets.vpc_id(count, "vpc-XXXXXX").cidr_block(count, x[0]).map_public_ip_on_launch(count, false)
+        subnets = subnets.default_for_az(count, false).state(count, "available").subnet_id(count, x[1])
+        subnets = subnets.available_ip_address_count(count, 251)
+        count+=1
+      end
+      subnets= subnets.get_json
+
+      desc_subnets = ZAWS::External::AWSCLI::Commands::EC2::DescribeSubnets.new
+      aws_command = ZAWS::External::AWSCLI::Commands::AWS.new
+      desc_subnets.filter.vpc_id("my_vpc_id")
+      desc_subnets.aws.output("json").region("us-west-1").subcommand(desc_subnets)
+
+      expect(@shellout).to receive(:cli).with(desc_subnets.aws.get_command, nil).and_return(subnets)
+
+      security_groups = ZAWS::External::AWSCLI::Generators::Result::EC2::SecurityGroups.new
+      security_groups = security_groups.description(0, "My security group").group_name(0, "my_security_group_name")
+      security_groups = security_groups.owner_id(0, "123456789012").group_id(0, "sg-903004f8")
+      sgroups = security_groups.get_json
+
+      desc_sec_grps = ZAWS::External::AWSCLI::Commands::EC2::DescribeSecurityGroups.new
+      desc_sec_grps.filter.vpc_id("my_vpc_id").group_name("my_security_group_name")
+      desc_sec_grps.aws.output("json").region("us-west-1")
+
+      expect(@shellout).to receive(:cli).with(desc_sec_grps.aws.get_command, nil).and_return(sgroups)
+
+      network_interfaces=ZAWS::External::AWSCLI::Commands::EC2::NetworkInterfaces.new
+      network_interfaces=network_interfaces.add_group(0, "sg-903004f8").private_ip_address(0, "10.0.0.6")
+      network_interfaces=network_interfaces.device_index(0, 0).subnet_id(0, "subnet-YYYYYY")
+
+      bdm = @aws.ec2.compute.network_interface_json('us-west-1', nil, 'my_vpc_id', '10.0.0.6', 'my_security_group_name')
+      expect(bdm).to eq(network_interfaces.get_network_interfaces_array_to_json)
+    end
+
+  end
+end
+
+
+