nick
3 years ago
12 changed files with 193 additions and 63 deletions
@ -0,0 +1,99 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* 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.thingsboard.server.transport.lwm2m.security.diffPort; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.eclipse.leshan.core.peer.IpPeer; |
|||
import org.eclipse.leshan.core.peer.LwM2mPeer; |
|||
import org.eclipse.leshan.core.peer.SocketIdentity; |
|||
import org.eclipse.leshan.server.registration.RegistrationStore; |
|||
import org.eclipse.leshan.server.registration.RegistrationUpdate; |
|||
import org.junit.Assert; |
|||
import org.mockito.invocation.InvocationOnMock; |
|||
import org.mockito.stubbing.Answer; |
|||
import org.springframework.boot.test.mock.mockito.SpyBean; |
|||
import org.thingsboard.server.common.data.device.profile.Lwm2mDeviceProfileTransportConfiguration; |
|||
import org.thingsboard.server.dao.service.DaoSqlTest; |
|||
import org.thingsboard.server.transport.lwm2m.security.AbstractSecurityLwM2MIntegrationTest; |
|||
|
|||
import java.net.InetAddress; |
|||
import java.net.InetSocketAddress; |
|||
import java.util.concurrent.TimeUnit; |
|||
|
|||
import static org.awaitility.Awaitility.await; |
|||
import static org.mockito.ArgumentMatchers.any; |
|||
import static org.mockito.Mockito.doAnswer; |
|||
import static org.thingsboard.server.transport.lwm2m.Lwm2mTestHelper.LwM2MClientState.ON_REGISTRATION_STARTED; |
|||
import static org.thingsboard.server.transport.lwm2m.Lwm2mTestHelper.LwM2MClientState.ON_REGISTRATION_SUCCESS; |
|||
import static org.thingsboard.server.transport.lwm2m.Lwm2mTestHelper.LwM2MClientState.ON_UPDATE_SUCCESS; |
|||
|
|||
@DaoSqlTest |
|||
@Slf4j |
|||
public abstract class AbstractLwM2MIntegrationDiffPortTest extends AbstractSecurityLwM2MIntegrationTest { |
|||
|
|||
@SpyBean |
|||
private RegistrationStore registrationStoreTest; |
|||
|
|||
protected void basicTestConnectionDifferentPort(Lwm2mDeviceProfileTransportConfiguration transportConfiguration, |
|||
String awaitAlias) throws Exception { |
|||
|
|||
createDeviceProfile(transportConfiguration); |
|||
createDevice(deviceCredentials, clientEndpoint); |
|||
createNewClient(security, null, COAP_CONFIG, true, clientEndpoint); |
|||
lwM2MTestClient.start(true); |
|||
await(awaitAlias) |
|||
.atMost(40, TimeUnit.SECONDS) |
|||
.until(() -> lwM2MTestClient.getClientStates().contains(ON_REGISTRATION_SUCCESS) || lwM2MTestClient.getClientStates().contains(ON_REGISTRATION_STARTED)); |
|||
Assert.assertTrue(lwM2MTestClient.getClientStates().containsAll(expectedStatusesRegistrationLwm2mSuccess)); |
|||
|
|||
await(awaitAlias) |
|||
.atMost(40, TimeUnit.SECONDS) |
|||
.until(() -> { |
|||
doAnswer(new Answer<Object>() { |
|||
@Override |
|||
public Object answer(InvocationOnMock invocation) throws Throwable { |
|||
Object[] arguments = invocation.getArguments(); |
|||
if (arguments.length > 0 && arguments[0] instanceof RegistrationUpdate) { |
|||
int portOld = ((RegistrationUpdate) arguments[0]).getPort(); |
|||
int portValueChange = 5; |
|||
arguments[0] = registrationUpdateNewPort((RegistrationUpdate) arguments[0], portValueChange); |
|||
int portNew = ((RegistrationUpdate) arguments[0]).getPort(); |
|||
Assert.assertEquals((portNew - portOld), portValueChange); |
|||
} |
|||
return invocation.callRealMethod(); |
|||
} |
|||
}).when(registrationStoreTest).updateRegistration(any(RegistrationUpdate.class)); |
|||
return lwM2MTestClient.getClientStates().contains(ON_UPDATE_SUCCESS); |
|||
}); |
|||
Assert.assertTrue(lwM2MTestClient.getClientStates().containsAll(expectedStatusesRegistrationLwm2mSuccessUpdate)); |
|||
} |
|||
|
|||
private RegistrationUpdate registrationUpdateNewPort (RegistrationUpdate update, int portValueChange) { |
|||
Integer portOld = update.getPort(); |
|||
Integer portNew = portOld + portValueChange; |
|||
log.warn("portOld: [{}], portNew: [{}]", portOld, portNew); |
|||
InetAddress addressOld = update.getAddress(); |
|||
InetSocketAddress socketAddressUpdate = new InetSocketAddress(addressOld, portNew); |
|||
SocketIdentity socketIdentity = new SocketIdentity(socketAddressUpdate); |
|||
LwM2mPeer sender = new IpPeer(new InetSocketAddress(addressOld, portNew), socketIdentity); |
|||
return new RegistrationUpdate(update.getRegistrationId(), sender, |
|||
update.getLifeTimeInSec(), update.getSmsNumber(), update.getBindingMode(), |
|||
update.getObjectLinks(), update.getAlternatePath(), |
|||
update.getSupportedContentFormats(), update.getSupportedObjects(), |
|||
update.getAvailableInstances(), update.getAdditionalAttributes(), |
|||
update.getApplicationData()); |
|||
} |
|||
} |
|||
@ -0,0 +1,45 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* 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.thingsboard.server.transport.lwm2m.security.diffPort; |
|||
|
|||
import org.junit.Test; |
|||
import org.thingsboard.server.common.data.device.profile.Lwm2mDeviceProfileTransportConfiguration; |
|||
|
|||
import static org.thingsboard.server.common.data.device.credentials.lwm2m.LwM2MSecurityMode.NO_SEC; |
|||
import static org.thingsboard.server.common.data.device.credentials.lwm2m.LwM2MSecurityMode.PSK; |
|||
import static org.thingsboard.server.transport.lwm2m.Lwm2mTestHelper.LwM2MProfileBootstrapConfigType.NONE; |
|||
|
|||
public class LwM2MIntegrationDiffPortTest extends AbstractLwM2MIntegrationDiffPortTest { |
|||
|
|||
@Test |
|||
public void testWithNoSecConnectLwm2mSuccess_AfterRegistration_UpdateRegistrationFromDifferentPort_Ok() throws Exception { |
|||
String awaitAlias = "await on client state (NoSec different port)"; |
|||
Lwm2mDeviceProfileTransportConfiguration transportConfiguration = getTransportConfiguration(OBSERVE_ATTRIBUTES_WITHOUT_PARAMS, getBootstrapServerCredentialsSecure(NO_SEC, NONE)); |
|||
initDeviceCredentialsNoSek(); |
|||
basicTestConnectionDifferentPort( |
|||
transportConfiguration, |
|||
awaitAlias); |
|||
} |
|||
@Test |
|||
public void testWithPskConnectLwm2mSuccess_AfterRegistration_UpdateRegistrationFromDifferentPort_Ok() throws Exception { |
|||
String awaitAlias = "await on client state (Psk different port)"; |
|||
Lwm2mDeviceProfileTransportConfiguration transportConfiguration = getTransportConfiguration(OBSERVE_ATTRIBUTES_WITHOUT_PARAMS, getBootstrapServerCredentialsSecure(PSK, NONE)); |
|||
initDeviceCredentialsPsk(); |
|||
basicTestConnectionDifferentPort( |
|||
transportConfiguration, |
|||
awaitAlias); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue