From 9ac2f5fd9b5664470b8e99f91332efb305b8550b Mon Sep 17 00:00:00 2001
From: Lars Bergstrom <lars@lars.com>
Date: Wed, 15 Jan 2014 15:54:47 -0600
Subject: [PATCH] Also handle the exit after load and output file cases for ref
 tests

---
 src/components/main/compositing/compositor.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/components/main/compositing/compositor.rs b/src/components/main/compositing/compositor.rs
index d65cab22c71..1d3af84ca8d 100644
--- a/src/components/main/compositing/compositor.rs
+++ b/src/components/main/compositing/compositor.rs
@@ -663,14 +663,16 @@ impl IOCompositor {
             let res = png::store_png(&img, &path);
             assert!(res.is_ok());
 
-            self.done = true;
+            debug!("shutting down the constellation after generating an output file");
+            self.constellation_chan.send(ExitMsg);
         }
 
         self.window.present();
 
         let exit = self.opts.exit_after_load;
         if exit {
-            self.done = true;
+            debug!("shutting down the constellation for exit_after_load");
+            self.constellation_chan.send(ExitMsg);
         }
     }