crowdsec/pkg/cwhub/item_test.go

package cwhub

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestItemStatus(t *testing.T) {
	hub := envSetup(t)

	// get existing map
	x := hub.GetItemMap(COLLECTIONS)
	require.NotEmpty(t, x)

	// Get item: good and bad
	for k := range x {
		item := hub.GetItem(COLLECTIONS, k)
		require.NotNil(t, item)

		item.State.Installed = true
		item.State.UpToDate = false
		item.State.Tainted = false
		item.State.Downloaded = true

		txt := item.State.Text()
		require.Equal(t, "enabled,update-available", txt)

		item.State.Installed = true
		item.State.UpToDate = false
		item.State.Tainted = false
		item.State.Downloaded = false

		txt = item.State.Text()
		require.Equal(t, "enabled,local", txt)
	}

	stats := hub.ItemStats()
	require.Equal(t, []string{
		"Loaded: 2 parsers, 1 scenarios, 3 collections",
		"Unmanaged items: 3 local, 0 tainted",
	}, stats)
}

func TestGetters(t *testing.T) {
	hub := envSetup(t)

	// get non existing map
	empty := hub.GetItemMap("ratata")
	require.Nil(t, empty)

	// get existing map
	x := hub.GetItemMap(COLLECTIONS)
	require.NotEmpty(t, x)

	// Get item: good and bad
	for k := range x {
		empty := hub.GetItem(COLLECTIONS, k+"nope")
		require.Nil(t, empty)

		item := hub.GetItem(COLLECTIONS, k)
		require.NotNil(t, item)

		// Add item and get it
		item.Name += "nope"
		hub.addItem(item)

		newitem := hub.GetItem(COLLECTIONS, item.Name)
		require.NotNil(t, newitem)
	}
}