Skip to content

Commit bfc8072

Browse files
Made HuggingFaceChatCompletionServiceSettingsTests extend AbstractBWCWireSerializationTestCase for future serialization testing
1 parent 4688901 commit bfc8072

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/huggingface/completion/HuggingFaceChatCompletionServiceSettingsTests.java

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,14 @@
77

88
package org.elasticsearch.xpack.inference.services.huggingface.completion;
99

10+
import org.elasticsearch.TransportVersion;
1011
import org.elasticsearch.common.Strings;
1112
import org.elasticsearch.common.ValidationException;
1213
import org.elasticsearch.common.io.stream.Writeable;
13-
import org.elasticsearch.test.AbstractWireSerializingTestCase;
1414
import org.elasticsearch.xcontent.XContentBuilder;
1515
import org.elasticsearch.xcontent.XContentFactory;
1616
import org.elasticsearch.xcontent.XContentType;
17+
import org.elasticsearch.xpack.core.ml.AbstractBWCWireSerializationTestCase;
1718
import org.elasticsearch.xpack.inference.services.ConfigurationParseContext;
1819
import org.elasticsearch.xpack.inference.services.ServiceFields;
1920
import org.elasticsearch.xpack.inference.services.ServiceUtils;
@@ -27,7 +28,7 @@
2728
import static org.hamcrest.Matchers.containsString;
2829
import static org.hamcrest.Matchers.is;
2930

30-
public class HuggingFaceChatCompletionServiceSettingsTests extends AbstractWireSerializingTestCase<
31+
public class HuggingFaceChatCompletionServiceSettingsTests extends AbstractBWCWireSerializationTestCase<
3132
HuggingFaceChatCompletionServiceSettings> {
3233

3334
public static final String MODEL_ID = "some model";
@@ -226,6 +227,14 @@ protected HuggingFaceChatCompletionServiceSettings mutateInstance(HuggingFaceCha
226227
return randomValueOtherThan(instance, HuggingFaceChatCompletionServiceSettingsTests::createRandomWithNonNullUrl);
227228
}
228229

230+
@Override
231+
protected HuggingFaceChatCompletionServiceSettings mutateInstanceForVersion(
232+
HuggingFaceChatCompletionServiceSettings instance,
233+
TransportVersion version
234+
) {
235+
return instance;
236+
}
237+
229238
private static HuggingFaceChatCompletionServiceSettings createRandomWithNonNullUrl() {
230239
return createRandom(randomAlphaOfLength(15));
231240
}

0 commit comments

Comments
 (0)