main_test.go

/*
 Copyright 2020 Keyporttech Inc.
 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 Licens}ze 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"
	"net/http"
	"testing"
	"encoding/json"
	"io/ioutil"
)

type GiteaEvent struct {
	secret string `json:"secret"`
}

// tests validate payload
func TestValidatePayload(t *testing.T) {

  //load the sample request from files
	file, _ := ioutil.ReadFile("testEvent.json")
	giteaEvent := GiteaEvent{}
	_ = json.Unmarshal([]byte(file), &giteaEvent)
	defaultBody := string(file)

	const defaultSignature = "1887c76452d0b1b4798a775ca9f9e62d57ae535708e089d6674aef4fb8de3e89"
	secretKey := []byte("YOUR_secret")
	validTests := []struct {
		signature   string
		eventID     string
		event       string
		wantEventID string
		wantEvent   string
		wantPayload string
	}{
		// The following tests generate expected errors:
		//{signature: "yo"},     // Signature not hex string
		//{signature: "012345"}, // Invalid signature

		{
			signature:   defaultSignature,
			eventID:     "ping test",
			event:       "ping",
			wantEventID: "ping test",
			wantEvent:   "ping",
			wantPayload: defaultBody,
		},
		{
			signature:   defaultSignature,
			eventID:     "push test",
			event:       "push",
			wantEvent:   "push",
			wantPayload: defaultBody,
		},
		{
			signature:   defaultSignature,
			eventID:     "pull_request",
			event:       "pull_request",
			wantEvent:   "pull_request",
			wantPayload: defaultBody,
		},
	}

	// invalidTests := []struct {
	// 	signature   string
	// 	eventID     string
	// 	event       string
	// 	wantEventID string
	// 	wantEvent   string
	// 	wantPayload string
	// }{
	// 	// The following tests generate expected errors:
	// 	{},                    // Missing signature
	// 	{signature: "yo"},     // Signature not hex string
	// 	{signature: "012345"}, // Invalid signature
	// 	// The following tests expect err=nil:
	// 	{
	// 		signature:   defaultSignature,
	// 		eventID:     "caesar-salad",
	// 		event:       "ping",
	// 		wantEventID: "caesar-salad",
	// 		wantEvent:   "ping",
	// 		wantPayload: defaultBody,
	// 	},
	// 	{
	// 		signature:   defaultSignature,
	// 		event:       "ping",
	// 		wantEvent:   "ping",
	// 		wantPayload: defaultBody,
	// 	},
	// 	{
	// 		signature:   "b1f8020f5b4cd42042f807dd939015c4a418bc1ff7f604dd55b0a19b5d953d9b",
	// 		event:       "ping",
	// 		wantEvent:   "ping",
	// 		wantPayload: defaultBody,
	// 	},
	// }

	for _, test := range validTests {
		buf := bytes.NewBufferString(defaultBody)
		req, err := http.NewRequest("GET", "http://localhost/event", buf)
		if err != nil {
			t.Fatalf("NewRequest: %v", err)
		}
		if test.signature != "" {
			req.Header.Set(signatureHeader, test.signature)
		}
		req.Header.Set("Content-Type", "application/json")

		got, err := ValidatePayload(req, secretKey)
		if err != nil {
			if test.wantPayload != "" {
				t.Errorf("err = %v", err)
			}
			continue
		}
		if string(got) != test.wantPayload {
			t.Errorf("Event Id: %s - ValidatePayload = %q, want %q", test.eventID, got, test.wantPayload)
		}
	}
}