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