Heckle iasgn.

[git-p4: depot-paths = "//src/heckle/dev/": change = 3178]

History.txt

@@ -3,6 +3,7 @@
 * N major enhancements:
   * :call nodes are now heckled.
   * :lasgn nodes are now heckled.
+  * :iasgn nodes are now heckled.
 * N minor enhancements:
   * Added focus mode to feel the Eye of Sauron.
 

lib/heckle.rb

@@ -260,6 +260,26 @@ def process_defn(exp)
     reset_node_count
   end
 
+  def process_iasgn(exp)
+    mutate_node [:iasgn, exp.shift, process(exp.shift)]
+  end
+
+  ##
+  # Replaces the value of the iasgn with nil if its some value, and 42 if its
+  # nil.
+
+  def mutate_iasgn(node)
+    node.shift
+    var = node.shift
+    val = node.last
+
+    if val.first == :nil then
+      [:iasgn, var, [:lit, 42]]
+    else
+      [:iasgn, var, [:nil]]
+    end
+  end
+
   def process_lasgn(exp)
     mutate_node [:lasgn, exp.shift, process(exp.shift)]
   end

test/fixtures/heckled.rb

@@ -9,6 +9,11 @@ def uses_call
     some_func + some_other_func
   end
 
+  def uses_iasgn
+    @ivar = 5
+    @ivar = nil
+  end
+
   def uses_lasgn
     lvar = 5
     lvar = nil

test/test_heckle.rb

@@ -461,6 +461,41 @@ def test_heckle_class_methods
   end
 end
 
+class TestHeckleIasgn < HeckleTestCase
+
+  def setup
+    @nodes = [:iasgn]
+    super
+  end
+
+  def test_iasgn_val
+    expected = [:defn, :uses_iasgn,
+                [:scope,
+                 [:block,
+                  [:args],
+                  [:iasgn, :@ivar, [:nil]],
+                  [:iasgn, :@ivar, [:nil]]]]]
+
+    @heckler.process(@heckler.current_tree)
+    assert_equal expected, @heckler.current_tree
+  end
+
+  def test_iasgn_nil
+    expected = [:defn, :uses_iasgn,
+                [:scope,
+                 [:block,
+                  [:args],
+                  [:iasgn, :@ivar, [:lit, 5]],
+                  [:iasgn, :@ivar, [:lit, 42]]]]]
+
+    @heckler.process(@heckler.current_tree)
+    @heckler.reset_tree
+    @heckler.process(@heckler.current_tree)
+    assert_equal expected, @heckler.current_tree
+  end
+
+end
+
 class TestHeckleLasgn < HeckleTestCase
 
   def setup