mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-06-20 10: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.
78 lines
3.8 KiB
Java
78 lines
3.8 KiB
Java
/**
|
|
*
|
|
* Copyright 2018 Paul Schaub
|
|
*
|
|
* 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.omemo;
|
|
|
|
import static junit.framework.TestCase.assertEquals;
|
|
import static junit.framework.TestCase.assertFalse;
|
|
import static junit.framework.TestCase.assertTrue;
|
|
|
|
import java.io.IOException;
|
|
|
|
import org.jivesoftware.smack.SmackException;
|
|
import org.jivesoftware.smack.XMPPException;
|
|
|
|
import org.jivesoftware.smackx.omemo.exceptions.CryptoFailedException;
|
|
import org.jivesoftware.smackx.omemo.exceptions.ReadOnlyDeviceException;
|
|
import org.jivesoftware.smackx.omemo.exceptions.UndecidedOmemoIdentityException;
|
|
|
|
import org.igniterealtime.smack.inttest.SmackIntegrationTest;
|
|
import org.igniterealtime.smack.inttest.SmackIntegrationTestEnvironment;
|
|
import org.igniterealtime.smack.inttest.TestNotPossibleException;
|
|
|
|
public class ReadOnlyDeviceIntegrationTest extends AbstractTwoUsersOmemoIntegrationTest {
|
|
|
|
public ReadOnlyDeviceIntegrationTest(SmackIntegrationTestEnvironment environment) throws XMPPException.XMPPErrorException, SmackException.NotConnectedException, InterruptedException, SmackException.NoResponseException, TestNotPossibleException {
|
|
super(environment);
|
|
}
|
|
|
|
@SmackIntegrationTest
|
|
public void test() throws InterruptedException, SmackException.NoResponseException,
|
|
SmackException.NotLoggedInException, SmackException.NotConnectedException, CryptoFailedException,
|
|
UndecidedOmemoIdentityException, IOException {
|
|
boolean prevIgnoreReadOnlyConf = OmemoConfiguration.getIgnoreReadOnlyDevices();
|
|
int prevMaxMessageCounter = OmemoConfiguration.getMaxReadOnlyMessageCount();
|
|
|
|
OmemoConfiguration.setIgnoreReadOnlyDevices(true);
|
|
// Set the maxReadOnlyMessageCount to ridiculously low threshold of 5.
|
|
// This means that Alice will be able to encrypt 5 messages for Bob, while the 6th will not be encrypted for Bob.
|
|
OmemoConfiguration.setMaxReadOnlyMessageCount(5);
|
|
|
|
// Reset counter to begin test
|
|
alice.getOmemoService().getOmemoStoreBackend().storeOmemoMessageCounter(alice.getOwnDevice(), bob.getOwnDevice(), 0);
|
|
|
|
// Since the max threshold is 5, we must be able to encrypt 5 messages for Bob.
|
|
for (int i = 0; i < 5; i++) {
|
|
assertEquals(i, alice.getOmemoService().getOmemoStoreBackend().loadOmemoMessageCounter(alice.getOwnDevice(), bob.getOwnDevice()));
|
|
OmemoMessage.Sent message = alice.encrypt(bob.getOwnJid(), "Hello World!");
|
|
assertFalse(message.getSkippedDevices().containsKey(bob.getOwnDevice()));
|
|
}
|
|
|
|
// Now the message counter must be too high and Bobs device must be skipped.
|
|
OmemoMessage.Sent message = alice.encrypt(bob.getOwnJid(), "Hello World!");
|
|
Throwable exception = message.getSkippedDevices().get(bob.getOwnDevice());
|
|
assertTrue(exception instanceof ReadOnlyDeviceException);
|
|
assertEquals(bob.getOwnDevice(), ((ReadOnlyDeviceException) exception).getDevice());
|
|
|
|
// Reset the message counter
|
|
alice.getOmemoService().getOmemoStoreBackend().storeOmemoMessageCounter(alice.getOwnDevice(), bob.getOwnDevice(), 0);
|
|
|
|
// Reset the configuration to previous values
|
|
OmemoConfiguration.setMaxReadOnlyMessageCount(prevMaxMessageCounter);
|
|
OmemoConfiguration.setIgnoreReadOnlyDevices(prevIgnoreReadOnlyConf);
|
|
}
|
|
}
|