Merge pull request #2306 from herwinw/gc_enable_disable

include/natalie/gc_module.hpp

@@ -8,15 +8,8 @@ namespace Natalie {
 
 class GCModule : public Object {
 public:
-    static bool enable() {
-        Heap::the().gc_enable();
-        return true;
-    }
-
-    static bool disable() {
-        Heap::the().gc_disable();
-        return true;
-    }
+    static bool disable();
+    static bool enable();
 
     static Value start(Env *env) {
         Heap::the().collect();

spec/core/gc/disable_spec.rb

@@ -0,0 +1,18 @@
+require_relative '../../spec_helper'
+
+describe "GC.disable" do
+  after :each do
+    GC.enable
+  end
+
+  it "returns true if and only if the garbage collection was previously disabled" do
+    GC.enable
+    GC.disable.should == false
+    GC.disable.should == true
+    GC.disable.should == true
+    GC.enable
+    GC.disable.should == false
+    GC.disable.should == true
+  end
+
+end

spec/core/gc/enable_spec.rb

@@ -0,0 +1,13 @@
+require_relative '../../spec_helper'
+
+describe "GC.enable" do
+
+  it "returns true if and only if the garbage collection was already disabled" do
+    GC.enable
+    GC.enable.should == false
+    GC.disable
+    GC.enable.should == true
+    GC.enable.should == false
+  end
+
+end

src/gc_module.cpp

@@ -0,0 +1,19 @@
+#include "natalie/gc_module.hpp"
+
+namespace Natalie {
+
+bool GCModule::disable() {
+    if (!Heap::the().gc_enabled())
+        return true;
+    Heap::the().gc_disable();
+    return false;
+}
+
+bool GCModule::enable() {
+    if (Heap::the().gc_enabled())
+        return false;
+    Heap::the().gc_enable();
+    return true;
+}
+
+}