Merge pull request #5170 from stellar/pass_user_agent_in_ledger_backend

ingest/ledgerbackend - Pass user agent in ledger backend

historyarchive/archive_pool_test.go

@@ -0,0 +1,39 @@
+// Copyright 2016 Stellar Development Foundation and contributors. Licensed
+// under the Apache License, Version 2.0. See the COPYING file at the root
+// of this distribution or at http://www.apache.org/licenses/LICENSE-2.0
+
+package historyarchive
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/stellar/go/support/storage"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestConfiguresHttpUserAgentForArchivePool(t *testing.T) {
+	var userAgent string
+	var archiveURLs []string
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		userAgent = r.Header["User-Agent"][0]
+		w.WriteHeader(http.StatusOK)
+	}))
+	defer server.Close()
+	archiveURLs = append(archiveURLs, server.URL)
+
+	archiveOptions := ArchiveOptions{
+		ConnectOptions: storage.ConnectOptions{
+			UserAgent: "uatest",
+		},
+	}
+
+	archivePool, err := NewArchivePool(archiveURLs, archiveOptions)
+	assert.NoError(t, err)
+
+	ok, err := archivePool.BucketExists(EmptyXdrArrayHash())
+	assert.True(t, ok)
+	assert.NoError(t, err)
+	assert.Equal(t, userAgent, "uatest")
+}

ingest/ledgerbackend/captive_core_backend.go

@@ -179,7 +179,8 @@ func NewCaptive(config CaptiveCoreConfig) (*CaptiveStellarCore, error) {
 			NetworkPassphrase:   config.NetworkPassphrase,
 			CheckpointFrequency: config.CheckpointFrequency,
 			ConnectOptions: storage.ConnectOptions{
-				Context: config.Context,
+				Context:   config.Context,
+				UserAgent: config.UserAgent,
 			},
 		},
 	)