telegraf/plugins/inputs/snmp/snmp_mocks_generate.go

// +build generate

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"os"
	"os/exec"
	"strings"
)

// This file is a generator used to generate the mocks for the commands used by the tests.

// These are the commands to be mocked.
var mockedCommands = [][]string{
	{"snmptranslate", "-Td", "-Ob", "-m", "all", ".1.0.0.0"},
	{"snmptranslate", "-Td", "-Ob", "-m", "all", ".1.0.0.1.1"},
	{"snmptranslate", "-Td", "-Ob", "-m", "all", ".1.0.0.1.2"},
	{"snmptranslate", "-Td", "-Ob", "-m", "all", "1.0.0.1.1"},
	{"snmptranslate", "-Td", "-Ob", "-m", "all", ".1.0.0.0.1.1"},
	{"snmptranslate", "-Td", "-Ob", "-m", "all", ".1.0.0.0.1.1.0"},
	{"snmptranslate", "-Td", "-Ob", "-m", "all", ".999"},
	{"snmptranslate", "-Td", "-Ob", "TEST::server"},
	{"snmptranslate", "-Td", "-Ob", "TEST::server.0"},
	{"snmptranslate", "-Td", "-Ob", "TEST::testTable"},
	{"snmptranslate", "-Td", "-Ob", "TEST::connections"},
	{"snmptranslate", "-Td", "-Ob", "TEST::latency"},
	{"snmptranslate", "-Td", "-Ob", "TEST::hostname"},
	{"snmptranslate", "-Td", "-Ob", "IF-MIB::ifPhysAddress.1"},
	{"snmptranslate", "-Td", "-Ob", "BRIDGE-MIB::dot1dTpFdbAddress.1"},
	{"snmptranslate", "-Td", "-Ob", "TCP-MIB::tcpConnectionLocalAddress.1"},
	{"snmptranslate", "-Td", "TEST::testTable.1"},
	{"snmptable", "-Ch", "-Cl", "-c", "public", "127.0.0.1", "TEST::testTable"},
}

type mockedCommandResult struct {
	stdout    string
	stderr    string
	exitError bool
}

func main() {
	if err := generate(); err != nil {
		fmt.Fprintf(os.Stderr, "error: %s\n", err)
		os.Exit(1)
	}
}

func generate() error {
	f, err := os.OpenFile("snmp_mocks_test.go", os.O_RDWR, 0644)
	if err != nil {
		return err
	}
	br := bufio.NewReader(f)
	var i int64
	for l, err := br.ReadString('\n'); err == nil; l, err = br.ReadString('\n') {
		i += int64(len(l))
		if l == "// BEGIN GO GENERATE CONTENT\n" {
			break
		}
	}
	f.Truncate(i)
	f.Seek(i, 0)

	fmt.Fprintf(f, "var mockedCommandResults = map[string]mockedCommandResult{\n")

	for _, cmd := range mockedCommands {
		ec := exec.Command(cmd[0], cmd[1:]...)
		out := bytes.NewBuffer(nil)
		err := bytes.NewBuffer(nil)
		ec.Stdout = out
		ec.Stderr = err
		ec.Env = []string{
			"MIBDIRS=+./testdata",
		}

		var mcr mockedCommandResult
		if err := ec.Run(); err != nil {
			if err, ok := err.(*exec.ExitError); !ok {
				mcr.exitError = true
			} else {
				return fmt.Errorf("executing %v: %s", cmd, err)
			}
		}
		mcr.stdout = string(out.Bytes())
		mcr.stderr = string(err.Bytes())
		cmd0 := strings.Join(cmd, "\000")
		mcrv := fmt.Sprintf("%#v", mcr)[5:] // trim `main.` prefix
		fmt.Fprintf(f, "%#v: %s,\n", cmd0, mcrv)
	}
	f.Write([]byte("}\n"))
	f.Close()

	return exec.Command("gofmt", "-w", "snmp_mocks_test.go").Run()
}