feat(tests): Add better support for testing