common/schema: make enabled/disabled columns configurable

common/schema/root_test.go

@@ -0,0 +1,33 @@
+// 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")
+	}
+
+}