Expose available_disk in stager advertisement

cloudfoundry-attic · Apr 7, 2014 · 35769a6 · 35769a6
1 parent 4bd08bf
commit 35769a6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/dea/responders/staging_locator.rb b/lib/dea/responders/staging_locator.rb
@@ -33,7 +33,8 @@ def advertise
       nats.publish("staging.advertise", {
         "id" => dea_id,
         "stacks" => config["stacks"],
-        "available_memory" => resource_manager.remaining_memory
+        "available_memory" => resource_manager.remaining_memory,
+        "available_disk" => resource_manager.remaining_disk,
       })
     rescue => e
       logger.error("staging_locator.advertise", error: e, backtrace: e.backtrace)

diff --git a/spec/unit/responders/staging_locator_spec.rb b/spec/unit/responders/staging_locator_spec.rb
@@ -100,11 +100,13 @@
     it "publishes 'staging.advertise' message" do
       config["stacks"] = ["lucid64"]
       resource_manager.stub(:remaining_memory => 45678)
+      resource_manager.stub(:remaining_disk => 12345)
 
       nats_mock.should_receive(:publish).with("staging.advertise", JSON.dump(
         "id" => dea_id,
         "stacks" => ["lucid64"],
         "available_memory" => 45678,
+        "available_disk" => 12345,
       ))
       subject.advertise
     end