spec/deployment_file_spec.rb in bosh-cloudfoundry-0.7.0.alpha.10 vs spec/deployment_file_spec.rb in bosh-cloudfoundry-0.7.0.alpha.11

@@ -37,46 +37,47 @@
 
   # aws & openstack
   # v132 & v133
   # medium & large
 
-  context "generates deployment (aws)" do
-    let(:bosh_cpi) { "aws" }
-    let(:bosh_status) { {"cpi" => bosh_cpi, "uuid" => "UUID"} }
-    let(:attributes) { {
-      name: "demo",
-      dns: "mycloud.com",
-      ip_addresses: ['1.2.3.4'],
-      common_password: "qwertyasdfgh",
-      deployment_size: "medium"
-    } }
-    let(:release_version_cpi) { Bosh::Cloudfoundry::ReleaseVersionCpi.for_cpi(133, bosh_cpi) }
-    let(:release_version_cpi_medium) { Bosh::Cloudfoundry::ReleaseVersionCpiSize.new(release_version_cpi, "medium") }
-    let(:deployment_attributes) do
-      Bosh::Cloudfoundry::DeploymentAttributes.new(mock("director"), bosh_status, release_version_cpi_medium, attributes)
-    end
+  %w[medium large].each do |deployment_size|
+    context "generates deployment (aws)" do
+      let(:bosh_cpi) { "aws" }
+      let(:bosh_status) { {"cpi" => bosh_cpi, "uuid" => "UUID"} }
+      let(:attributes) { {
+        name: "demo",
+        dns: "mycloud.com",
+        ip_addresses: ['1.2.3.4'],
+        common_password: "qwertyasdfgh",
+        deployment_size: deployment_size
+      } }
+      let(:release_version_cpi) { Bosh::Cloudfoundry::ReleaseVersionCpi.for_cpi(133, bosh_cpi) }
+      let(:release_version_cpi_size) { Bosh::Cloudfoundry::ReleaseVersionCpiSize.new(release_version_cpi, deployment_size) }
+      let(:deployment_attributes) do
+        Bosh::Cloudfoundry::DeploymentAttributes.new(mock("director"), bosh_status, release_version_cpi_size, attributes)
+      end
 
-    subject { Bosh::Cloudfoundry::DeploymentFile.new(release_version_cpi_medium, deployment_attributes, bosh_status) }
+      subject { Bosh::Cloudfoundry::DeploymentFile.new(release_version_cpi_size, deployment_attributes, bosh_status) }
 
-    before do
-      subject.biff.stub(:deployment).and_return(home_file("deployments/cf/demo.yml"))
-      deployment_cmd = mock("deployment_cmd")
-      deployment_cmd.stub(:set_current).with(home_file("deployments/cf/demo.yml"))
-      deployment_cmd.stub(:perform)
-      subject.stub(:deployment_cmd).and_return(deployment_cmd)
-    end
+      before do
+        subject.biff.stub(:deployment).and_return(home_file("deployments/cf/demo.yml"))
+        deployment_cmd = mock("deployment_cmd")
+        deployment_cmd.stub(:set_current).with(home_file("deployments/cf/demo.yml"))
+        deployment_cmd.stub(:perform)
+        subject.stub(:deployment_cmd).and_return(deployment_cmd)
+      end
 
-    it "medium size" do
-      in_home_dir do
-        subject.prepare_environment
+      it "#{deployment_size} size" do
+        in_home_dir do
+          subject.prepare_environment
 
-        subject.create_deployment_file
-        files_match(spec_asset("v133/aws/medium.yml"), subject.deployment_file)
+          subject.create_deployment_file
+          files_match(spec_asset("v133/aws/#{deployment_size}.yml"), subject.deployment_file)
 
-        subject.deploy(non_interactive: true)
+          subject.deploy(non_interactive: true)
+        end
       end
     end
-
     # it "large size" do
     #   in_home_dir do
     #     command.add_option(:deployment_size, "large")
     # 
     #     command.create_cf