mirror of
https://github.com/akvorado/akvorado.git
synced 2025-12-11 22:14:02 +01:00
34 lines
852 B
Go
34 lines
852 B
Go
// SPDX-FileCopyrightText: 2023 Free Mobile
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
package schema_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"akvorado/common/schema"
|
|
)
|
|
|
|
func TestEnableDisableColumns(t *testing.T) {
|
|
config := schema.DefaultConfiguration()
|
|
config.Enabled = []schema.ColumnKey{schema.ColumnDstVlan, schema.ColumnSrcVlan}
|
|
config.Disabled = []schema.ColumnKey{schema.ColumnSrcCountry, schema.ColumnDstCountry}
|
|
c, err := schema.New(config)
|
|
if err != nil {
|
|
t.Fatalf("New() error:\n%+v", err)
|
|
}
|
|
|
|
if column, ok := c.LookupColumnByKey(schema.ColumnDstVlan); !ok {
|
|
t.Fatal("DstVlan not found")
|
|
} else if column.Disabled {
|
|
t.Fatal("DstVlan is still disabled")
|
|
}
|
|
|
|
if column, ok := c.LookupColumnByKey(schema.ColumnDstCountry); !ok {
|
|
t.Fatal("DstCountry not found")
|
|
} else if !column.Disabled {
|
|
t.Fatal("DstCountry is not disabled")
|
|
}
|
|
|
|
}
|