jhttp/src/test/java/io/harness/jhttp/server/HeaderListTest.java at main · rajpropelo/jhttp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package io.harness.jhttp.server;

import java.util.List;

import org.junit.Test;

import static org.junit.Assert.*;

public class HeaderListTest {

    @Test
    public void testAddHeader() {
        final HeaderList headerList = new HeaderList();
        headerList.addHeader("h1", "v1");
        headerList.addHeader("h2", "v2");

        assertEquals(2, headerList.getHeaders().size());
        assertEquals("v1", headerList.getHeader("h1"));
        assertEquals("v2", headerList.getHeader("h2"));
    }

    @Test
    public void testSetHeader() {
        final HeaderList headerList = new HeaderList();
        headerList.addHeader("h1", "v1");
        headerList.addHeader("h2", "v2");
        headerList.setHeader("h1", "xyz");

        assertEquals(2, headerList.getHeaders().size());
        assertEquals("xyz", headerList.getHeader("h1"));
        assertEquals("v2", headerList.getHeader("h2"));
    }

    @Test
    public void testSetFirstHeader() {
        final HeaderList headerList = new HeaderList();
        headerList.addHeader("h1", "v1");
        headerList.addHeader("h1", "v2");
        headerList.setHeader("h1", "xyz");

        assertEquals(2, headerList.getHeaders().size());
        final List<String> headers = headerList.getHeaders("h1");
        assertEquals("xyz", headers.get(0));
        assertEquals("v2", headers.get(1));
    }

    @Test
    public void testAddMultipleHeaders() {
        final HeaderList headerList = new HeaderList();
        headerList.addHeader("h1", "v1");
        headerList.addHeader("h1", "v2");
        headerList.addHeader("h2", "v3");

        assertEquals(3, headerList.getHeaders().size());
        final List<String> headers = headerList.getHeaders("h1");
        assertEquals("v1", headers.get(0));
        assertEquals("v2", headers.get(1));
        assertEquals("v3", headerList.getHeader("h2"));
    }

}