rtest.go

package main

import (
	"bufio"
	"fmt"

	"io/ioutil"
	"net/http"
	"net/url"

	"strconv"
	"strings"

	"github.com/rawoke083/ColorPrint"
	"log"
	"os"

	"errors"
)

type TestCase struct {
	HttpMethod       string
	Url              string
	HttpReturnCode   string
	ResponseTXTCheck string
	Pass             bool
}

func (test *TestCase) runATest(mparams map[string]string) bool {

	for i, k := range mparams {
		test.Url = strings.Replace(test.Url, i, k, -1)
	}

	ColorPrint.ColWrite("\n\nTEST:"+ColorPrint.ToColor(test.HttpMethod, ColorPrint.CL_LIGHT_CYAN)+" "+test.Url, ColorPrint.CL_WHITE)

	err := errors.New("")
	resp := new(http.Response)

	if test.HttpMethod == "POST" {
		u, _ := url.Parse(test.Url)
		ColorPrint.ColWrite("  ::: ARGS:"+u.RawQuery, ColorPrint.CL_WHITE)
		q, _ := url.ParseQuery(u.RawQuery)
		resp, err = http.PostForm(test.Url, q)
	} else {
		resp, err = http.Get(test.Url)
	}
	if err != nil {
		// handle error
		ColorPrint.ColWrite(fmt.Sprintf("\n\n=====>HTTP-ERROR:%s", test.Url), ColorPrint.CL_RED)
		return false
	}
	defer resp.Body.Close()

	test.Pass = true

	http_code := strconv.Itoa(resp.StatusCode)

	body, _ := ioutil.ReadAll(resp.Body)
	s := string(body)

	if strings.TrimSpace(test.HttpReturnCode) != http_code {

		test.Pass = false
		//ColorPrint.ColWrite("\n=>FAILED  - HttpCode Excepted |"+test.HttpReturnCode+"| but got |"+http_code+"|"+s, ColorPrint.CL_RED)
		ColorPrint.ColWrite("\n=>FAILED  - HttpCode Excepted |"+test.HttpReturnCode+"| but got |"+http_code+"|", ColorPrint.CL_RED)
		return false
	}

	if len(test.ResponseTXTCheck) > 1 {

		if !strings.Contains(s, test.ResponseTXTCheck) {
			test.Pass = false

			ColorPrint.ColWrite("\n=>FAILED  - ResponseText ("+test.ResponseTXTCheck+") not found. "+s, ColorPrint.CL_RED)
			return false

		}

	}

	return true
}

func runTestSuite(testCases []TestCase, mparams map[string]string) int {

	var testRunCount int
	var testOKCount int

	for _, test := range testCases {

		testRunCount++
		if test.runATest(mparams) {
			testOKCount++
		}

	} //end for

	s_total_test := fmt.Sprintf("\n\nTotal Test %d", testRunCount)
	s_total_test_ok := fmt.Sprintf("\nTest(s) OK %d", testOKCount)
	s_total_test_failed := fmt.Sprintf("\nTest(s) Failed %d\n", int(testRunCount-testOKCount))

	ColorPrint.ColWrite(s_total_test, ColorPrint.CL_LIGHT_BLUE)
	ColorPrint.ColWrite(s_total_test_ok, ColorPrint.CL_LIGHT_GREEN)
	ColorPrint.ColWrite(s_total_test_failed, ColorPrint.CL_RED)

	return (testRunCount - testOKCount)

}

func LoadTest(filename string) []TestCase {

	//	valid_http_methods := []string{"GET", "POST", "PUT", "DELETE"}

	f, err := os.Open(filename)
	if err != nil {
		log.Fatal(err)
	}

	//max of 64 test for now...
	testCaseList := [64]TestCase{}
	testCount := 0

	scanner := bufio.NewScanner(f)
	for scanner.Scan() {

		if strings.HasPrefix(scanner.Text(), "#") {
			continue
		}

		fields := strings.Split(scanner.Text(), "|")
		field_count := len(fields)

		if field_count < 2 {
			continue
		}

		testCaseList[testCount].HttpMethod = fields[0]
		testCaseList[testCount].Url = fields[1]
		testCaseList[testCount].HttpReturnCode = fields[2]
		if field_count > 3 {

			testCaseList[testCount].ResponseTXTCheck = fields[3]
		}
		testCount++

	} //for scanner

	return testCaseList[0:testCount]

}

func printUsage() {

	ColorPrint.ColWrite("Usage:\n", ColorPrint.CL_WHITE)
	ColorPrint.ColWrite("\nrtest filename=<url-list-file> [-DStringToReplace=NewString] [-DMoreStringToReplace=MoreNewString]\n\n", ColorPrint.CL_WHITE)
}

func main() {
	ColorPrint.ColWrite("\n###### REST Test ########\n", ColorPrint.CL_YELLOW)

	var mparams = make(map[string]string)

	var fileName string

	for i := range os.Args {

		if strings.Contains(os.Args[i], "-D") {

			fields := strings.Split(strings.Replace(os.Args[i], "-D", "", -1), "=")
			mparams[fields[0]] = fields[1]
		}

		if strings.Contains(os.Args[i], "filename=") {
			fields := strings.Split(os.Args[i], "=")
			fileName = fields[1]
		}

	}

	if len(fileName) < 1 {

		fmt.Println("\nError:No filename\n")
		printUsage()
		os.Exit(1)
	}

	os.Exit(runTestSuite(LoadTest(fileName), mparams))
}