Update plugin_test.go

Signed-off-by: Laura Lin <[email protected]>

flyteplugins/go/tasks/plugins/webapi/athena/plugin_test.go

@@ -22,3 +22,18 @@ func TestCreateTaskInfo(t *testing.T) {
 	assert.Len(t, taskInfo.ExternalResources, 1)
 	assert.Equal(t, taskInfo.ExternalResources[0].ExternalID, "query_id")
 }
+
+
+func TestCreateTaskInfoGovAWS(t *testing.T) {
+	taskInfo := createTaskInfo("query_id", awsSdk.Config{
+		Region: "us-gov-east-1",
+	})
+	assert.EqualValues(t, []*idlCore.TaskLog{
+		{
+			Uri:  "https://us-gov-east-1.console.amazonaws-us-gov.com/athena/home?force&region=us-gov-east-1#query/history/query_id",
+			Name: "Athena Query Console",
+		},
+	}, taskInfo.Logs)
+	assert.Len(t, taskInfo.ExternalResources, 1)
+	assert.Equal(t, taskInfo.ExternalResources[0].ExternalID, "query_id")
+}