Add trait for getting all unit tests of a module.