包名重构

protocol/http/down_test.go

@@ -0,0 +1,154 @@
+package http
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"reflect"
+	"testing"
+	"time"
+)
+
+func TestDown(t *testing.T) {
+	type args struct {
+		request *Request
+	}
+	tests := []struct {
+		name string
+		args args
+	}{
+		{
+			"proxyee-down.ico",
+			args{
+				&Request{
+					"get",
+					"https://raw.githubusercontent.com/proxyee-down-org/proxyee-down/master/front/public/favicon.ico",
+					map[string]string{
+						"Host":            "github.com",
+						"User-Agent":      "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/72.0.3626.121 Safari/537.36",
+						"Accept":          "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
+						"Referer":         "http://github.com/proxyee-down-org/proxyee-down/releases",
+						"Accept-Encoding": "gzip, deflate, br",
+						"Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
+					},
+					nil,
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			os.Remove("favicon.ico")
+			err := Down(tt.args.request)
+			if err != nil {
+				t.Errorf("error down= %v", err)
+				return
+			}
+			downMd5 := fileMd5("favicon.ico")
+			os.Remove("favicon.ico")
+			if "8de7a6a2e786861013d61b77b2394012" != downMd5 {
+				t.Errorf("error md5= %v", downMd5)
+				return
+			}
+		})
+	}
+}
+
+func TestResolve(t *testing.T) {
+	// os.Setenv("HTTP_PROXY", "http://127.0.0.1:8888")
+	type args struct {
+		request *Request
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    *Response
+		wantErr bool
+	}{
+		{
+			"proxyee-down.ico",
+			args{
+				&Request{
+					"get",
+					"https://raw.githubusercontent.com/proxyee-down-org/proxyee-down/master/front/public/favicon.ico",
+					map[string]string{
+						"Host":            "github.com",
+						"User-Agent":      "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/72.0.3626.121 Safari/537.36",
+						"Accept":          "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
+						"Referer":         "http://github.com/proxyee-down-org/proxyee-down/releases",
+						"Accept-Encoding": "gzip, deflate, br",
+						"Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
+					},
+					nil,
+				},
+			},
+			&Response{"favicon.ico", 116093, true},
+			false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := Resolve(tt.args.request)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("Resolve() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("Resolve() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestTemp(t *testing.T) {
+	f, err := os.Create("e:/testbt/test.data")
+	if err != nil {
+		log.Fatal(err)
+	}
+	if err := f.Truncate(1024 * 1024 * 512); err != nil {
+		log.Fatal(err)
+	}
+	bts := []byte{1, 2, 3, 4, 5, 6, 7, 8, 9}
+	s := time.Now().UnixNano()
+	f.Write(bts)
+	fmt.Printf("write from 0 use: %d\n", time.Now().UnixNano()-s)
+	s = time.Now().UnixNano()
+	f.Seek(1024*1024*128, 0)
+	f.Write(bts)
+	fmt.Printf("write from 128M use %d\n", time.Now().UnixNano()-s)
+	s = time.Now().UnixNano()
+	f.Write(bts)
+	fmt.Printf("write from 128M+9b use %d\n", time.Now().UnixNano()-s)
+	s = time.Now().UnixNano()
+	f.Seek(0, 0)
+	f.Write(bts)
+	fmt.Printf("write from 0 agein use %d\n", time.Now().UnixNano()-s)
+	s = time.Now().UnixNano()
+	f.Seek(1024*1024*64, 0)
+	f.Write(bts)
+	fmt.Printf("write from agein 64M use %d\n", time.Now().UnixNano()-s)
+}
+
+func fileMd5(filePath string) string {
+	file, _ := os.Open(filePath)
+
+	// Tell the program to call the following function when the current function returns
+	defer file.Close()
+
+	// Open a new hash interface to write to
+	hash := md5.New()
+
+	// Copy the file in the hash interface and check for any error
+	if _, err := io.Copy(hash, file); err != nil {
+		return ""
+	}
+	return hex.EncodeToString(hash.Sum(nil))
+}
+
+func TestTemp2(t *testing.T) {
+	fmt.Println("123456789"[2:3])
+}