chore(http): test update of scraper targets
Signed-off-by: Leonardo Di Donato <leodidonato@gmail.com>pull/10616/head
parent
4932f2aadd
commit
e79be29224
|
@ -134,8 +134,7 @@ func (h *ScraperHandler) handleGetScraperTargets(w http.ResponseWriter, r *http.
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func decodeScraperTargetUpdateRequest(ctx context.Context, r *http.Request) (
|
func decodeScraperTargetUpdateRequest(ctx context.Context, r *http.Request) (*platform.ScraperTarget, error) {
|
||||||
*platform.ScraperTarget, error) {
|
|
||||||
update := &platform.ScraperTarget{}
|
update := &platform.ScraperTarget{}
|
||||||
if err := json.NewDecoder(r.Body).Decode(update); err != nil {
|
if err := json.NewDecoder(r.Body).Decode(update); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
|
@ -476,7 +476,166 @@ func TestService_handlePostScraperTarget(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestService_handlePatchScraperTarget(t *testing.T) {
|
func TestService_handlePatchScraperTarget(t *testing.T) {
|
||||||
|
type fields struct {
|
||||||
|
Service platform.ScraperTargetStoreService
|
||||||
|
}
|
||||||
|
|
||||||
|
type args struct {
|
||||||
|
id string
|
||||||
|
update *platform.ScraperTarget
|
||||||
|
}
|
||||||
|
|
||||||
|
type wants struct {
|
||||||
|
statusCode int
|
||||||
|
contentType string
|
||||||
|
body string
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields fields
|
||||||
|
args args
|
||||||
|
wants wants
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "update a scraper target",
|
||||||
|
fields: fields{
|
||||||
|
Service: &mock.ScraperTargetStoreService{
|
||||||
|
UpdateTargetF: func(ctx context.Context, t *platform.ScraperTarget) (*platform.ScraperTarget, error) {
|
||||||
|
if t.ID == targetOneID {
|
||||||
|
return t, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, fmt.Errorf("not found")
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
id: targetOneIDString,
|
||||||
|
update: &platform.ScraperTarget{
|
||||||
|
ID: targetOneID,
|
||||||
|
Name: "name",
|
||||||
|
BucketName: "buck",
|
||||||
|
Type: platform.PrometheusScraperType,
|
||||||
|
URL: "www.example.url",
|
||||||
|
OrgName: "orgg",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
statusCode: http.StatusOK,
|
||||||
|
contentType: "application/json; charset=utf-8",
|
||||||
|
body: fmt.Sprintf(
|
||||||
|
`
|
||||||
|
{
|
||||||
|
"id":"%[1]s",
|
||||||
|
"name":"name",
|
||||||
|
"type":"prometheus",
|
||||||
|
"url":"www.example.url",
|
||||||
|
"org":"orgg",
|
||||||
|
"bucket":"buck",
|
||||||
|
"links":{
|
||||||
|
"self":"/api/v2/scrapertargets/%[1]s"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
`,
|
||||||
|
targetOneIDString,
|
||||||
|
),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
// fixme > scraper update needs to be done through an intermediate struct (eg., patchScraperRequest) to check almost 1 field is updated
|
||||||
|
// todo > introduce a platform.ScraperTargetUpdate struct
|
||||||
|
// {
|
||||||
|
// name: "update a scraper target with empty request body",
|
||||||
|
// fields: fields{
|
||||||
|
// Service: &mock.ScraperTargetStoreService{
|
||||||
|
// UpdateTargetF: func(ctx context.Context, upd *platform.ScraperTarget) (*platform.ScraperTarget, error) {
|
||||||
|
// return nil, fmt.Errorf("not found")
|
||||||
|
// },
|
||||||
|
// },
|
||||||
|
// },
|
||||||
|
// args: args{
|
||||||
|
// id: targetTwoIDString,
|
||||||
|
// },
|
||||||
|
// wants: wants{
|
||||||
|
// statusCode: http.StatusBadRequest,
|
||||||
|
// },
|
||||||
|
// },
|
||||||
|
{
|
||||||
|
name: "scraper target not found",
|
||||||
|
fields: fields{
|
||||||
|
Service: &mock.ScraperTargetStoreService{
|
||||||
|
UpdateTargetF: func(ctx context.Context, upd *platform.ScraperTarget) (*platform.ScraperTarget, error) {
|
||||||
|
return nil, &platform.Error{
|
||||||
|
Code: platform.ENotFound,
|
||||||
|
Msg: platform.ErrScraperTargetNotFound,
|
||||||
|
}
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
id: targetOneIDString,
|
||||||
|
update: &platform.ScraperTarget{
|
||||||
|
ID: targetOneID,
|
||||||
|
Name: "name",
|
||||||
|
BucketName: "buck",
|
||||||
|
Type: platform.PrometheusScraperType,
|
||||||
|
URL: "www.example.url",
|
||||||
|
OrgName: "orgg",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wants: wants{
|
||||||
|
statusCode: http.StatusNotFound,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
h := NewScraperHandler()
|
||||||
|
h.ScraperStorageService = tt.fields.Service
|
||||||
|
|
||||||
|
var err error
|
||||||
|
st := make([]byte, 0)
|
||||||
|
if tt.args.update != nil {
|
||||||
|
st, err = json.Marshal(*tt.args.update)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to unmarshal scraper target: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println(string(st))
|
||||||
|
|
||||||
|
r := httptest.NewRequest("GET", "http://any.tld", bytes.NewReader(st))
|
||||||
|
|
||||||
|
r = r.WithContext(context.WithValue(
|
||||||
|
context.Background(),
|
||||||
|
httprouter.ParamsKey,
|
||||||
|
httprouter.Params{
|
||||||
|
{
|
||||||
|
Key: "id",
|
||||||
|
Value: tt.args.id,
|
||||||
|
},
|
||||||
|
}))
|
||||||
|
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
|
||||||
|
h.handlePatchScraperTarget(w, r)
|
||||||
|
|
||||||
|
res := w.Result()
|
||||||
|
content := res.Header.Get("Content-Type")
|
||||||
|
body, _ := ioutil.ReadAll(res.Body)
|
||||||
|
|
||||||
|
if res.StatusCode != tt.wants.statusCode {
|
||||||
|
t.Errorf("%q. handlePatchScraperTarget() = %v, want %v", tt.name, res.StatusCode, tt.wants.statusCode)
|
||||||
|
}
|
||||||
|
if tt.wants.contentType != "" && content != tt.wants.contentType {
|
||||||
|
t.Errorf("%q. handlePatchScraperTarget() = %v, want %v", tt.name, content, tt.wants.contentType)
|
||||||
|
}
|
||||||
|
if eq, diff, _ := jsonEqual(string(body), tt.wants.body); tt.wants.body != "" && !eq {
|
||||||
|
t.Errorf("%q. handlePatchScraperTarget() = ***%s***", tt.name, diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func initScraperService(f platformtesting.TargetFields, t *testing.T) (platform.ScraperTargetStoreService, string, func()) {
|
func initScraperService(f platformtesting.TargetFields, t *testing.T) (platform.ScraperTargetStoreService, string, func()) {
|
||||||
|
|
Loading…
Reference in New Issue