forked from ContentSquare/chproxy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
io_test.go
29 lines (26 loc) · 1.66 KB
/
io_test.go
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
package main
import (
"bytes"
"io"
"net/http/httptest"
"testing"
)
func TestCachedReadCloser(t *testing.T) {
b := makeQuery(1000)
crc := &cachedReadCloser{
ReadCloser: io.NopCloser(bytes.NewReader(b)),
}
req := httptest.NewRequest("POST", "http://localhost", crc)
res, err := io.ReadAll(req.Body)
if err != nil {
t.Fatalf("cannot obtain response: %s", err)
}
if string(res) != string(b) {
t.Fatalf("unexpected query read %q; expecting %q", res, b)
}
expectedStart := "SELECT column col0, col1, col2, col3, col4, col5, col6, col7, col8, col9, col10, col11, col12, col13, col14, col15, col16, col17, col18, col19, col20, col21, col22, col23, col24, col25, col26, col27, col28, col29, col30, col31, col32, col33, col34, col35, col36, col37, col38, col39, col40, col41, col42, col43, col44, col45, col46, col47, col48, col49, col50, col51, col52, col53, col54, col55, col56, col57, col58, col59, col60, col61, col62, col63, col64, col65, col66, col67, col68, col69, col70, col71, col72, col73, col74, col75, col76, col77, col78, col79, col80, col81, col82, col83, col84, col85, col86, col87, col88, col89, col90, col91, col92, col93, col94, col95, col96, col97, col98, col99, col100, col101, col102, col103, col104, col105, col106, col107, col108, col109, col110, col111, col112, col113, col114, col115, col116, col117, col118, col119, col120, col121, col122, col123, col124, col125, col126, col127, col128, col129, col130, col131, col132, col133, col134, col135, col136, col137, col138, col139, ..."
start := crc.String()
if start != expectedStart {
t.Fatalf("unexpected query start read: (%d) %q; expecting (%d) %q", len(start), start, len(expectedStart), expectedStart)
}
}