import org.jmock.Expectations;
import org.jmock.Mockery;
import org.junit.Test;
public class JMockExampleTest {
@Test
public void testJMockExample() {
Mockery context = new Mockery();
final DatabaseConnection mockedConnection = context.mock(DatabaseConnection.class);
context.checking(new Expectations() {{
\tallowing(mockedConnection).connect(); will(returnValue(true));
\tallowing(mockedConnection).executeQuery("SELECT * FROM users"); will(returnValue("User1, User2, User3"));
}});
DatabaseService databaseService = new DatabaseService();
databaseService.setConnection(mockedConnection);
boolean isConnected = databaseService.connectToDatabase();
String queryResult = databaseService.executeQuery("SELECT * FROM users");
context.assertIsSatisfied();
assertEquals(true, isConnected);
assertEquals("User1, User2, User3", queryResult);
}
}