@mock.patch.object(rest_client.KioxiaProvisioner, 'expand_volume')
def test_extend_volume_success(self, mock_expand_volume):
    testvol = _stub_volume()
    mock_expand_volume.return_value = success_prov_response
    new_size = (VOL_SIZE + 2)
    result = self.driver.extend_volume(testvol, new_size)
    mock_expand_volume.assert_any_call(new_size, testvol['id'])
    self.assertIsNone(result)
@mock.patch.object(rest_client.KioxiaProvisioner, 'expand_volume')
def test_extend_volume_success(self, mock_expand_volume):
    testvol = _stub_volume()
    mock_expand_volume.return_value = success_prov_response
    new_size = (VOL_SIZE + 2)
    result = self.driver.extend_volume(testvol, new_size)
    mock_expand_volume.assert_any_call(new_size, testvol['id'])
    self.assertIsNone(result)