[Unit Tests] Refactor and provide utility classes for convenience

* Put common tasks in convenience wrapper classes