KER-239: Aligned the unit tests with other integrations