lib/rollo/commands/host_cluster.rb in rollo-0.1.0 vs lib/rollo/commands/host_cluster.rb in rollo-0.2.0

@@ -21,11 +21,11 @@
           region, asg_name, _,
               host_cluster = nil)
         batch_size = options[:batch_size]
 
         host_cluster = host_cluster ||
-            Rollo::HostCluster.new(asg_name, region)
+            Rollo::Model::HostCluster.new(asg_name, region)
 
         say("Increasing host cluster desired capacity by #{batch_size}...")
         with_padding do
           host_cluster.increase_capacity_by(batch_size) do |on|
             on.prepare do |current, target|
@@ -64,13 +64,13 @@
           region, asg_name, ecs_cluster_name,
               host_cluster = nil, service_cluster = nil)
         batch_size = options[:batch_size]
 
         host_cluster = host_cluster ||
-            Rollo::HostCluster.new(asg_name, region)
+            Rollo::Model::HostCluster.new(asg_name, region)
         service_cluster = service_cluster ||
-            Rollo::ServiceCluster.new(ecs_cluster_name, region)
+            Rollo::Model::ServiceCluster.new(ecs_cluster_name, region)
 
         say("Decreasing host cluster desired capacity by #{batch_size}...")
         with_padding do
           host_cluster.decrease_capacity_by(batch_size) do |on|
             on.prepare do |current, target|
@@ -131,12 +131,12 @@
 
         service_start_wait_minutes = options[:startup_time]
         service_start_wait_seconds = 60 * service_start_wait_minutes
 
         host_cluster = host_cluster ||
-            Rollo::HostCluster.new(asg_name, region)
+            Rollo::Model::HostCluster.new(asg_name, region)
         service_cluster = service_cluster ||
-            Rollo::ServiceCluster.new(ecs_cluster_name, region)
+            Rollo::Model::ServiceCluster.new(ecs_cluster_name, region)
 
         hosts = host_cluster.hosts.select {|h| instance_ids.include?(h.id) }
         host_batches = hosts.each_slice(batch_size).to_a
 
         say(