package main

import (
	"errors"
	"fmt"
	"strconv"
	"strings"
)

func CheckResults(testCase *TestCase) (bool, error) {

	// Check headers
	for k, v := range testCase.Expect.Header.Contains {
		fmt.Printf("Header contains check '%s'='%s'\n", k, v)

		if k == "http_status" { // http status is a special case
			statusCode, err := strconv.Atoi(v)
			if err != nil {
				error := fmt.Sprintf("unable to convert http_status expected value '%s' to int : %s", v, err)
				return false, errors.New(error)
			}
			if statusCode != testCase.ResultStatusCode {
				return false, nil
			}
		} else { // any other contains check
			checkFound := false
			for hKey, hVal := range testCase.ResultHeaders {
				if k == hKey {
					fmt.Printf("Found header key '%s', checking value '%s' matches test definition '%s'\n", hKey, hVal, v)

					for i := range hVal {
						if strings.Contains(hVal[i], v) {
							fmt.Printf("Found match '%s' = '%s'\n", hVal[i], v)
							checkFound = true
						}
					}
				}
			}

			if !checkFound {
				fmt.Printf("Expected to find header key-value of '%s:%s' but no matching value was found\n", k, v)
				return false, nil
			}
		}
	}

	return true, nil
}