removed client isLocal()

This commit is contained in:
Josh Yan
2024-07-08 10:33:47 -07:00
parent 3aee405dfa
commit 8377dc48d0
3 changed files with 17 additions and 135 deletions

View File

@@ -1,8 +1,6 @@
package api
import (
"net/http"
"net/url"
"testing"
"github.com/ollama/ollama/envconfig"
@@ -48,80 +46,3 @@ func TestClientFromEnvironment(t *testing.T) {
})
}
}
// Test function
func TestIsLocal(t *testing.T) {
type test struct {
client *Client
want bool
err error
}
tests := map[string]test{
"localhost": {
client: func() *Client {
baseURL, _ := url.Parse("http://localhost:1234")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: true,
err: nil,
},
"127.0.0.1": {
client: func() *Client {
baseURL, _ := url.Parse("http://127.0.0.1:1234")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: true,
err: nil,
},
"example.com": {
client: func() *Client {
baseURL, _ := url.Parse("http://example.com:1111")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: false,
err: nil,
},
"8.8.8.8": {
client: func() *Client {
baseURL, _ := url.Parse("http://8.8.8.8:1234")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: false,
err: nil,
},
"empty host with port": {
client: func() *Client {
baseURL, _ := url.Parse("http://:1234")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: true,
err: nil,
},
"empty host without port": {
client: func() *Client {
baseURL, _ := url.Parse("http://")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: true,
err: nil,
},
"remote host without port": {
client: func() *Client {
baseURL, _ := url.Parse("http://example.com")
return &Client{base: baseURL, http: &http.Client{}}
}(),
want: false,
err: nil,
},
}
for name, tc := range tests {
t.Run(name, func(t *testing.T) {
got := tc.client.IsLocal()
if got != tc.want {
t.Errorf("test %s failed: got %v, want %v", name, got, tc.want)
}
})
}
}