mirror of
https://github.com/activecm/rita
synced 2026-06-08 13:02:45 +00:00
Naomi Kramer
77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
package cmd_test
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/activecm/rita/v5/cmd"
|
|
"github.com/activecm/rita/v5/database"
|
|
"github.com/activecm/rita/v5/util"
|
|
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func (c *CmdTestSuite) TestFormatListTable() {
|
|
require := require.New(c.T())
|
|
|
|
_, err := cmd.RunImportCmd(time.Now(), c.cfg, afero.NewOsFs(), "../test_data/valid_tsv", "dnscat", false, true)
|
|
require.NoError(err, "importing dnscat data should not produce an error")
|
|
|
|
_, err = cmd.RunImportCmd(time.Now(), c.cfg, afero.NewOsFs(), "../test_data/proxy", "proxy", false, true)
|
|
require.NoError(err, "importing proxy data should not produce an error")
|
|
|
|
// get droplet subnet
|
|
_, dropletSubnet, err := net.ParseCIDR("64.225.56.201/32")
|
|
require.NoError(err)
|
|
|
|
// update config to include droplet subnet
|
|
c.cfg.Filtering.InternalSubnets = append(c.cfg.Filtering.InternalSubnets, util.Subnet{IPNet: dropletSubnet})
|
|
c.cfg.Filtering.FilterExternalToInternal = false
|
|
|
|
_, err = cmd.RunImportCmd(time.Now(), c.cfg, afero.NewOsFs(), "../test_data/missing_host/2024-04-19", "fake_rolling", true, true)
|
|
require.NoError(err, "importing fake_rolling data should not produce an error")
|
|
|
|
_, err = cmd.RunImportCmd(time.Now(), c.cfg, afero.NewOsFs(), "../test_data/missing_host/2024-04-20", "fake_rolling", true, false)
|
|
require.NoError(err, "importing 2nd fake_rolling data should not produce an error")
|
|
|
|
// connect to server
|
|
server, err := database.ConnectToServer(context.Background(), c.cfg)
|
|
require.NoError(err)
|
|
|
|
dbs, err := server.ListImportDatabases()
|
|
require.NoError(err)
|
|
|
|
output := cmd.FormatListTable(dbs)
|
|
|
|
lines := strings.Split(output.String(), "\n")
|
|
require.Len(lines, 7)
|
|
lines = lines[3:6]
|
|
|
|
expectedDBs := []struct {
|
|
name string
|
|
rolling string
|
|
tsRange string
|
|
}{
|
|
{name: "fake_rolling", rolling: "true", tsRange: "2024-04-18 20:07 - 2024-04-20 23:59"},
|
|
{name: "proxy", rolling: "false", tsRange: "2022-12-22 18:48 - 2023-01-05 18:48"},
|
|
{name: "dnscat", rolling: "false", tsRange: "2018-01-30 18:00 - 2018-01-31 18:14"},
|
|
}
|
|
for i, line := range lines {
|
|
cols := strings.Split(line, "│")
|
|
require.Len(cols, 5)
|
|
cols = cols[1:4]
|
|
require.Equal(expectedDBs[i].name, strings.TrimSpace(cols[0]))
|
|
require.Equal(expectedDBs[i].rolling, strings.TrimSpace(cols[1]))
|
|
require.Equal(expectedDBs[i].tsRange, strings.TrimSpace(cols[2]))
|
|
}
|
|
|
|
// clean up
|
|
for _, db := range expectedDBs {
|
|
err := c.server.DeleteSensorDB(db.name)
|
|
require.NoError(err)
|
|
}
|
|
}
|