cmake: Added code to handle unit tests