Method Process

internal/output/options.go:40–76

Process generates formatted test code from the header and function signatures.

(head *models.Header, funcs []*models.Function)

// Process generates formatted test code from the header and function signatures.
func (o *Options) Process(head *models.Header, funcs []*models.Function) ([]byte, error) {
	o.render = render.New()

	switch {
	case o.providesTemplateDir():
		if err := o.render.LoadCustomTemplates(o.TemplateDir); err != nil {
			return nil, fmt.Errorf("loading custom templates: %v", err)
		}
	case o.providesTemplate():
		if err := o.render.LoadCustomTemplatesName(o.Template); err != nil {
			return nil, fmt.Errorf("loading custom templates of name: %v", err)
		}
	case o.providesTemplateData():
		o.render.LoadFromData(o.TemplateData)
	}

	//
	tf, err := ioutil.TempFile("", "gotests_")
	if err != nil {
		return nil, fmt.Errorf("ioutil.TempFile: %v", err)
	}
	defer tf.Close()
	defer os.Remove(tf.Name())

	// create physical copy of test
	b := &bytes.Buffer{}
	if err := o.writeTests(b, head, funcs); err != nil {
		return nil, err
	}

	// format file
	out, err := imports.Process(tf.Name(), b.Bytes(), nil)
	if err != nil {
		return nil, fmt.Errorf("imports.Process: %v", err)
	}
	return out, nil
}

func (o *Options) providesTemplateData() bool {
	return o != nil && len(o.TemplateData) > 0