test/test_SimpleQueryExpander.rb in taskjuggler-3.1.0 vs test/test_SimpleQueryExpander.rb in taskjuggler-3.2.0

@@ -18,39 +18,66 @@
 require 'taskjuggler/SimpleQueryExpander'
 require 'taskjuggler/MessageHandler'
 
 class TestSimpleQueryExpander < Test::Unit::TestCase
 
+  class Scenario
+
+    def id
+      'scId'
+    end
+
+  end
+
+  class Project
+
+    def initialize
+    end
+
+    def scenario(foo)
+      Scenario.new
+    end
+
+  end
+
   class Query
 
     def initialize
     end
 
     def process
     end
 
+    def project
+      Project.new
+    end
+
+    def scenarioIdx
+      0
+    end
+
     def attributeId=(value)
     end
 
     def ok
       true
     end
 
     def to_s
       'XXX'
     end
+
   end
 
   def setup
   end
 
   def teardown
   end
 
   def test_expand
     exp = TaskJuggler::SimpleQueryExpander.new('foo <-bar-> foo',
-                                               Query.new,
-                                               nil, nil)
+                                               Query.new, nil)
     assert_equal('foo XXX foo', exp.expand)
   end
 
 end