Daria Shevchenko
3 days ago
committed by
GitHub
GitHub
24 changed files with 515 additions and 18 deletions
@ -0,0 +1,82 @@ |
|||
/** |
|||
* Copyright © 2016-2026 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.coap; |
|||
|
|||
import org.eclipse.californium.core.server.resources.CoapExchange; |
|||
import org.junit.jupiter.api.BeforeEach; |
|||
import org.junit.jupiter.api.Test; |
|||
import org.junit.jupiter.api.extension.ExtendWith; |
|||
import org.mockito.Mock; |
|||
import org.mockito.junit.jupiter.MockitoExtension; |
|||
import org.thingsboard.server.coapserver.CoapServerContext; |
|||
import org.thingsboard.server.common.data.id.DeviceId; |
|||
import org.thingsboard.server.common.transport.TransportDeviceProfileCache; |
|||
import org.thingsboard.server.common.transport.TransportService; |
|||
import org.thingsboard.server.queue.discovery.PartitionService; |
|||
import org.thingsboard.server.transport.coap.client.DefaultCoapClientContext; |
|||
import org.thingsboard.server.transport.coap.client.TbCoapClientState; |
|||
import org.thingsboard.server.transport.coap.client.TbCoapObservationState; |
|||
|
|||
import static org.mockito.Mockito.mock; |
|||
import static org.mockito.Mockito.verifyNoInteractions; |
|||
import static org.mockito.Mockito.when; |
|||
|
|||
@ExtendWith(MockitoExtension.class) |
|||
public class CoapSessionListenerTest { |
|||
|
|||
@Mock |
|||
CoapServerContext config; |
|||
@Mock |
|||
CoapTransportContext transportContext; |
|||
@Mock |
|||
TransportService transportService; |
|||
@Mock |
|||
TransportDeviceProfileCache profileCache; |
|||
@Mock |
|||
PartitionService partitionService; |
|||
|
|||
private DefaultCoapClientContext coapClientContext; |
|||
|
|||
@BeforeEach |
|||
void setUp() { |
|||
coapClientContext = new DefaultCoapClientContext(config, transportContext, transportService, profileCache, partitionService); |
|||
} |
|||
|
|||
@Test |
|||
void onTenantDeleted_shouldNotInteractWithCore() { |
|||
// GIVEN
|
|||
TbCoapClientState state = mock(TbCoapClientState.class); |
|||
|
|||
TbCoapObservationState rpcObs = mock(TbCoapObservationState.class); |
|||
when(rpcObs.getToken()).thenReturn("rpc-token"); |
|||
when(rpcObs.getExchange()).thenReturn(mock(CoapExchange.class)); |
|||
|
|||
TbCoapObservationState attrsObs = mock(TbCoapObservationState.class); |
|||
when(attrsObs.getToken()).thenReturn("attr-token"); |
|||
when(attrsObs.getExchange()).thenReturn(mock(CoapExchange.class)); |
|||
when(state.getRpc()).thenReturn(rpcObs); |
|||
when(state.getAttrs()).thenReturn(attrsObs); |
|||
|
|||
DefaultCoapClientContext.CoapSessionListener listener = coapClientContext.new CoapSessionListener(state); |
|||
|
|||
// WHEN
|
|||
listener.onTenantDeleted(mock(DeviceId.class)); |
|||
|
|||
// THEN
|
|||
verifyNoInteractions(transportService, partitionService); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
/** |
|||
* Copyright © 2016-2026 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.common.transport; |
|||
|
|||
import lombok.Getter; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.queue.discovery.event.TbApplicationEvent; |
|||
|
|||
public final class TenantDeletedEvent extends TbApplicationEvent { |
|||
|
|||
private static final long serialVersionUID = -3801673609913423334L; |
|||
@Getter |
|||
private final TenantId tenantId; |
|||
|
|||
public TenantDeletedEvent(TenantId tenantId) { |
|||
super(new Object()); |
|||
this.tenantId = tenantId; |
|||
} |
|||
} |
|||
@ -0,0 +1,26 @@ |
|||
/** |
|||
* Copyright © 2016-2026 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.common.transport.session; |
|||
|
|||
public enum SessionCloseReason { |
|||
|
|||
TENANT_DELETED; |
|||
|
|||
public static boolean shouldNotifyCore(SessionCloseReason reason) { |
|||
return reason != TENANT_DELETED; |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue