1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19 package org.apache.shiro.web;
20
21 import groovy.lang.Closure;
22
23 import java.io.Closeable;
24 import java.util.Collections;
25 import java.util.Map;
26 import java.util.Properties;
27
28 /**
29 * Wrapper that will restore System properties after test methods.
30 *
31 * Based on: https://github.com/stefanbirkner/system-rules/blob/master/src/main/java/org/junit/contrib/java/lang/system/RestoreSystemProperties.java
32 */
33 public class RestoreSystemProperties implements Closeable {
34
35 private final Properties originalProperties;
36
37 public RestoreSystemProperties() {
38 originalProperties = System.getProperties();
39 System.setProperties(copyOf(originalProperties));
40 }
41
42 public void restore() {
43 System.setProperties(originalProperties);
44 }
45
46 private Properties copyOf(Properties source) {
47 Properties copy = new Properties();
48 copy.putAll(source);
49 return copy;
50 }
51
52 public static <T> T withProperties(Closure<T> closure) {
53 return withProperties(Collections.emptyMap(), closure);
54 }
55
56 public static <T> T withProperties(Map<String, String> properties, Closure<T> closure) {
57
58 try (RestoreSystemPropertiesmProperties">RestoreSystemProperties restoreSystemProperties = new RestoreSystemProperties()) {
59 properties.forEach(System::setProperty);
60
61 return closure.call();
62 }
63 }
64
65 @Override
66 public void close() {
67 restore();
68 }
69 }