/*
* Licensed to Elasticsearch under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.elasticsearch.common.settings;
import org.elasticsearch.common.Strings;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.common.xcontent.json.JsonXContent;
import org.elasticsearch.rest.RestRequest;
import org.elasticsearch.test.ESTestCase;
import org.elasticsearch.test.rest.FakeRestRequest;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashSet;
import static org.hamcrest.CoreMatchers.equalTo;
public class SettingsFilterTests extends ESTestCase {
public void testAddingAndRemovingFilters() {
HashSet<String> hashSet = new HashSet<>(Arrays.asList("foo", "bar", "baz"));
SettingsFilter settingsFilter = new SettingsFilter(Settings.EMPTY, hashSet);
assertEquals(settingsFilter.getPatterns(), hashSet);
}
public void testSettingsFiltering() throws IOException {
testFiltering(Settings.builder()
.put("foo", "foo_test")
.put("foo1", "foo1_test")
.put("bar", "bar_test")
.put("bar1", "bar1_test")
.put("bar.2", "bar2_test")
.build(),
Settings.builder()
.put("foo1", "foo1_test")
.build(),
"foo", "bar*"
);
testFiltering(Settings.builder()
.put("foo", "foo_test")
.put("foo1", "foo1_test")
.put("bar", "bar_test")
.put("bar1", "bar1_test")
.put("bar.2", "bar2_test")
.build(),
Settings.builder()
.put("foo", "foo_test")
.put("foo1", "foo1_test")
.build(),
"bar*"
);
testFiltering(Settings.builder()
.put("foo", "foo_test")
.put("foo1", "foo1_test")
.put("bar", "bar_test")
.put("bar1", "bar1_test")
.put("bar.2", "bar2_test")
.build(),
Settings.builder()
.build(),
"foo", "bar*", "foo*"
);
testFiltering(Settings.builder()
.put("foo", "foo_test")
.put("bar", "bar_test")
.put("baz", "baz_test")
.build(),
Settings.builder()
.put("foo", "foo_test")
.put("bar", "bar_test")
.put("baz", "baz_test")
.build()
);
testFiltering(Settings.builder()
.put("a.b.something.d", "foo_test")
.put("a.b.something.c", "foo1_test")
.build(),
Settings.builder()
.put("a.b.something.c", "foo1_test")
.build(),
"a.b.*.d"
);
}
private void testFiltering(Settings source, Settings filtered, String... patterns) throws IOException {
SettingsFilter settingsFilter = new SettingsFilter(Settings.EMPTY, Arrays.asList(patterns));
// Test using direct filtering
Settings filteredSettings = settingsFilter.filter(source);
assertThat(filteredSettings.getAsMap().entrySet(), equalTo(filtered.getAsMap().entrySet()));
// Test using toXContent filtering
RestRequest request = new FakeRestRequest();
settingsFilter.addFilterSettingParams(request);
XContentBuilder xContentBuilder = XContentBuilder.builder(JsonXContent.jsonXContent);
xContentBuilder.startObject();
source.toXContent(xContentBuilder, request);
xContentBuilder.endObject();
String filteredSettingsString = xContentBuilder.string();
filteredSettings = Settings.builder().loadFromSource(filteredSettingsString, xContentBuilder.contentType()).build();
assertThat(filteredSettings.getAsMap().entrySet(), equalTo(filtered.getAsMap().entrySet()));
}
}