[chore] Fixing unit and e2e tests

Added more comprehensive unit testing for params and fixed broken tests.

@settinghead @donnapep @stulees please review & merge. Thanks!