Add basic unit tests support