Changed the syntax some. All the handling of variable argument lists is

tests/lib/unittest/unittest_test.dart

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 // TODO(gram):
 // Unfortunately I can't seem to test anything that involves timeouts, e.g.
 // insufficient callbacks, because the timeout is controlled externally
 // (test.dart?), and we would need to use a shorter timeout for the inner tests
 // so the outer timeout doesn't fire. So I removed all such tests.
 // I'd like to revisit this at some point.
@@ skipping 54 matching lines @@
   TestConfiguration(this._port);
 
   void onDone(int passed, int failed, int errors, List<TestCase> results,
       String uncaughtError) {
     var result = buildStatusString(passed, failed, errors, results, count,
         setup, teardown, uncaughtError);
     _port.send(result);
   }
 }
 
+class MockList extends Mock implements List {
+}
+
+class Foo {
+  sum(a, b, c) => a + b + c;
+}
+
+class FooSpy extends Mock implements Foo {
+  FooSpy real;
+  FooSpy() {
+    real = new Foo();
+    this.when(callsTo('sum')).alwaysCall(real.sum);
+  }
+}
+
 runTest() {
   port.receive((testName, sendport) {
     configure(_testconfig = new TestConfiguration(sendport));
     if (testName == 'single correct test') {
       test(testName, () => expect(2 + 3, equals(5)));
     } else if (testName == 'single failing test') {
       test(testName, () => expect(2 + 2, equals(5)));
     } else if (testName == 'exception test') {
       test(testName, () { throw new Exception('fail'); });
     } else if (testName == 'group name test') {
@@ skipping 32 matching lines @@
       test(testName, () {
              var _callback;
              _callback = expectAsyncUntil0(() {
                if (++_testconfig.count < 10) {
                  _defer(_callback);
                }
              },
              () => (_testconfig.count == 10));
              _defer(_callback);
       });
+    } else if (testName == 'mock test 1 (Mock)') {
+      test(testName, () {
+        var m = new Mock();
+
+        m.when(callsTo('foo', 1, 2)).thenReturn('A').thenReturn('B');
+        m.when(callsTo('foo', 1, 1)).thenReturn('C');
+        m.when(callsTo('foo', 9, anything)).thenReturn('D');
+        m.when(callsTo('bar', anything, anything)).thenReturn('E');
+        m.when(callsTo('foobar')).thenReturn('F');
+
+        var s = '${m.foo(1,2)}${m.foo(1,1)}${m.foo(9,10)}'
+            '${m.bar(1,1)}${m.foo(1,2)}';
+        getLogs(m, callsTo('foo', anything, anything)).verify(calledExactly(4));
+        getLogs(m, callsTo('foo', 1, anything)).verify(calledExactly(3));
+        getLogs(m, callsTo('foo', 9, anything)).verify(calledOnce);
+        getLogs(m, callsTo('foo', anything, 2)).verify(calledExactly(2));
+        getLogs(m, callsTo('foobar')).verify(neverCalled);
+        getLogs(m, callsTo('foo', 10, anything)).verify(neverCalled);
+        expect(s, 'ACDEB');
+      });
+    } else if (testName == 'mock test 2 (MockList)') {
+      test(testName, () {
+        var l = new MockList();
+        l.when(callsTo('length')).thenReturn(1);
+        l.when(callsTo('add', anything)).alwaysReturn(0);
+        l.add('foo');
+        expect(l.length(), 1);
+
+        var m = new MockList();
+        m.when(callsTo('add', anything)).alwaysReturn(0);
+
+        m.add('foo');
+        m.add('bar');
+
+        getLogs(m, callsTo('add')).verify(calledExactly(2));
+        getLogs(m, callsTo('add', 'foo')).verify(calledOnce);
+      });
+    } else if (testName == 'mock test 3 (Spy)') {
+      test(testName, () {
+        var p = new FooSpy();
+        p.sum(1, 2, 3);
+        getLogs(p, callsTo('sum')).verify(calledOnce);
+        p.sum(2, 2, 2);
+        getLogs(p, callsTo('sum')).verify(calledExactly(2));
+        getLogs(p, callsTo('sum')).verify(sometimeReturned(6));
+        getLogs(p, callsTo('sum')).verify(alwaysReturned(6));
+        getLogs(p, callsTo('sum')).verify(neverReturned(5));
+        p.sum(2, 2, 1);
+        getLogs(p, callsTo('sum')).verify(sometimeReturned(5));
+      });
+    } else if (testName == 'mock test 4 (Excess calls)') {
+      test(testName, () {
+        var m = new Mock();
+        m.when(callsTo('foo')).alwaysReturn(null);
+        m.foo();
+        m.foo();
+        getLogs(m, callsTo('foo')).verify(calledOnce);
+      });
+    } else if (testName == 'mock test 5 (No behavior)') {
+      test(testName, () {
+        var m = new Mock();
+        m.when(callsTo('foo')).thenReturn(null);
+        m.foo();
+        m.foo();
+      });
+    } else if (testName == 'mock test 6 (No matching return)') {
+      test(testName, () {
+        var p = new FooSpy();
+        p.sum(1, 2, 3);
+        getLogs(p, callsTo('sum')).verify(sometimeReturned(0));
+      });
     }
   });
 }
 
 void nextTest(int testNum) {
   SendPort sport = spawnFunction(runTest);
   sport.call(tests[testNum]).then((msg) {
     actual.add(msg);
     if (actual.length == expected.length) {
       for (var i = 0; i < tests.length; i++) {
         test(tests[i], () => expect(actual[i], equals(expected[i])));
       }
     } else {
       nextTest(testNum+1);
     }
   });
 }
 
 main() {
   tests = [
     'single correct test',
     'single failing test',
     'exception test',
     'group name test',
     'setup test',
     'teardown test',
     'setup and teardown test',
     'correct callback test',
     'excess callback test',
-    'completion test'
+    'completion test',
+    'mock test 1 (Mock)',
+    'mock test 2 (MockList)',
+    'mock test 3 (Spy)',
+    'mock test 4 (Excess calls)',
+    'mock test 5 (No behavior)',
+    'mock test 6 (No matching return)'
   ];
 
   expected = [
     buildStatusString(1, 0, 0, tests[0]),
     buildStatusString(0, 1, 0, tests[1], message: 'Expected: <5> but: was <4>'),
     buildStatusString(0, 1, 0, tests[2], message: 'Caught Exception: fail'),
     buildStatusString(2, 0, 0, 'a a::a b b'),
     buildStatusString(1, 0, 0, 'a ${tests[4]}', 0, 'setup'),
     buildStatusString(1, 0, 0, 'a ${tests[5]}', 0, '', 'teardown'),
     buildStatusString(1, 0, 0, 'a ${tests[6]}', 0,
         'setup', 'teardown'),
     buildStatusString(1, 0, 0, tests[7], 1),
     buildStatusString(0, 0, 1, tests[8], 1,
         message: 'Callback called more times than expected (2 > 1)'),
-    buildStatusString(1, 0, 0, tests[9], 10)
+    buildStatusString(1, 0, 0, tests[9], 10),
+    buildStatusString(1, 0, 0, tests[10]),
+    buildStatusString(1, 0, 0, tests[11]),
+    buildStatusString(1, 0, 0, tests[12]),
+    buildStatusString(0, 1, 0, tests[13],
+        message: 'Expected foo() to be called 1 times but:'
+            ' was called 2 times'),
+    buildStatusString(0, 1, 0, tests[14],
+        message: 'Caught Exception: No behavior specified for method foo'),
+    buildStatusString(0, 1, 0, tests[15],
+        message: 'Expected sum() to sometimes return <0> but: never did')
   ];
 
   actual = [];
 
   nextTest(0);
 }