github.com/hernad/nomad@v1.6.112/e2e/parameterized/parameterized.go (about)

     1  // Copyright (c) HashiCorp, Inc.
     2  // SPDX-License-Identifier: MPL-2.0
     3  
     4  package parameterized
     5  
     6  import (
     7  	"fmt"
     8  
     9  	"github.com/hernad/nomad/e2e/e2eutil"
    10  	"github.com/hernad/nomad/e2e/framework"
    11  	"github.com/hernad/nomad/helper/uuid"
    12  	"github.com/hernad/nomad/testutil"
    13  	"github.com/stretchr/testify/require"
    14  )
    15  
    16  type ParameterizedTest struct {
    17  	framework.TC
    18  	jobIDs []string
    19  }
    20  
    21  func init() {
    22  	framework.AddSuites(&framework.TestSuite{
    23  		Component:   "Parameterized",
    24  		CanRunLocal: true,
    25  		Cases: []framework.TestCase{
    26  			new(ParameterizedTest),
    27  		},
    28  	})
    29  }
    30  
    31  func (tc *ParameterizedTest) BeforeAll(f *framework.F) {
    32  	e2eutil.WaitForLeader(f.T(), tc.Nomad())
    33  }
    34  
    35  func (tc *ParameterizedTest) AfterEach(f *framework.F) {
    36  	nomadClient := tc.Nomad()
    37  	j := nomadClient.Jobs()
    38  
    39  	for _, id := range tc.jobIDs {
    40  		j.Deregister(id, true, nil)
    41  	}
    42  	_, err := e2eutil.Command("nomad", "system", "gc")
    43  	f.NoError(err)
    44  }
    45  
    46  func (tc *ParameterizedTest) TestParameterizedDispatch_Basic(f *framework.F) {
    47  	t := f.T()
    48  
    49  	uuid := uuid.Generate()
    50  	jobID := fmt.Sprintf("dispatch-%s", uuid[0:8])
    51  	tc.jobIDs = append(tc.jobIDs, jobID)
    52  
    53  	// register job
    54  	require.NoError(t, e2eutil.Register(jobID, "parameterized/input/simple.nomad"))
    55  
    56  	// force dispatch
    57  	dispatched := 4
    58  
    59  	for i := 0; i < dispatched; i++ {
    60  		require.NoError(t, e2eutil.Dispatch(jobID, map[string]string{"i": fmt.Sprintf("%v", i)}, ""))
    61  	}
    62  
    63  	testutil.WaitForResult(func() (bool, error) {
    64  		children, err := e2eutil.DispatchedJobs(jobID)
    65  		if err != nil {
    66  			return false, err
    67  		}
    68  
    69  		dead := 0
    70  		for _, c := range children {
    71  			if c["Status"] != "dead" {
    72  				return false, fmt.Errorf("expected periodic job to be dead")
    73  			}
    74  			dead++
    75  		}
    76  
    77  		if dead != dispatched {
    78  			return false, fmt.Errorf("expected %d but found %d children", dispatched, dead)
    79  		}
    80  
    81  		return true, nil
    82  	}, func(err error) {
    83  		require.NoError(t, err)
    84  	})
    85  
    86  	// Assert there are no pending children
    87  	summary, err := e2eutil.ChildrenJobSummary(jobID)
    88  	require.NoError(t, err)
    89  	require.Len(t, summary, 1)
    90  	require.Equal(t, summary[0]["Pending"], "0")
    91  	require.Equal(t, summary[0]["Running"], "0")
    92  	require.Equal(t, summary[0]["Dead"], fmt.Sprintf("%v", dispatched))
    93  }