filedrop/utils_test.go at master · foxcpp/filedrop · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package filedrop_test

import (
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"testing"

	"github.com/foxcpp/filedrop"
	_ "github.com/mattn/go-sqlite3"
)

var TestDB = os.Getenv("TEST_DB")
var TestDSN = os.Getenv("TEST_DSN")

var file = `Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow
Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow
Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow Meow`

func initServ(conf filedrop.Config) *filedrop.Server {
	tempDir, err := ioutil.TempDir("", "filedrop-tests-")
	if err != nil {
		panic(err)
	}
	conf.StorageDir = tempDir

	conf.DB.Driver = TestDB
	conf.DB.DSN = TestDSN

	// This is meant for DB debugging.
	if TestDB == "" || TestDSN == "" {
		log.Println("Using sqlite3 DB in temporary directory.")
		conf.DB.Driver = "sqlite3"
		conf.DB.DSN = filepath.Join(tempDir, "index.db")
	}

	serv, err := filedrop.New(conf)
	if err != nil {
		panic(err)
	}
	if testing.Verbose() {
		serv.DebugLogger = log.New(os.Stderr, "filedrop/debug ", log.Lshortfile)
	}
	return serv
}

func cleanServ(serv *filedrop.Server) {
	if _, err := serv.DB.Exec(`DROP TABLE filedrop`); err != nil {
		panic(err)
	}
	serv.Close()
	os.Remove(serv.Conf.StorageDir)
}

func doPOST(t *testing.T, c *http.Client, url string, contentType string, reqBody io.Reader) []byte {
	t.Helper()

	resp, err := c.Post(url, contentType, reqBody)
	if err != nil {
		t.Error("POST:", err)
		t.FailNow()
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		t.Error("ioutil.ReadAll:", err)
		t.FailNow()
	}
	if resp.StatusCode/100 != 2 {
		t.Error("POST: HTTP", resp.StatusCode, resp.Status)
		t.Error("Body:", string(body))
		t.FailNow()
	}
	return body
}

func doPOSTFail(t *testing.T, c *http.Client, url string, contentType string, reqBody io.Reader) int {
	t.Helper()

	resp, err := c.Post(url, contentType, reqBody)
	if err != nil {
		t.Error("POST:", err)
		t.FailNow()
	}
	defer resp.Body.Close()
	if resp.StatusCode/100 == 2 {
		t.Error("POST: HTTP", resp.StatusCode, resp.Status)
		t.FailNow()
	}
	return resp.StatusCode
}

func doGET(t *testing.T, c *http.Client, url string) []byte {
	t.Helper()

	resp, err := c.Get(url)
	if err != nil {
		t.Error("GET:", err)
		t.FailNow()
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		t.Error("ioutil.ReadAll:", err)
		t.FailNow()
	}
	if resp.StatusCode/100 != 2 {
		t.Error("GET: HTTP", resp.Status)
		t.Error("Body:", string(body))
		t.FailNow()
	}
	return body
}

func doGETFail(t *testing.T, c *http.Client, url string) int {
	t.Helper()

	resp, err := c.Get(url)
	if err != nil {
		t.Error("GET:", err)
		t.FailNow()
	}
	defer resp.Body.Close()
	if resp.StatusCode/100 == 2 {
		t.Error("GET: HTTP", resp.StatusCode, resp.Status)
		t.FailNow()
	}
	return resp.StatusCode
}