[STABLE] better tests

lib/celery_script/tree_climber.rb

@@ -11,6 +11,13 @@ module CeleryScript
       results
     end
 
+    def self.find_all_with_arg(node, arg_name)
+      results = []
+      filter = -> (node) { results.push(node) if node.args.has_key?(arg_name) }
+      travel(node, filter)
+      results
+    end
+
   private
 
     def self.visit_node(node, callable)

spec/lib/celery_script/tree_climber_spec.rb

@@ -23,4 +23,10 @@ describe CeleryScript::TreeClimber do
     expect(query.length).to eq(1)
     expect(query.first.kind).to eq("var_get")
   end
+
+  it "searches the tree for a specific `arg`" do
+    query = CeleryScript::TreeClimber.find_all_with_arg(node, :data_value)
+    expect(query.length).to eq(3)
+    expect(query.map(&:kind).sort).to eq(["blah", "other", "whatever"])
+  end
 end

spec/mutations/sequences/improved_ast_fixture.json

@@ -26,7 +26,7 @@
             "kind": "whatever",
             "args": {
                 "data_type": "integer",
-                "data_value": 1000
+                "data_value": 2000
             }
         }
     ]