mirror of
https://github.com/ollama/ollama.git
synced 2026-04-26 18:55:53 +02:00
removed client isLocal()
This commit is contained in:
@@ -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)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user