mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-06-14 07:34:50 +02:00
e98d42790a
This commit adds - SmackReactor / NIO - a framework for finite state machine connections - support for Java 8 - pretty printed XML debug output It also - reworks the integration test framework - raises the minimum Android API level to 19 - introduces XmppNioTcpConnection Furthermore fixes SMACK-801 (at least partly). Java 8 language features are available, but not all runtime library methods. For that we would need to raise the Android API level to 24 or higher.
106 lines
4.1 KiB
Java
106 lines
4.1 KiB
Java
/**
|
|
*
|
|
* Copyright 2017-2019 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);
|
|
}
|
|
}
|