fix test with mocks

nkallen · Sep 8, 2010 · a18f15a · a18f15a
1 parent 386a150
commit a18f15a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/scala/com/twitter/querulous/unit/MemoizingDatabaseFactorySpec.scala b/src/test/scala/com/twitter/querulous/unit/MemoizingDatabaseFactorySpec.scala
@@ -18,8 +18,8 @@ class MemoizingDatabaseFactorySpec extends Specification with JMocker {
       val memoizingDatabase = new MemoizingDatabaseFactory(databaseFactory)
 
       expect {
-        one(databaseFactory).apply(hosts, "bar", username, password) willReturn database1
-        one(databaseFactory).apply(hosts, "baz", username, password) willReturn database2
+        one(databaseFactory).apply(hosts, "bar", username, password, null) willReturn database1
+        one(databaseFactory).apply(hosts, "baz", username, password, null) willReturn database2
       }
       memoizingDatabase(hosts, "bar", username, password) mustBe database1
       memoizingDatabase(hosts, "bar", username, password) mustBe database1