add functionality for unit and integration tests