kraken

1
// Copyright (c) 2016-2019 Uber Technologies, Inc.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
// See the License for the specific language governing permissions and
13
// limitations under the License.
14
package blobserver
15

16
import (
17
	"net/http"
18
	"testing"
19

20
	"github.com/uber/kraken/utils/handler"
21
	"github.com/stretchr/testify/require"
22
)
23

24
func TestParseContentRangeHeaderBadRequests(t *testing.T) {
25
	tests := []struct {
26
		description string
27
		value       string
28
	}{
29
		{"empty value", ""},
30
		{"invalid format", "blah"},
31
		{"invalid start", "blah-5"},
32
		{"invalid end", "5-blah"},
33
	}
34
	for _, test := range tests {
35
		t.Run(test.description, func(t *testing.T) {
36
			require := require.New(t)
37

38
			h := http.Header{}
39
			h.Add("Content-Range", test.value)
40
			start, end, err := parseContentRange(h)
41
			require.Error(err)
42
			require.Equal(http.StatusBadRequest, err.(*handler.Error).GetStatus())
43
			require.Equal(int64(0), start)
44
			require.Equal(int64(0), end)
45
		})
46
	}
47
}
48