mirror of
				https://gitea.com/Lydanne/buildx.git
				synced 2025-11-04 01:53:42 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			45 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			45 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package store
 | 
						|
 | 
						|
import (
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/docker/buildx/util/platformutil"
 | 
						|
	"github.com/stretchr/testify/require"
 | 
						|
)
 | 
						|
 | 
						|
func TestNodeGroupUpdate(t *testing.T) {
 | 
						|
	t.Parallel()
 | 
						|
 | 
						|
	ng := &NodeGroup{}
 | 
						|
	err := ng.Update("foo", "foo0", []string{"linux/amd64"}, true, false, []string{"--debug"}, "", nil)
 | 
						|
	require.NoError(t, err)
 | 
						|
 | 
						|
	err = ng.Update("foo1", "foo1", []string{"linux/arm64", "linux/arm/v7"}, true, true, nil, "", nil)
 | 
						|
	require.NoError(t, err)
 | 
						|
 | 
						|
	require.Equal(t, len(ng.Nodes), 2)
 | 
						|
 | 
						|
	// update
 | 
						|
	err = ng.Update("foo", "foo2", []string{"linux/amd64", "linux/arm"}, true, false, nil, "", nil)
 | 
						|
	require.NoError(t, err)
 | 
						|
 | 
						|
	require.Equal(t, len(ng.Nodes), 2)
 | 
						|
	require.Equal(t, []string{"linux/amd64", "linux/arm/v7"}, platformutil.Format(ng.Nodes[0].Platforms))
 | 
						|
	require.Equal(t, []string{"linux/arm64"}, platformutil.Format(ng.Nodes[1].Platforms))
 | 
						|
 | 
						|
	require.Equal(t, "foo2", ng.Nodes[0].Endpoint)
 | 
						|
	require.Equal(t, []string{"--debug"}, ng.Nodes[0].BuildkitdFlags)
 | 
						|
	require.Equal(t, []string(nil), ng.Nodes[1].BuildkitdFlags)
 | 
						|
 | 
						|
	// duplicate endpoint
 | 
						|
	err = ng.Update("foo1", "foo2", nil, true, false, nil, "", nil)
 | 
						|
	require.Error(t, err)
 | 
						|
	require.Contains(t, err.Error(), "duplicate endpoint")
 | 
						|
 | 
						|
	err = ng.Leave("foo")
 | 
						|
	require.NoError(t, err)
 | 
						|
 | 
						|
	require.Equal(t, len(ng.Nodes), 1)
 | 
						|
	require.Equal(t, []string{"linux/arm64"}, platformutil.Format(ng.Nodes[0].Platforms))
 | 
						|
}
 |