Adding in unit test with easymock.