Extending unit tests