21846: applied select to context menu open
[arvados.git] / services / keep-web / ranges_test.go
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 package keepweb
6
7 import (
8         "fmt"
9         "net/http"
10         "net/http/httptest"
11
12         "git.arvados.org/arvados.git/sdk/go/arvadosclient"
13         "git.arvados.org/arvados.git/sdk/go/arvadostest"
14         "git.arvados.org/arvados.git/sdk/go/keepclient"
15         check "gopkg.in/check.v1"
16 )
17
18 func (s *IntegrationSuite) TestRanges(c *check.C) {
19         blocksize := 1000000
20         var uuid string
21         {
22                 testdata := make([]byte, blocksize)
23                 for i := 0; i < blocksize; i++ {
24                         testdata[i] = byte(' ')
25                 }
26                 copy(testdata[1:4], []byte("foo"))
27                 arv, err := arvadosclient.MakeArvadosClient()
28                 c.Assert(err, check.Equals, nil)
29                 arv.ApiToken = arvadostest.ActiveToken
30                 kc, err := keepclient.MakeKeepClient(arv)
31                 c.Assert(err, check.Equals, nil)
32                 loc, _, err := kc.PutB(testdata[:])
33                 c.Assert(err, check.Equals, nil)
34                 loc2, _, err := kc.PutB([]byte{'Z'})
35                 c.Assert(err, check.Equals, nil)
36
37                 mtext := fmt.Sprintf(". %s %s %s %s %s 1:%d:testdata.bin 0:1:space.txt\n", loc, loc, loc, loc, loc2, blocksize*4)
38                 coll := map[string]interface{}{}
39                 err = arv.Create("collections",
40                         map[string]interface{}{
41                                 "collection": map[string]interface{}{
42                                         "name":          "test data for keep-web TestRanges",
43                                         "manifest_text": mtext,
44                                 },
45                         }, &coll)
46                 c.Assert(err, check.Equals, nil)
47                 uuid = coll["uuid"].(string)
48                 defer arv.Delete("collections", uuid, nil, nil)
49         }
50
51         url := mustParseURL("http://" + uuid + ".collections.example.com/testdata.bin")
52         for _, trial := range []struct {
53                 header     string
54                 expectObey bool
55                 expectBody string
56         }{
57                 {"0-2", true, "foo"},
58                 {"-2", true, " Z"},
59                 {"1-4", true, "oo  "},
60                 {"z-y", false, ""},
61                 {"1000000-1000003", true, "foo "},
62                 {"999999-1000003", true, " foo "},
63                 {"2000000-2000003", true, "foo "},
64                 {"1999999-2000002", true, " foo"},
65                 {"3999998-3999999", true, " Z"},
66                 {"3999998-4000004", true, " Z"},
67                 {"3999998-", true, " Z"},
68         } {
69                 c.Logf("trial: %#v", trial)
70                 resp := httptest.NewRecorder()
71                 req := &http.Request{
72                         Method:     "GET",
73                         URL:        url,
74                         Host:       url.Host,
75                         RequestURI: url.RequestURI(),
76                         Header: http.Header{
77                                 "Authorization": {"OAuth2 " + arvadostest.ActiveToken},
78                                 "Range":         {"bytes=" + trial.header},
79                         },
80                 }
81                 s.handler.ServeHTTP(resp, req)
82                 if trial.expectObey {
83                         c.Check(resp.Code, check.Equals, http.StatusPartialContent)
84                         c.Check(resp.Body.Len(), check.Equals, len(trial.expectBody))
85                         if resp.Body.Len() > 1000 {
86                                 c.Check(resp.Body.String()[:1000]+"[...]", check.Equals, trial.expectBody)
87                         } else {
88                                 c.Check(resp.Body.String(), check.Equals, trial.expectBody)
89                         }
90                 } else {
91                         c.Check(resp.Code, check.Equals, http.StatusRequestedRangeNotSatisfiable)
92                 }
93         }
94 }