HttpDataTest Class — netty Architecture

Source Code

codec-http/src/test/java/io/netty/handler/codec/http/multipart/HttpDataTest.java lines 38–150

class HttpDataTest {
    private static final byte[] BYTES = new byte[64];

    @Retention(RetentionPolicy.RUNTIME)
    @Target(ElementType.METHOD)
    @ParameterizedTest(name = "{displayName}({0})")
    @MethodSource("data")
    @interface ParameterizedHttpDataTest {
    }

    static HttpData[] data() {
        return new HttpData[]{
                new MemoryAttribute("test", 10),
                new MemoryFileUpload("test", "", "text/plain", null, CharsetUtil.UTF_8, 10),
                new MixedAttribute("test", 10, -1),
                new MixedFileUpload("test", "", "text/plain", null, CharsetUtil.UTF_8, 10, -1),
                new DiskAttribute("test", 10),
                new DiskFileUpload("test", "", "text/plain", null, CharsetUtil.UTF_8, 10)
        };
    }

    @BeforeAll
    static void setUp() {
        Random rndm = new Random();
        rndm.nextBytes(BYTES);
    }

    @ParameterizedHttpDataTest
    void testAddContentEmptyBuffer(HttpData httpData) throws IOException {
        ByteBuf content = PooledByteBufAllocator.DEFAULT.buffer();
        httpData.addContent(content, false);
        assertThat(content.refCnt()).isEqualTo(0);
    }

    @ParameterizedHttpDataTest
    void testCompletedFlagPreservedAfterRetainDuplicate(HttpData httpData) throws IOException {
        httpData.addContent(Unpooled.wrappedBuffer("foo".getBytes(CharsetUtil.UTF_8)), false);
        assertThat(httpData.isCompleted()).isFalse();
        HttpData duplicate = httpData.retainedDuplicate();
        assertThat(duplicate.isCompleted()).isFalse();
        assertThat(duplicate.release()).isTrue();
        httpData.addContent(Unpooled.wrappedBuffer("bar".getBytes(CharsetUtil.UTF_8)), true);
        assertThat(httpData.isCompleted()).isTrue();
        duplicate = httpData.retainedDuplicate();
        assertThat(duplicate.isCompleted()).isTrue();
        assertThat(duplicate.release()).isTrue();
    }

    @Test
    void testAddContentExceedsDefinedSizeDiskFileUpload() {
        doTestAddContentExceedsSize(
                new DiskFileUpload("test", "", "application/json", null, CharsetUtil.UTF_8, 10),
                "Out of size: 64 > 10");
    }

    @Test
    void testAddContentExceedsDefinedSizeMemoryFileUpload() {
        doTestAddContentExceedsSize(
                new MemoryFileUpload("test", "", "application/json", null, CharsetUtil.UTF_8, 10),
                "Out of size: 64 > 10");
    }

    @ParameterizedHttpDataTest
    void testAddContentExceedsMaxSize(final HttpData httpData) {
        httpData.setMaxSize(10);
        doTestAddContentExceedsSize(httpData, "Size exceed allowed maximum capacity");
    }

    @ParameterizedHttpDataTest
    void testSetContentExceedsDefinedSize(final HttpData httpData) {
        doTestSetContentExceedsSize(httpData, "Out of size: 64 > 10");
    }

    @ParameterizedHttpDataTest
    void testSetContentExceedsMaxSize(final HttpData httpData) {
        httpData.setMaxSize(10);
        doTestSetContentExceedsSize(httpData, "Size exceed allowed maximum capacity");
    }

    private static void doTestAddContentExceedsSize(final HttpData httpData, String expectedMessage) {
        final ByteBuf content = PooledByteBufAllocator.DEFAULT.buffer();