feedback

Author: manodnyab

plugins/core/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/telemetry/OpenedFileTypeMetricsTest.kt

@@ -3,35 +3,34 @@
 
 package software.aws.toolkits.jetbrains.services.telemetry
 
-import org.junit.After
-import org.junit.Before
-import org.junit.Test
-import org.junit.jupiter.api.Assertions.assertEquals
-import org.junit.jupiter.api.Assertions.assertTrue
+import org.assertj.core.api.Assertions.assertThat
+import org.junit.jupiter.api.AfterEach
+import org.junit.jupiter.api.BeforeEach
+import org.junit.jupiter.api.Test
 
 class OpenedFileTypeMetricsTest {
 
     private lateinit var service: OpenedFileTypesMetricsService
 
-    @Before
+    @BeforeEach
     fun setup() {
         service = OpenedFileTypesMetricsService()
     }
 
-    @After
+    @AfterEach
     fun teardown() {
         service.dispose()
     }
 
     @Test
     fun `test addToExistingTelemetryBatch with allowed extension`() {
         service.addToExistingTelemetryBatch("kt")
-        assertTrue(service.getOpenedFileTypes().contains(".kt"))
+        assertThat(service.getOpenedFileTypes().contains(".kt"))
     }
 
     @Test
     fun `test addToExistingTelemetryBatch with disallowed extension`() {
         service.addToExistingTelemetryBatch("txt")
-        assertEquals(service.getOpenedFileTypes(), emptySet<String>())
+        assertThat(service.getOpenedFileTypes().isEmpty())
     }
 }