View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  package org.apache.juneau.http.header;
18  
19  import static org.apache.juneau.TestUtils.*;
20  import static org.apache.juneau.http.HttpHeaders.*;
21  import static org.junit.jupiter.api.Assertions.*;
22  
23  import java.io.*;
24  import java.util.function.*;
25  
26  import org.apache.juneau.*;
27  import org.apache.juneau.annotation.*;
28  import org.apache.juneau.common.utils.*;
29  import org.apache.juneau.http.annotation.*;
30  import org.apache.juneau.rest.annotation.*;
31  import org.apache.juneau.rest.client.*;
32  import org.apache.juneau.rest.mock.*;
33  import org.junit.jupiter.api.*;
34  
35  class Connection_Test extends TestBase {
36  
37  	private static final String HEADER = "Connection";
38  	private static final String VALUE = "foo";
39  	private static final String PARSED = "foo";
40  
41  	@Rest
42  	public static class A {
43  		@RestOp
44  		public StringReader get(@Header(name=HEADER) @Schema(cf="multi") String[] h) {
45  			return reader(h == null ? "null" : Utils.join(h, ','));
46  		}
47  	}
48  
49  	//------------------------------------------------------------------------------------------------------------------
50  	// Method tests
51  	//------------------------------------------------------------------------------------------------------------------
52  
53  	@Test void a01_basic() throws Exception {
54  		var c = client().build();
55  
56  		// Normal usage.
57  		c.get().header(connection(VALUE)).run().assertContent(VALUE);
58  		c.get().header(connection(VALUE)).run().assertContent(VALUE);
59  		c.get().header(connection(PARSED)).run().assertContent(VALUE);
60  		c.get().header(connection(()->PARSED)).run().assertContent(VALUE);
61  
62  		// Invalid usage.
63  		c.get().header(connection((String)null)).run().assertContent("Keep-Alive");
64  		c.get().header(connection((Supplier<String>)null)).run().assertContent("Keep-Alive");
65  		c.get().header(connection(()->null)).run().assertContent("Keep-Alive");
66  	}
67  
68  	@Test void a02_other() {
69  		assertTrue(connection("Keep-Alive").isKeepAlive());
70  		assertTrue(connection("keep-alive").isKeepAlive());
71  		assertFalse(connection("foo").isKeepAlive());
72  		assertTrue(connection("Close").isClose());
73  		assertTrue(connection("close").isClose());
74  		assertFalse(connection("foo").isClose());
75  		assertTrue(connection("Upgrade").isUpgrade());
76  		assertTrue(connection("upgrade").isUpgrade());
77  		assertFalse(connection("foo").isUpgrade());
78  	}
79  
80  	//------------------------------------------------------------------------------------------------------------------
81  	// Helper methods.
82  	//------------------------------------------------------------------------------------------------------------------
83  
84  	private static RestClient.Builder client() {
85  		return MockRestClient.create(A.class);
86  	}
87  }