Как написать модульные тесты с помощью commons-fileupload? - PullRequest
1 голос
/ 17 декабря 2010

Я работаю в среде, которая использует commons-fileupload.

Теперь мне нужно написать классы модульного теста.Я сомневаюсь, как написать фиктивные тесты, если ServletFileUpload не является интерфейсом?Есть еще один способ проверить мои классы?

Вчера я написал модульные тесты, используя mockito для частей сервлета, и это легко.Но я не могу думать, как написать тесты для загрузки файла.

Ответы [ 3 ]

1 голос
/ 09 августа 2016

Вы можете использовать Commons HTTPClient для создания многочастного потока:

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.apache.http.HttpEntity;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.mime.MultipartEntityBuilder;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import javax.servlet.ServletException;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class ServletFileUploadMock {

    public static class FileUpload {

        private final String filename;
        private final String mimeType;
        private final byte[] contents;

        public FileUpload(String filename, String mimeType, byte[] contents) {
            this.filename = filename;
            this.mimeType = mimeType;
            this.contents = contents;
        }

    }

    public static HttpServletResponse mockServletFileUpload(HttpServlet servlet, List<FileUpload> files,
            Map<String, String> queryParams) throws IOException, ServletException {

        MultipartEntityBuilder builder = MultipartEntityBuilder.create();
        for (FileUpload f : files) {
            builder = builder.addBinaryBody(f.filename, f.contents, ContentType.create(f.mimeType), f.filename);
        }
        HttpEntity entity = builder.build();

        ByteArrayOutputStream os = new ByteArrayOutputStream();
        entity.writeTo(os);
        ByteArrayInputStream is = new ByteArrayInputStream(os.toByteArray());

        HttpServletRequest mockRequest = mock(HttpServletRequest.class);
        when(mockRequest.getMethod()).thenReturn("POST");
        when(mockRequest.getContentType()).thenReturn(entity.getContentType().getValue());
        when(mockRequest.getContentLength()).thenReturn((int)entity.getContentLength());
        when(mockRequest.getInputStream()).thenReturn(new MockServletInputStream(is));

        // Mock query params
        for (Entry<String, String> p : queryParams.entrySet()) {
            when(mockRequest.getParameter(p.getKey())).thenReturn(p.getValue());
        }

        HttpServletResponse mockResponse = mock(HttpServletResponse.class);

        servlet.service(mockRequest, mockResponse);

        return mockResponse;
    }

    public static class MockServletInputStream extends ServletInputStream {

        private final InputStream delegate;

        public MockServletInputStream(InputStream delegate) {
            this.delegate = delegate;
        }

        @Override
        public int read() throws IOException {
            return delegate.read();
        }

    }
}
0 голосов
/ 17 декабря 2010

Я бы посоветовал обернуть ваши вызовы FileUpload в другой слой. Хотя это может звучать как излишнее, это позволит вам очень быстро переключать загружаемые библиотеки с добавленной возможностью для лучшего модульного тестирования ваших звонков ... Кажется, вы тесно связываете свое приложение с commons-fileupload ...

0 голосов
/ 17 декабря 2010

Вы можете вернуться к основам и прокрутить фиктивный объект вручную, обернув или расширив и переопределив. Иногда легче не зависеть от насмешливых рамок для всего.

Результаты издевательских издевательств и испытаний

...