container-core/src/test/java/com/yahoo/container/jdisc/HttpResponseTestCase.java


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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.jdisc;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import com.yahoo.jdisc.Response;
import com.yahoo.text.Utf8;

/**
 * API test for HttpResponse.
 *
 * @author Steinar Knutsen
 */
public class HttpResponseTestCase {

    private static final String COM_YAHOO_CONTAINER_JDISC_HTTP_RESPONSE_TEST_CASE_TEST_RESPONSE = "com.yahoo.container.jdisc.HttpResponseTestCase.TestResponse";

    private static class TestResponse extends HttpResponse {

        public TestResponse(int status) {
            super(status);
        }

        @Override
        public void render(OutputStream outputStream) throws IOException {
            outputStream.write(Utf8.toBytes(COM_YAHOO_CONTAINER_JDISC_HTTP_RESPONSE_TEST_CASE_TEST_RESPONSE));
        }
    }

    HttpResponse r;

    @BeforeEach
    public void setUp() throws Exception {
        r = new TestResponse(Response.Status.OK);
    }

    @AfterEach
    public void tearDown() throws Exception {
        r = null;
    }

    @Test
    final void testRender() throws IOException {
        ByteArrayOutputStream o = new ByteArrayOutputStream(1024);
        r.render(o);
        assertEquals(COM_YAHOO_CONTAINER_JDISC_HTTP_RESPONSE_TEST_CASE_TEST_RESPONSE, Utf8.toString(o.toByteArray()));
    }

    @Test
    final void testGetStatus() {
        assertEquals(Response.Status.OK, r.getStatus());
    }

    @Test
    final void testHeaders() {
        assertNotNull(r.headers());
    }

    @Test
    final void testGetJdiscResponse() {
        assertNotNull(r.getJdiscResponse());
    }

    @Test
    final void testGetContentType() {
        assertEquals(HttpResponse.DEFAULT_MIME_TYPE, r.getContentType());
    }

    @Test
    final void testGetCharacterEncoding() {
        assertEquals(HttpResponse.DEFAULT_CHARACTER_ENCODING, r.getCharacterEncoding());
    }

}