robot_test.go

// Copyright (c) Huawei Technologies Co., Ltd. 2024. All rights reserved.
//
// 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 main

import (
	"bytes"
	"errors"
	"github.com/agiledragon/gomonkey/v2"
	kafka "github.com/opensourceways/kafka-lib/agent"
	"github.com/opensourceways/robot-framework-lib/framework"
	"github.com/sirupsen/logrus"
	"github.com/stretchr/testify/assert"
	"net/http"
	"net/http/httptest"
	"testing"
)

const (
	headerEventType            = "X-GitCode-Event"
	headerEventTypeValue       = "Note Hook"
	headerEventGUID            = "X-GitCode-Delivery"
	headerEventGUIDValue       = "gsadiuoady"
	headerEventToken           = "X-GitCode-Signature-256"
	headerEventTokenValue      = "sha256=ed14c459bd15a2460ae7dedc058ce6624bbb32f5b3266dfa237d9b02b118d7fb"
	headerUserAgent            = "User-Agent"
	headerUserAgentValue       = "git-gitcode-hook"
	headerContentTypeName      = "Content-Type"
	headerContentTypeJsonValue = "application/json"

	mockRequestBody          = "fihoagdshajbolkhasdb"
	mockWebHookCorrectSecret = "*****"
	mockWebHookWrongSecret   = "**************"
	mockRobotUserAgentValue  = "robot"
)

func TestDelivery(t *testing.T) {
	d := delivery{
		topic:     "t1",
		userAgent: mockRobotUserAgentValue,
		hmac:      []byte(mockWebHookWrongSecret),
		log:       framework.NewLogger().WithField("component", component),
	}

	w := httptest.NewRecorder()
	req, _ := http.NewRequest(http.MethodPost, "http://localhost:8080/case1", bytes.NewBufferString(mockRequestBody))
	req.Header.Set(headerUserAgent, headerUserAgentValue)
	req.Header.Set(headerContentTypeName, headerContentTypeJsonValue)
	req.Header.Set(headerEventType, headerEventTypeValue)
	req.Header.Set(headerEventToken, headerEventTokenValue)
	req.Header.Set(headerEventGUID, headerEventGUIDValue)
	d.ServeHTTP(w, req)

	assert.Equal(t, http.StatusUnauthorized, w.Result().StatusCode)

	patch := gomonkey.ApplyFunc(kafka.Publish, func(topic string, header map[string]string, msg []byte) error {
		return nil
	})
	defer patch.Reset()
	d.hmac = []byte(mockWebHookCorrectSecret)
	w1 := httptest.NewRecorder()
	req1, _ := http.NewRequest(http.MethodPost, "http://localhost:8080/case1", bytes.NewBufferString(mockRequestBody))
	req1.Header.Set(headerUserAgent, headerUserAgentValue)
	req1.Header.Set(headerContentTypeName, headerContentTypeJsonValue)
	req1.Header.Set(headerEventType, headerEventTypeValue)
	req1.Header.Set(headerEventToken, headerEventTokenValue)
	req1.Header.Set(headerEventGUID, headerEventGUIDValue)
	logrus.SetLevel(logrus.DebugLevel)

	d.ServeHTTP(w1, req1)

	assert.Equal(t, http.StatusOK, w1.Result().StatusCode)
}

func TestDeliveryError(t *testing.T) {
	d := delivery{
		topic:     "t2",
		userAgent: mockRobotUserAgentValue,
		hmac:      []byte(mockWebHookCorrectSecret),
		log:       framework.NewLogger().WithField("component", component),
	}

	w := httptest.NewRecorder()
	req, _ := http.NewRequest(http.MethodPost, "http://localhost:8080/case2", bytes.NewBufferString(mockRequestBody))
	req.Header.Set(headerUserAgent, headerUserAgentValue)
	req.Header.Set(headerContentTypeName, headerContentTypeJsonValue)
	req.Header.Set(headerEventType, headerEventTypeValue)
	req.Header.Set(headerEventToken, headerEventTokenValue)
	req.Header.Set(headerEventGUID, headerEventGUIDValue)

	patch := gomonkey.ApplyFunc(kafka.Publish, func(topic string, header map[string]string, msg []byte) error {
		return errors.New("mock kafka error")
	})
	defer patch.Reset()

	logrus.SetLevel(logrus.DebugLevel)
	d.ServeHTTP(w, req)
	assert.Equal(t, http.StatusOK, w.Result().StatusCode)
}