diff options
Diffstat (limited to 'core/src/test/java/org/elasticsearch/common/xcontent/support/filtering/YamlFilteringGeneratorTests.java')
-rw-r--r-- | core/src/test/java/org/elasticsearch/common/xcontent/support/filtering/YamlFilteringGeneratorTests.java | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/core/src/test/java/org/elasticsearch/common/xcontent/support/filtering/YamlFilteringGeneratorTests.java b/core/src/test/java/org/elasticsearch/common/xcontent/support/filtering/YamlFilteringGeneratorTests.java index c85fbc9225..a101da0e4b 100644 --- a/core/src/test/java/org/elasticsearch/common/xcontent/support/filtering/YamlFilteringGeneratorTests.java +++ b/core/src/test/java/org/elasticsearch/common/xcontent/support/filtering/YamlFilteringGeneratorTests.java @@ -22,7 +22,7 @@ package org.elasticsearch.common.xcontent.support.filtering; import org.elasticsearch.common.xcontent.XContentBuilder; import org.elasticsearch.common.xcontent.XContentType; -public class YamlFilteringGeneratorTests extends AbstractFilteringJsonGeneratorTestCase { +public class YamlFilteringGeneratorTests extends AbstractXContentFilteringTestCase { @Override protected XContentType getXContentType() { @@ -30,7 +30,11 @@ public class YamlFilteringGeneratorTests extends AbstractFilteringJsonGeneratorT } @Override - protected void assertXContentBuilder(XContentBuilder expected, XContentBuilder builder) { - assertString(expected, builder); + protected void assertFilterResult(XContentBuilder expected, XContentBuilder actual) { + if (randomBoolean()) { + assertXContentBuilderAsString(expected, actual); + } else { + assertXContentBuilderAsBytes(expected, actual); + } } } |