The Requirements For Mocking
Following are the requirements that we wanted to fill up (for starters). Though these requirements are specific to mine, I'm sure almost every project would have the same / similar mocking requirements :
- Objective : The most important point of using mocking frameworks was to emulate certain behaviour from instances of external dependencies.
- Lets continue with this example :
| MyLoginService --uses--> BackendAuthService |
| | |
| authenticates via |
| | |
| v |
| DataSource |
- Here the objective is to unit test MyLoginService's login API, which in its integration environment would talk to multiple backend services.
- Lets assume that the BackendAuthService has an API named login, which accepts a custom class object. This custom class object contains the details of the credentials to authenticate.
- Further, lets say this API returns another custom class object which is the authenticated token after successful authentication. Also, if the authentication fails when this API throws an exception.
- Note that this implies that the developer first needs to know possible behaviours exhibited by these dependencies (SuccessCase1, SuccessCase2, FailCase1, FailCase2, ExceptionCase1, ....).
- Then s/he should code the unit test case, such that when the external dependencies are called / invoked, they exhibit the behaviour the unit test case is supposed to test.
- Generally, there would be at least one unit test case for each expected scenario. In addition there would be negative test cases as well.
- A point to explain what I mean by "mock-out the behaviour". In the above example, for a positive test case, we need (to simulate) the backend service returning a valid authenticated token, signifying successful authentication. Using the mocking framework, we can induce this simulation on the backend service instance.
- Modifying the production code flow, for adding any unit testing capability is forbidden.
- Hence, we want to re-use the injection framework, configuration files and the same instance look up code, to seamlessly work while unit testing.
Before I explicitly state the solution we adopted, I want to discuss the various possibilities we considered. This section talks about those.
- Spring supports two bean scopes by default for standalone applications - prototype and singleton (default). What we needed was a way to define mock implementation classes in the spring configuration, such that each thread would have its own mocked instance - basically thread scope. This way the thread safety of the mocked instance and different-behaviour-per-thread requirements can be achieved.
- One way out here was to define a custom scope in Spring. (see also the SimpleThreadScope class in available in 3.0.x version).
- Another way out is to ask Spring to use factories for bean creation and then take control over this aspect by implementing a factory.
- Spring provides a convenient way to do this - using FactoryBeans. By implementing FactoryBean interface, we can write a factory class that will generate beans as well as control singleton behaviour.
- Other methods using instance factories and static factories are also possible
- Use an implementation of Spring's FactoryBean to serve as a factory for creating beans. This implementation, lets name it MockServiceLookUp, will be called whenever the code flow requests for some bean.
- The MockServiceLookUp implementation is responsible for creating mocked instances for the class that is being looked up - one instance per thread.
- The MockServiceLookUp implementation will rely on the use of a MockProvider to create actual mock objects. The MockProvider is the class that will actually use the mocker library to create mocks.
- Ideally, MockProvider should be an interface, with multiple implementations for each type of mocker you want to support (so that those can be plugged in as needed).
- Alternately, MockProvider can be skipped completely and the mocker library can be used in the look up implementation to create mock instances.
- The general case is that unit test will fetch the mock instance for a class that the actual code the unit test is for will be using. Then the unit test would "apply" behaviour mocking on this instance (i.e. behavior stubbing) and finally call the actual API to test.
- Since the API would also look up the same class from Spring (which in turn will request MockServiceLookUp for the instance), the API would get a mock instance with appropriate methods stubbed. These methods would be those that the unit test expects the API to call on the backend/dependency. Hence, the unit test would be complete.
- This was a useful API for us - to start afresh from within a unit test - where we wanted to discard all previous stubbing
- This API was also called by default from out unit testing infrastructure whenever a new unit test case started - hence ensuring all unit tests started clean.
This is what the mock framework class diagram (taken with permission - thanks to my colleague Siju) is like :
Finally, a unit test would run like this :
- Unit Test Case Start
- Use Spring to fetch the implementation for BackendAuthService.
- Spring would invoke the MockServiceLookUp's getObject() API to get the implementation of BackendAuthService.
- MockServiceLookUp would first check if an existing implementation of the class exists already, in its ThreadLocal. If this is the first getObject call for this class, then the mock method of the MockProvider is called. Otherwise, the mock instance from ThreadLocal is returned.
- The MockProvider will use Mockito to create the mock instance of the specified class.
- The unit test will override the behaviour of the BackendAuthService's login API
- The behaviour override can be to return valid values (auth tokens) or invalid values (excetpions, errors, invalid auth tokens) depending on what is being tested
- Then the test would make a call to MyLoginService's login API
- This implementation would again request Spring for the implementation of the BackendAuthService class, which this time will be the stubbed mock instance.
- When the login API on BackendAuthService is called the stubbed code executes returning the auth token / exception.
- After the API call completes, the unit test must verify that the auth token returned is valid or not (or if an exception was excepted, whether it was thrown or not).
- Unit Test Case End