go.mondoo.com/cnquery@v0.0.0-20231005093811-59568235f6ea/providers/os/resources/updates/win_updates_test.go (about)

     1  // Copyright (c) Mondoo, Inc.
     2  // SPDX-License-Identifier: BUSL-1.1
     3  
     4  package updates
     5  
     6  import (
     7  	"errors"
     8  	"strings"
     9  	"testing"
    10  
    11  	"github.com/stretchr/testify/assert"
    12  	"github.com/stretchr/testify/require"
    13  	"go.mondoo.com/cnquery/providers-sdk/v1/inventory"
    14  	"go.mondoo.com/cnquery/providers/os/connection/mock"
    15  	"go.mondoo.com/cnquery/providers/os/resources/powershell"
    16  )
    17  
    18  func TestWinOSUpdatesParser(t *testing.T) {
    19  	mock, err := mock.New("./testdata/updates_win.toml", &inventory.Asset{
    20  		Platform: &inventory.Platform{Name: "windows"},
    21  	})
    22  	if err != nil {
    23  		t.Fatal(err)
    24  	}
    25  
    26  	cmd := powershell.Encode(WINDOWS_QUERY_WSUS_AVAILABLE)
    27  	c, err := mock.RunCommand(cmd)
    28  	if err != nil {
    29  		t.Fatal(err)
    30  	}
    31  
    32  	m, err := ParseWindowsUpdates(c.Stdout)
    33  	assert.Nil(t, err)
    34  	assert.Equal(t, 6, len(m), "detected the right amount of packages")
    35  
    36  	pkg, err := findKb(m, "890830")
    37  	require.NoError(t, err)
    38  	assert.Equal(t, "890830", pkg.Name, "update id detected")
    39  	assert.Equal(t, "Windows Malicious Software Removal Tool x64 - March 2020 (KB890830)", pkg.Description, "update title detected")
    40  
    41  	pkg, err = findKb(m, "4538461")
    42  	require.NoError(t, err)
    43  	assert.Equal(t, "4538461", pkg.Name, "update id detected")
    44  	assert.Equal(t, "2020-03 Cumulative Update for Windows Server 2019 (1809) for x64-based Systems (KB4538461)", pkg.Description, "update title detected")
    45  
    46  	// check empty return
    47  	m, err = ParseWindowsUpdates(strings.NewReader(""))
    48  	assert.Nil(t, err)
    49  	assert.Equal(t, 0, len(m), "detected the right amount of packages")
    50  }
    51  
    52  func findKb(pkgs []OperatingSystemUpdate, name string) (OperatingSystemUpdate, error) {
    53  	for i := range pkgs {
    54  		if pkgs[i].Name == name {
    55  			return pkgs[i], nil
    56  		}
    57  	}
    58  
    59  	return OperatingSystemUpdate{}, errors.New("not found")
    60  }