-
Notifications
You must be signed in to change notification settings - Fork 7
/
errors.go
78 lines (68 loc) · 2.46 KB
/
errors.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
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
/******************************************************************************
*
* Copyright 2019 SAP SE
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
******************************************************************************/
package test
import (
"encoding/json"
"fmt"
"testing"
"github.com/sapcc/keppel/internal/keppel"
)
// ErrorCode wraps keppel.RegistryV2ErrorCode with an implementation of the
// assert.HTTPResponseBody interface.
type ErrorCode keppel.RegistryV2ErrorCode
// AssertResponseBody implements the assert.HTTPResponseBody interface.
func (e ErrorCode) AssertResponseBody(t *testing.T, requestInfo string, responseBody []byte) bool {
t.Helper()
wrapped := ErrorCodeWithMessage{keppel.RegistryV2ErrorCode(e), ""}
return wrapped.AssertResponseBody(t, requestInfo, responseBody)
}
// ErrorCodeWithMessage extends ErrorCode with an expected detail message.
type ErrorCodeWithMessage struct {
Code keppel.RegistryV2ErrorCode
Message string
}
// AssertResponseBody implements the assert.HTTPResponseBody interface.
func (e ErrorCodeWithMessage) AssertResponseBody(t *testing.T, requestInfo string, responseBody []byte) bool {
t.Helper()
var data struct {
Errors []struct {
Code keppel.RegistryV2ErrorCode `json:"code"`
Message string `json:"message"`
} `json:"errors"`
}
err := json.Unmarshal(responseBody, &data)
if err != nil {
t.Errorf("%s: cannot decode JSON: %s", requestInfo, err.Error())
t.Logf("\tresponse body = %q", string(responseBody))
return false
}
expectedStr := string(e.Code)
if e.Message != "" {
expectedStr = fmt.Sprintf("%s with message: %s", e.Code, e.Message)
}
matches := len(data.Errors) == 1 && data.Errors[0].Code == e.Code
if matches {
matches = e.Message == "" || data.Errors[0].Message == e.Message
}
if !matches {
t.Errorf(requestInfo + ": got unexpected error")
t.Logf("\texpected = %q\n", expectedStr)
t.Logf("\tactual = %q\n", string(responseBody))
}
return matches
}