From 95178d12ea2f874f1ea29439428c3533239c1dad Mon Sep 17 00:00:00 2001
From: Andrii Landiak <andriylandiak@gmail.com>
Date: Thu, 11 Jul 2024 14:48:57 +0300
Subject: [PATCH] Fix OAuth2EdgeTest

---
 .../org/thingsboard/server/edge/NotificationEdgeTest.java  | 1 -
 .../java/org/thingsboard/server/edge/OAuth2EdgeTest.java   | 7 +------
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/application/src/test/java/org/thingsboard/server/edge/NotificationEdgeTest.java b/application/src/test/java/org/thingsboard/server/edge/NotificationEdgeTest.java
index 890ec00913..6d4ebb3676 100644
--- a/application/src/test/java/org/thingsboard/server/edge/NotificationEdgeTest.java
+++ b/application/src/test/java/org/thingsboard/server/edge/NotificationEdgeTest.java
@@ -18,7 +18,6 @@ package org.thingsboard.server.edge;
 import com.google.protobuf.AbstractMessage;
 import org.junit.Assert;
 import org.junit.Test;
-import org.springframework.test.context.TestPropertySource;
 import org.thingsboard.common.util.JacksonUtil;
 import org.thingsboard.server.common.data.StringUtils;
 import org.thingsboard.server.common.data.notification.NotificationDeliveryMethod;
diff --git a/application/src/test/java/org/thingsboard/server/edge/OAuth2EdgeTest.java b/application/src/test/java/org/thingsboard/server/edge/OAuth2EdgeTest.java
index c25410b562..beae36718d 100644
--- a/application/src/test/java/org/thingsboard/server/edge/OAuth2EdgeTest.java
+++ b/application/src/test/java/org/thingsboard/server/edge/OAuth2EdgeTest.java
@@ -59,12 +59,7 @@ public class OAuth2EdgeTest extends AbstractEdgeTest {
         oAuth2Info.setEnabled(false);
         oAuth2Info.setEdgeEnabled(false);
         doPost("/api/oauth2/config", oAuth2Info, OAuth2Info.class);
-        Assert.assertTrue(edgeImitator.waitForMessages());
-        latestMessage = edgeImitator.getLatestMessage();
-        Assert.assertTrue(latestMessage instanceof OAuth2UpdateMsg);
-        oAuth2UpdateMsg = (OAuth2UpdateMsg) latestMessage;
-        result = JacksonUtil.fromString(oAuth2UpdateMsg.getEntity(), OAuth2Info.class, true);
-        Assert.assertEquals(oAuth2Info, result);
+        Assert.assertFalse(edgeImitator.waitForMessages(5));
 
         edgeImitator.ignoreType(OAuth2UpdateMsg.class);
         loginTenantAdmin();