BATCH-2149: Updated test to validate both the simple and fault tolerant use cases