endpoint as registryKey

Author: samgst-amazon

plugins/core/jetbrains-community/resources/META-INF/aws.toolkit.core.xml

@@ -77,6 +77,9 @@
                      restartRequired="true"/>
         <registryKey key="aws.toolkit.developerMode" description="Enables features to facilitate development of the toolkit" restartRequired="false"
                      defaultValue="false"/>
+        <registryKey key="aws.toolkit.notification.endpoint" description="Endpoint for AWS Toolkit notifications"
+                     defaultValue=""/>
+
 
         <notificationGroup id="aws.plugin.version.mismatch" displayType="STICKY_BALLOON" key="aws.settings.title"/>
 

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/notifications/NotificationPollingService.kt

@@ -7,12 +7,12 @@ import com.fasterxml.jackson.module.kotlin.readValue
 import com.intellij.openapi.Disposable
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.components.Service
+import com.intellij.openapi.util.registry.Registry
 import com.intellij.util.Alarm
 import com.intellij.util.AlarmFactory
 import com.intellij.util.io.HttpRequests
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.runBlocking
-import org.jetbrains.annotations.VisibleForTesting
 import software.aws.toolkits.core.utils.RemoteResolveParser
 import software.aws.toolkits.core.utils.RemoteResource
 import software.aws.toolkits.core.utils.error
@@ -40,23 +40,11 @@ object NotificationFileValidator : RemoteResolveParser {
 }
 
 object NotificationEndpoint {
-    private var overriddenEndpoint: String? = null
-
-    fun getEndpoint(): String = overriddenEndpoint ?: DEFAULT_ENDPOINT
-
-    @VisibleForTesting
-    fun setTestEndpoint(endpoint: String) {
-        overriddenEndpoint = endpoint
-    }
-
-    @VisibleForTesting
-    fun resetEndpoint() {
-        overriddenEndpoint = null
-    }
-
-    private const val DEFAULT_ENDPOINT = "" // TODO: Replace with actual endpoint
+    fun getEndpoint(): String =
+        Registry.get("aws.toolkit.notification.endpoint").asString()
 }
 
+
 @Service(Service.Level.APP)
 internal final class NotificationPollingService : Disposable {
     private val isFirstPoll = AtomicBoolean(true)