summaryrefslogtreecommitdiff
path: root/vendor/github.com/utahta/go-openuri/openuri_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/utahta/go-openuri/openuri_test.go')
-rw-r--r--vendor/github.com/utahta/go-openuri/openuri_test.go74
1 files changed, 74 insertions, 0 deletions
diff --git a/vendor/github.com/utahta/go-openuri/openuri_test.go b/vendor/github.com/utahta/go-openuri/openuri_test.go
new file mode 100644
index 0000000..7f653e6
--- /dev/null
+++ b/vendor/github.com/utahta/go-openuri/openuri_test.go
@@ -0,0 +1,74 @@
+package openuri
+
+import (
+ "io/ioutil"
+ "net/http"
+ "strings"
+ "testing"
+)
+
+type dummyRoundTripper struct{}
+
+func (d *dummyRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
+ return &http.Response{Status: "dummy", Body: ioutil.NopCloser(strings.NewReader("body dummy"))}, nil
+}
+
+func TestNew(t *testing.T) {
+ _, err := New()
+ if err != nil {
+ t.Error(err)
+ }
+}
+
+func TestWithHTTPClient(t *testing.T) {
+ c, err := New(WithHTTPClient(&http.Client{Transport: &dummyRoundTripper{}}))
+ if err != nil {
+ t.Error(err)
+ }
+
+ resp, _ := c.httpClient.Get("test")
+ if resp.Status != "dummy" {
+ t.Errorf("Expected status dummy, got %s", resp.Status)
+ }
+}
+
+func TestOpen_File(t *testing.T) {
+ o, err := Open("./openuri.go")
+ if err != nil {
+ t.Error(err)
+ }
+
+ b, err := ioutil.ReadAll(o)
+ if err != nil {
+ t.Error(err)
+ }
+
+ if !strings.HasPrefix(string(b), "package openuri") {
+ t.Errorf("Expected open file, go %s", string(b))
+ }
+}
+
+func TestOpen_URL(t *testing.T) {
+ tests := []struct {
+ url string
+ }{
+ {"http://example.com"},
+ {"https://example.com"},
+ }
+
+ for _, test := range tests {
+ o, err := Open(test.url, WithHTTPClient(&http.Client{Transport: &dummyRoundTripper{}}))
+ if err != nil {
+ t.Error(err)
+ }
+
+ b, err := ioutil.ReadAll(o)
+ if err != nil {
+ t.Error(err)
+ }
+
+ if !strings.HasPrefix(string(b), "body dummy") {
+ t.Errorf("Expected open file, go %s", string(b))
+ }
+ }
+}