github.com/e154/smart-home@v0.17.2-0.20240311175135-e530a6e5cd45/system/terminal/command_test.go (about)

     1  // This file is part of the Smart Home
     2  // Program complex distribution https://github.com/e154/smart-home
     3  // Copyright (C) 2024, Filippov Alex
     4  //
     5  // This library is free software: you can redistribute it and/or
     6  // modify it under the terms of the GNU Lesser General Public
     7  // License as published by the Free Software Foundation; either
     8  // version 3 of the License, or (at your option) any later version.
     9  //
    10  // This library is distributed in the hope that it will be useful,
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    13  // Library General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU Lesser General Public
    16  // License along with this library.  If not, see
    17  // <https://www.gnu.org/licenses/>.
    18  
    19  package terminal
    20  
    21  import (
    22  	"bytes"
    23  	"github.com/spf13/cobra"
    24  	"strings"
    25  	"testing"
    26  )
    27  
    28  func executeCommandC(root *cobra.Command, args ...string) (c *cobra.Command, output string, err error) {
    29  	buf := new(bytes.Buffer)
    30  	root.SetOut(buf)
    31  	root.SetErr(buf)
    32  	root.SetArgs(args)
    33  
    34  	c, err = root.ExecuteC()
    35  
    36  	return c, buf.String(), err
    37  }
    38  
    39  func executeCommand(root *cobra.Command, args ...string) (output string, err error) {
    40  	_, output, err = executeCommandC(root, args...)
    41  	return output, err
    42  }
    43  
    44  const onetwo = "one two"
    45  
    46  func TestSingleCommand(t *testing.T) {
    47  	var rootCmdArgs []string
    48  	rootCmd := &cobra.Command{
    49  		Use:  "root",
    50  		Args: cobra.ExactArgs(2),
    51  		Run:  func(_ *cobra.Command, args []string) { rootCmdArgs = args },
    52  	}
    53  	aCmd := &cobra.Command{Use: "a", Args: cobra.NoArgs, Run: emptyRun}
    54  	bCmd := &cobra.Command{Use: "b", Args: cobra.NoArgs, Run: emptyRun}
    55  	rootCmd.AddCommand(aCmd, bCmd)
    56  
    57  	output, err := executeCommand(rootCmd, "one", "two")
    58  	if output != "" {
    59  		t.Errorf("Unexpected output: %v", output)
    60  	}
    61  	if err != nil {
    62  		t.Errorf("Unexpected error: %v", err)
    63  	}
    64  
    65  	got := strings.Join(rootCmdArgs, " ")
    66  	if got != onetwo {
    67  		t.Errorf("rootCmdArgs expected: %q, got: %q", onetwo, got)
    68  	}
    69  }