azure-container-networking/npm/cmd/debug_test.go

package main

import (
	"bytes"
	"io"
	"testing"

	dataplane "github.com/Azure/azure-container-networking/npm/pkg/dataplane/debug"
	"github.com/Azure/azure-container-networking/npm/util"
	"github.com/stretchr/testify/require"
)

const (
	iptableSaveFile = "../pkg/dataplane/testdata/iptablesave-v1"
	npmCacheFile    = "../pkg/dataplane/testdata/npmcachev1.json"
	nonExistingFile = "non-existing-iptables-file"

	npmCacheFlag         = "-c"
	iptablesSaveFileFlag = "-i"
	dstFlag              = "-d"
	srcFlag              = "-s"
	unknownShorthandFlag = "-z"

	testIP1 = "10.224.0.87" // from npmCacheWithCustomFormat.json
	testIP2 = "10.224.0.20" // ditto

	debugCmdString          = "debug"
	convertIPTableCmdString = "convertiptable"
	getTuplesCmdString      = "gettuples"
	parseIPTableCmdString   = "parseiptable"
)

type testCases struct {
	name    string
	args    []string
	wantErr bool
}

func testCommand(t *testing.T, tests []*testCases) {
	for _, tt := range tests {
		tt := tt
		t.Run(tt.name, func(t *testing.T) {
			rootCMD := NewRootCmd()
			b := bytes.NewBufferString("")
			rootCMD.SetOut(b)
			rootCMD.SetArgs(tt.args)
			err := rootCMD.Execute()

			if tt.wantErr {
				require.Error(t, err)
				return
			}

			require.NoError(t, err)

			out, err := io.ReadAll(b)
			require.NoError(t, err)
			if tt.wantErr {
				require.NotEmpty(t, out)
			} else {
				require.Empty(t, out)
			}
		})
	}
}

func concatArgs(baseArgs []string, args ...string) []string {
	return append(baseArgs, args...)
}

func TestPrettyPrint(t *testing.T) {
	c := &dataplane.Converter{}

	iptables, err := c.GetProtobufRulesFromIptableFile(
		util.IptablesFilterTable,
		npmCacheFile,
		iptableSaveFile,
	)

	require.NoError(t, err)
	require.NoError(t, prettyPrintIPTables(iptables))
}