Enterprise Java

Testing Abstract Classes and Template Method Pattern

From wikipedia “A template method defines the program skeleton of an algorithm. One or more of the algorithm steps can be overridden by subclasses to allow differing behaviors while ensuring that the overarching algorithm is still followed”.

Typically this pattern is composed by two or more classes, one that is an abstract class providing template methods (non-abstract) that have calls to abstract methods implemented by one or more concrete subclasses.

Often template abstract class and concrete implementations reside in the same project, but depending on the scope of the project, these concrete objects will be implemented into another project.

In this post we are going to see how to test template method pattern when concrete classes are implemented on external project, or more general how to test abstract classes.

Let’s see a simple example of template method pattern. Consider a class which is responsible of receiving a vector of integers and calculate the Euclidean norm. These integers could be received from multiple sources, and is left to each project to provide a way to obtain them.

The template class looks like:

public abstract class AbstractCalculator {

 public double euclideanNorm() {

  int[] vector = this.read();

  int total = 0;

  for(int element:vector) {
   total+= (element*element);  
  }

  return Math.sqrt(total);
 }

 public abstract int[] read();
}

Now another project could extend previous class and make an implementation of abstract calculator by providing an implementation of read() method .

public class ConsoleCalculator extends AbstractCalculator {

 public int[] read() {

  int [] data = new int[0];

  Scanner scanner = new Scanner(System.in);

  //data = read requried data from console

  return data; 

 }

}

Developer that has written a concrete implementation will test only read() method, he can “trust” that developer of abstract class has tested non-abstract methods.

But how are we going to write unit tests over calculate method if class is abstract and an implementation of read() method is required?

The first approach could be creating a fake implementation:

public class FakeCalculator extends AbstractCalculator {

 private int[] data;

 public FakeCalculator(int[] data) {
  this.data = data;
 }

 public int[] read() {
  return this.data;
 }

}

This is not a bad approach, but has some disadvantages:

  • Test will be less readable, readers should know the existence of these fake classes and must know exactly what are they doing.
  • As a test writer you will spend time in implementing fake classes, in this case it is simple, but your project could have more than one abstract class without implementation, or even with more than one abstract method.
  • Behaviour of fake classes are “hard-coded”.

A better way is using Mockito to mock only abstract method meanwhile implementation of non-abstract methods are called.

public class WhenCalculatingEuclideanNorm {

 @Test
 public void should_calculate_correctly() {

  AbstractCalculator abstractCalculator = mock(AbstractCalculator.class, Mockito.CALLS_REAL_METHODS);

  doReturn(new int[]{2,2}).when(abstractCalculator).read();
  assertThat(abstractCalculator.euclideanNorm(), is(2.8284271247461903));

 }

 @Test
 public void should_calculate_correctly_with_negative_values() {

  AbstractCalculator abstractCalculator = mock(AbstractCalculator.class, Mockito.CALLS_REAL_METHODS);

  doReturn(new int[]{-2,-2}).when(abstractCalculator).read();
  assertThat(abstractCalculator.euclideanNorm(), is(2.8284271247461903));

 }

}

Mockito simplifies the testing of abstract classes by calling real methods, and only stubbing abstract methods. See that in this case because we are calling real methods by default, instead of using the typical when() then() structure, doReturn schema must be used.

Of course this approach can be only used if your project does not contain a concrete implementation of algorithm or your project will be a part of a 3rd party library on another project. In the other cases the best way of attacking the problem is by testing the implemented class.

Download sourcecode

Reference: Testing Abstract Classes (and Template Method Pattern in Particular) from our JCG partner Alex Soto at the One Jar To Rule Them All blog.

Subscribe
Notify of
guest

This site uses Akismet to reduce spam. Learn how your comment data is processed.

3 Comments
Oldest
Newest Most Voted
Inline Feedbacks
View all comments
dharmendar rao
dharmendar rao
12 years ago

thanks for sharing it.

Benoit Guérout
12 years ago

Mockito rocks! Nevertheless always consider using Delegation instead of Template Method (Inheritance).
In your example, if AbtrsctaCalculator class was a concreate class with an injectable Reader instance, it should be easier to test:

public class Calculator {

private final Reader reader;

public Calculator(Reader reader){
this.reader = reader;
}

public double euclideanNorm() {

int[] vector = this.reader.read();

int total = 0;

for(int element:vector) {
total+= (element*element);
}

return Math.sqrt(total);
}
//no more abstract method to implement}

Krzysztof
Krzysztof
6 years ago

What is the best approach when there are a few implemented classes?
How to test without duplicating code in tests?

Back to top button