mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-17 06:52:26 +02:00
ab8de19506
* Add back pointer Item.hub * Hub.enableItem() -> Item.enable() * Rename variable i -> idx (i is used for item instances) * Move Hub.purgeItem() -> Item.purge() * Move Hub.disableItem() -> Item.disable() * Move Hub.downloadItem() -> Item.download() * Move Hub.downloadLatest() -> Item.downloadLatest() * Move Hub.DownloadDataIfNeeded() -> Item.DownloadDataIfNeeded() * Move Hub.InstallItem() -> Item.Install() * Move Hub.RemoveItem() -> Item.Remove() * Move Hub.UpgradeItem() -> Item.Upgrade() * store hub items as pointers * No need to re-add items to the hub if we use pointers * Fix parameter calling order + regression test
68 lines
1.3 KiB
Go
68 lines
1.3 KiB
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.Installed = true
|
|
item.UpToDate = false
|
|
item.Tainted = false
|
|
item.Downloaded = true
|
|
|
|
txt, _ := item.Status()
|
|
require.Equal(t, "enabled,update-available", txt)
|
|
|
|
item.Installed = true
|
|
item.UpToDate = false
|
|
item.Tainted = false
|
|
item.Downloaded = false
|
|
|
|
txt, _ = item.Status()
|
|
require.Equal(t, "enabled,local", txt)
|
|
}
|
|
|
|
stats := hub.ItemStats()
|
|
require.Equal(t, []string{"Loaded: 2 parsers, 1 scenarios, 3 collections"}, 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.Items[item.Type][item.Name] = item
|
|
|
|
newitem := hub.GetItem(COLLECTIONS, item.Name)
|
|
require.NotNil(t, newitem)
|
|
}
|
|
}
|