global.go

package frisby

import (
	"errors"
	"fmt"
	"net/http"
	"os"

	"github.com/mozillazg/request"
)

type global_data struct {
	Req  *request.Request
	Errs map[string][]error

	NumRequest int
	NumAsserts int
	NumErrored int

	PrintProgressName bool
	PrintProgressDot  bool
}

func init() {
	Global.Req = request.NewRequest(new(http.Client))
	Global.Errs = make(map[string][]error, 0)
	Global.PrintProgressDot = true
}

// Set BasicAuth values for the coming request
func (G *global_data) BasicAuth(user, passwd string) *global_data {
	G.Req.BasicAuth = request.BasicAuth{user, passwd}
	return G
}

// Set Proxy URL for the coming request
func (G *global_data) SetProxy(url string) *global_data {
	G.Req.Proxy = url
	return G
}

// Set a Header value for the coming request
func (G *global_data) SetHeader(key, value string) *global_data {
	if G.Req.Headers == nil {
		G.Req.Headers = make(map[string]string)
	}
	G.Req.Headers[key] = value
	return G
}

// Set several Headers for the coming request
func (G *global_data) SetHeaders(headers map[string]string) *global_data {
	if G.Req.Headers == nil {
		G.Req.Headers = make(map[string]string)
	}
	for key, value := range headers {
		G.Req.Headers[key] = value
	}
	return G
}

// Set a Cookie value for the coming request
func (G *global_data) SetCookie(key, value string) *global_data {
	if G.Req.Cookies == nil {
		G.Req.Cookies = make(map[string]string)
	}
	G.Req.Cookies[key] = value
	return G
}

// Set several Cookie values for the coming request
func (G *global_data) SetCookies(cookies map[string]string) *global_data {
	if G.Req.Cookies == nil {
		G.Req.Cookies = make(map[string]string)
	}
	for key, value := range cookies {
		G.Req.Cookies[key] = value
	}
	return G
}

// Set a Gorm data for the coming request
func (G *global_data) SetData(key, value string) *global_data {
	if G.Req.Data == nil {
		G.Req.Data = make(map[string]string)
	}
	G.Req.Data[key] = value
	return G
}

// Set several Gorm data for the coming request
func (G *global_data) SetDatas(datas map[string]string) *global_data {
	if G.Req.Data == nil {
		G.Req.Data = make(map[string]string)
	}
	for key, value := range datas {
		G.Req.Data[key] = value
	}
	return G
}

// Set a url Param for the coming request
func (G *global_data) SetParam(key, value string) *global_data {
	if G.Req.Params == nil {
		G.Req.Params = make(map[string]string)
	}
	G.Req.Params[key] = value
	return G
}

// Set several url Param for the coming request
func (G *global_data) SetParams(params map[string]string) *global_data {
	if G.Req.Params == nil {
		G.Req.Params = make(map[string]string)
	}
	for key, value := range params {
		G.Req.Params[key] = value
	}
	return G
}

// Set the JSON body for the coming request
func (G *global_data) SetJson(json interface{}) *global_data {
	G.Req.Json = json
	return G
}

// Add a file to the Gorm data for the coming request
func (G *global_data) AddFile(filename string) *global_data {
	file, err := os.Open(filename)
	if err != nil {
		G.AddError("Global", err.Error())
		fmt.Println("Error adding file to global")
	} else {
		fileField := request.FileField{"file", filename, file}
		G.Req.Files = append(G.Req.Files, fileField)
	}
	return G
}

// Manually add an error, if you need to
func (G *global_data) AddError(name, err_str string) *global_data {
	G.NumErrored++
	err := errors.New(err_str)
	G.Errs[name] = append(G.Errs[name], err)
	return G
}

// Get all errors for the global_data object
//
// This function should be called last
func (G *global_data) Errors() map[string][]error {
	return G.Errs
}

// Prints a report for the FrisbyGlobal Object
//
// If there are any errors, they will all be printed as well
func (G *global_data) PrintReport() *global_data {
	fmt.Printf("\nFor %d requests made\n", G.NumRequest)
	if len(G.Errs) == 0 {
		fmt.Printf("  All tests passed\n")
	} else {
		fmt.Printf("  FAILED  [%d/%d]\n", G.NumErrored, G.NumAsserts)
		for key, val := range G.Errs {
			fmt.Printf("      [%s]\n", key)
			for _, e := range val {
				fmt.Println("        - ", e)
			}
		}
	}

	return G
}