Added unit tests for test builder