go.mondoo.com/cnquery@v0.0.0-20231005093811-59568235f6ea/explorer/bundle_test.go (about)

     1  // Copyright (c) Mondoo, Inc.
     2  // SPDX-License-Identifier: BUSL-1.1
     3  
     4  package explorer
     5  
     6  import (
     7  	"os"
     8  	"testing"
     9  
    10  	"github.com/stretchr/testify/assert"
    11  	"github.com/stretchr/testify/require"
    12  )
    13  
    14  func TestBundleLoad(t *testing.T) {
    15  	t.Run("load bundle from file", func(t *testing.T) {
    16  		bundle, err := BundleFromPaths("../examples/os.mql.yaml")
    17  		require.NoError(t, err)
    18  		assert.Equal(t, 1, len(bundle.Packs))
    19  		assert.Equal(t, 3, len(bundle.Packs[0].Queries))
    20  
    21  		// ensure that the uid is generated
    22  		assert.True(t, len(bundle.Packs[0].Queries[0].Uid) > 0)
    23  	})
    24  
    25  	t.Run("load bundle from memory", func(t *testing.T) {
    26  		data, err := os.ReadFile("../examples/os.mql.yaml")
    27  		require.NoError(t, err)
    28  		bundle, err := BundleFromYAML(data)
    29  		require.NoError(t, err)
    30  		assert.Equal(t, 1, len(bundle.Packs))
    31  		assert.Equal(t, 3, len(bundle.Packs[0].Queries))
    32  
    33  		// ensure that the uid is generated
    34  		assert.True(t, len(bundle.Packs[0].Queries[0].Uid) > 0)
    35  	})
    36  }