github.com/greenboxal/deis@v1.12.1/deisctl/backend/fleet/status_test.go (about)

     1  package fleet
     2  
     3  import (
     4  	"bytes"
     5  	"fmt"
     6  	"sync"
     7  	"testing"
     8  	"time"
     9  
    10  	"github.com/coreos/fleet/machine"
    11  	"github.com/coreos/fleet/schema"
    12  )
    13  
    14  type mockStatusCommandRunner struct {
    15  	validUnits []string
    16  }
    17  
    18  func (mockStatusCommandRunner) LocalCommand(string) (int, error) {
    19  	return 0, nil
    20  }
    21  
    22  func (m mockStatusCommandRunner) RemoteCommand(cmd string, addr string, timeout time.Duration) (int, error) {
    23  	if addr != "1.1.1.1" || timeout != 0 {
    24  		return -1, fmt.Errorf("Got %s %d, which is unexpected", cmd, addr, timeout)
    25  	}
    26  
    27  	for _, unit := range m.validUnits {
    28  		if fmt.Sprintf("systemctl status -l %s", unit) == cmd {
    29  			return 0, nil
    30  		}
    31  	}
    32  
    33  	return -1, fmt.Errorf("Didn't find command %s to match with units %v", cmd, m.validUnits)
    34  }
    35  
    36  func TestStatus(t *testing.T) {
    37  	t.Parallel()
    38  
    39  	testMachines := []machine.MachineState{
    40  		machine.MachineState{
    41  			ID:       "test-1",
    42  			PublicIP: "1.1.1.1",
    43  			Metadata: nil,
    44  			Version:  "",
    45  		},
    46  	}
    47  
    48  	testUnits := []*schema.Unit{
    49  		&schema.Unit{
    50  			Name:         "deis-router@1.service",
    51  			CurrentState: "loaded",
    52  			MachineID:    "test-1",
    53  		},
    54  		&schema.Unit{
    55  			Name:         "deis-router@2.service",
    56  			CurrentState: "loaded",
    57  			MachineID:    "test-1",
    58  		},
    59  		&schema.Unit{
    60  			Name:      "deis-controller.service",
    61  			MachineID: "test-1",
    62  		},
    63  	}
    64  
    65  	testWriter := bytes.Buffer{}
    66  
    67  	c := &FleetClient{Fleet: &stubFleetClient{testUnits: testUnits, testMachineStates: testMachines,
    68  		unitsMutex: &sync.Mutex{}}, errWriter: &testWriter, runner: mockStatusCommandRunner{
    69  		validUnits: []string{"deis-router@1.service", "deis-router@2.service"}}}
    70  
    71  	err := c.Status("router")
    72  
    73  	if err != nil {
    74  		t.Error(err)
    75  	}
    76  
    77  	commandErr := testWriter.String()
    78  
    79  	if commandErr != "" {
    80  		t.Error(commandErr)
    81  	}
    82  
    83  	c.runner = mockStatusCommandRunner{validUnits: []string{}}
    84  	err = c.Status("foo")
    85  
    86  	actualErr := err.Error()
    87  	expectedErr := "could not find unit: foo"
    88  
    89  	if actualErr != expectedErr {
    90  		t.Errorf("Expected %s, Got %s", expectedErr, actualErr)
    91  	}
    92  
    93  	err = c.Status("controller")
    94  
    95  	if err != nil {
    96  		t.Error(err)
    97  	}
    98  
    99  	expectedErr = "Unit deis-controller.service does not appear to be running.\n"
   100  	commandErr = testWriter.String()
   101  
   102  	if commandErr != expectedErr {
   103  		t.Errorf("Expected '%s', Got '%s'", expectedErr, commandErr)
   104  	}
   105  }