Power of EasyMock 2 with Java 1.4

public class EasyMockHelper {

  private static MockControl lastInvokedControl;

  public static ReturnValueSetter expect(boolean b) {
    return new ReturnValueSetter(lastInvokedControl);
  }

  public static Object getMock(Class cl) {
    Object proxy = Proxy.newProxyInstance(cl.getClassLoader(), new Class[]{cl},
            new MockInvocationHandler(MockControl.createControl(cl)));
    return proxy;
  }

  private static class MockInvocationHandler implements InvocationHandler {

    private final MockControl control;

    public MockInvocationHandler(MockControl control) {
      this.control = control;
    }

    public MockControl getControl() {
      return control;
    }

    public Object invoke(Object proxy, Method method, Object[] args)
            throws Throwable {
      Object retValue;
      try {
        retValue = method.invoke(control.getMock(), args);
      } finally {
        lastInvokedControl = control;       
      }
      return retValue;
    }
  }

  public static void replay(Object mock) {
    getControl(mock).replay();
  }

  private static MockControl getControl(Object mock) {
    MockInvocationHandler invocationHandler = (MockInvocationHandler) Proxy
            .getInvocationHandler(mock);
    return invocationHandler.getControl();
  }

  public static void verify(Object mock) {
    getControl(mock).verify();
  }
}

public class ReturnValueSetter {

  private final MockControl lastInvokedControl;

  public ReturnValueSetter(MockControl lastInvokedControl) {
    this.lastInvokedControl = lastInvokedControl;     
  }
 
  public void andReturn(boolean b) {
    lastInvokedControl.setReturnValue(b);
  }

}

public class EasyMockHelperTest extends TestCase {

  private interface Sample {
    boolean getBoolean();
  }

  public void testExpectReturnTrue() throws Exception {
    Sample mock = (Sample) EasyMockHelper.getMock(Sample.class);

    EasyMockHelper.expect(mock.getBoolean()).andReturn(true);
    EasyMockHelper.replay(mock);

    assertTrue(mock.getBoolean());
  }

  public void testExpectReturnFalse() throws Exception {
    Sample mock = (Sample) EasyMockHelper.getMock(Sample.class);

    EasyMockHelper.expect(mock.getBoolean()).andReturn(false);
    EasyMockHelper.replay(mock);

    assertFalse(mock.getBoolean());
  }

  public void testExpectMultipleCalls() throws Exception {
    Sample mock = (Sample) EasyMockHelper.getMock(Sample.class);

    EasyMockHelper.expect(mock.getBoolean()).andReturn(false);
    EasyMockHelper.expect(mock.getBoolean()).andReturn(true);
    EasyMockHelper.replay(mock);

    assertFalse(mock.getBoolean());
    assertTrue(mock.getBoolean());
  }
}