More tests

macadmins · Jul 11, 2024 · 2f78cf7 · 2f78cf7
1 parent 2736ee8
commit 2f78cf7
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tables/sofa/client_test.go b/tables/sofa/client_test.go
@@ -301,7 +301,7 @@ func TestDownloadData(t *testing.T) {
 func TestNewSofaClient(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithCacheDir(cwd))
+	client, err := NewSofaClient(WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 	assert.Equal(t, SofaV1URL, client.endpoint)
 	assert.NotNil(t, client.httpClient)
@@ -310,23 +310,23 @@ func TestNewSofaClient(t *testing.T) {
 func TestWithHTTPClient(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithHTTPClient(&http.Client{}), WithCacheDir(cwd))
+	client, err := NewSofaClient(WithHTTPClient(&http.Client{}), WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 	assert.NotNil(t, client.httpClient)
 }
 
 func TestWithEndpoint(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithURL("http://example.com"), WithCacheDir(cwd))
+	client, err := NewSofaClient(WithURL("http://example.com"), WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 	assert.Equal(t, "http://example.com", client.endpoint)
 }
 
 func TestSetCachePaths(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithCacheDir(cwd))
+	client, err := NewSofaClient(WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 
 	client.setCachePaths()
@@ -338,7 +338,7 @@ func TestSetCachePaths(t *testing.T) {
 func TestWithCacheDir(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithCacheDir(cwd))
+	client, err := NewSofaClient(WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 	defer os.Remove(client.cacheDir) //nolint:errcheck
 
@@ -350,7 +350,7 @@ func TestWithCacheDir(t *testing.T) {
 func TestEtagPath(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithCacheDir(cwd))
+	client, err := NewSofaClient(WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 
 	client.setCachePaths()
@@ -361,7 +361,7 @@ func TestEtagPath(t *testing.T) {
 func TestCreateCacheDir(t *testing.T) {
 	cwd, err := os.Getwd()
 	assert.NoError(t, err)
-	client, err := NewSofaClient(WithCacheDir(cwd))
+	client, err := NewSofaClient(WithCacheDir(cwd), WithUserAgent("test"))
 	assert.NoError(t, err)
 
 	err = client.createCacheDir()