Partial mocking of objects using easymock classextensions

Posted on September 11, 2008

1


Ocassionally, some of your methods in your classes need to call some other public methods in the same class to get the job done. But when writing unit tests for the calling methods, you want to keep your tests isolated from the calls to other public methods. I encountered such a scenario recently and found that Easymock can help you by mocking some of the methods while keeping the rest of the methods intact.

Let us see an example,
The BlogSearchService has two public methods, first method filterBlogsByAuthor takes an author name and a list of blogs and returns all the blogs written by that particular order author.
The other business method searchBlogs will take a search paramter and the author, it will get all the blogs by the search parameter using a custom logic and do some other house keeping tasks like sorting them by published date and number of comments and then filter the methods by author.

package com.teja.blogs;

import java.util.Collections;import java.util.List;

/**** @author Teja*/public class BlogSearchService {

      public List<Blog> searchBlogs(String searchString, String author){

              //do some custom logic here on these blogs like sorting them by posted date.              return filterBlogsByAuthor(author, Collections.EMPTY_LIST);      }

      public List<Blog> filterBlogsByAuthor(String author, List<Blog> blogs){
             return Collections.EMPTY_LIST;
 }

}

When writing the test cases for the second method, I don’t want to test the other public method filterByAuthor as
(i)it has already been tested by other test cases,
(ii)I want to keep my test cases more unit.

So in my test case where I am testing my searchBlogs, I am creating the mock service class which has mock methods only for filterBlogsByAuthor, but the actual mehtod(searchBlogs) which is about to be tested is not.
As usual you can do expectations on the mock class, replay and verify as if it is some other external resource being mocked.

To create a partial mock class of the service, the extra parameter you need to pass is the name of the method to be mocked and the parameter types it accepts (coz, it need to know which version of the method you want to mock if you have overloaded methods)

Here is our test class:

package com.teja.blogs;

import java.util.Collections;import java.util.List;import org.junit.Test;import org.easymock.classextension.EasyMock;

/**** @author Teja*/public class BlogSearchServiceTest {

        @Test        
public void testSearchBlogs() throws NoSuchMethodException {
 String author = "Teja";
BlogSearchService blogSearchService =
EasyMock.createMock(BlogSearchService.class,BlogSearchService.class.getMethod("filterBlogsByAuthor", 
new Class[]{String.class, List.class}));
EasyMock.expect(blogSearchService.filterBlogsByAuthor(author, Collections.EMPTY_LIST)).andReturn(Collections.EMPTY_LIST);                EasyMock.replay(blogSearchService);                blogSearchService.searchBlogs("easymock", author);                EasyMock.verify(blogSearchService);        }

}
Advertisements
Posted in: easymock, junit, testing