mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-06-16 00:24:49 +02:00
This is a complete redesign of what was previously XmppNioTcpConnection. The new architecture allows to extend an XMPP client to server (c2s) connection with new transport bindings and other extensions.
106 lines
4.1 KiB
Java
106 lines
4.1 KiB
Java
/**
|
|
*
|
|
* Copyright 2017-2020 Florian Schmaus
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package org.jivesoftware.smackx.httpfileupload;
|
|
|
|
import static org.junit.Assert.assertArrayEquals;
|
|
|
|
import java.io.BufferedInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.net.HttpURLConnection;
|
|
import java.net.URL;
|
|
|
|
import org.jivesoftware.smack.SmackException;
|
|
import org.jivesoftware.smack.SmackException.NoResponseException;
|
|
import org.jivesoftware.smack.SmackException.NotConnectedException;
|
|
import org.jivesoftware.smack.XMPPException.XMPPErrorException;
|
|
|
|
import org.igniterealtime.smack.inttest.AbstractSmackIntegrationTest;
|
|
import org.igniterealtime.smack.inttest.SmackIntegrationTest;
|
|
import org.igniterealtime.smack.inttest.SmackIntegrationTestEnvironment;
|
|
import org.igniterealtime.smack.inttest.TestNotPossibleException;
|
|
|
|
public class HttpFileUploadIntegrationTest extends AbstractSmackIntegrationTest {
|
|
|
|
private static final int FILE_SIZE = 1024 * 128;
|
|
|
|
private final HttpFileUploadManager hfumOne;
|
|
|
|
public HttpFileUploadIntegrationTest(SmackIntegrationTestEnvironment environment) throws XMPPErrorException,
|
|
NotConnectedException, NoResponseException, InterruptedException, TestNotPossibleException {
|
|
super(environment);
|
|
hfumOne = HttpFileUploadManager.getInstanceFor(conOne);
|
|
if (!hfumOne.discoverUploadService()) {
|
|
throw new TestNotPossibleException(
|
|
"HttpFileUploadManager was unable to discover a HTTP File Upload service");
|
|
}
|
|
UploadService uploadService = hfumOne.getDefaultUploadService();
|
|
if (!uploadService.acceptsFileOfSize(FILE_SIZE)) {
|
|
throw new TestNotPossibleException("The upload service at " + uploadService.getAddress()
|
|
+ " does not accept files of size " + FILE_SIZE
|
|
+ ". It only accepts files with a maximum size of " + uploadService.getMaxFileSize());
|
|
}
|
|
hfumOne.setTlsContext(environment.configuration.tlsContext);
|
|
}
|
|
|
|
@SmackIntegrationTest
|
|
public void httpFileUploadTest() throws IOException, XMPPErrorException, InterruptedException, SmackException {
|
|
final int fileSize = FILE_SIZE;
|
|
File file = createNewTempFile();
|
|
FileOutputStream fos = new FileOutputStream(file.getCanonicalPath());
|
|
byte[] upBytes;
|
|
try {
|
|
upBytes = new byte[fileSize];
|
|
INSECURE_RANDOM.nextBytes(upBytes);
|
|
fos.write(upBytes);
|
|
}
|
|
finally {
|
|
fos.close();
|
|
}
|
|
|
|
URL getUrl = hfumOne.uploadFile(file, new UploadProgressListener() {
|
|
@Override
|
|
public void onUploadProgress(long uploadedBytes, long totalBytes) {
|
|
double progress = uploadedBytes / totalBytes;
|
|
LOGGER.fine("HTTP File Upload progress " + progress + "% (" + uploadedBytes + '/' + totalBytes + ')');
|
|
}
|
|
});
|
|
|
|
HttpURLConnection urlConnection = getHttpUrlConnectionFor(getUrl);
|
|
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream(fileSize);
|
|
byte[] buffer = new byte[4096];
|
|
int n;
|
|
try {
|
|
InputStream is = new BufferedInputStream(urlConnection.getInputStream());
|
|
while ((n = is.read(buffer)) != -1) {
|
|
baos.write(buffer, 0, n);
|
|
}
|
|
}
|
|
finally {
|
|
urlConnection.disconnect();
|
|
}
|
|
|
|
byte[] downBytes = baos.toByteArray();
|
|
|
|
assertArrayEquals(upBytes, downBytes);
|
|
}
|
|
}
|