Skip to content

Commit

Permalink
allow to define the nftables table name
Browse files Browse the repository at this point in the history
  • Loading branch information
aojea committed Oct 23, 2024
1 parent dff6a37 commit 26cd9e1
Showing 1 changed file with 69 additions and 0 deletions.
69 changes: 69 additions & 0 deletions pkg/networkpolicy/controller_test.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
package networkpolicy

import (
"testing"

v1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/informers"
Expand Down Expand Up @@ -125,3 +127,70 @@ func newTestController() *networkpolicyController {
informersFactory.Core().V1().Nodes().Informer().GetStore(),
}
}

func TestConfig_Defaults(t *testing.T) {
tests := []struct {
name string
config Config
expected Config
}{
{
name: "empty",
config: Config{
NodeName: "testnode", // nodename defaults to os.Hostname so we ignore for tests
},
expected: Config{
FailOpen: false,
AdminNetworkPolicy: false,
BaselineAdminNetworkPolicy: false,
QueueID: 100,
NodeName: "testnode", // nodename defaults to os.Hostname so we ignore for tests
NetfilterBug1766Fix: false,
NFTableName: "kube-network-policies",
},
}, {
name: "queue id",
config: Config{
NodeName: "testnode", // nodename defaults to os.Hostname so we ignore for tests
QueueID: 99,
},
expected: Config{
FailOpen: false,
AdminNetworkPolicy: false,
BaselineAdminNetworkPolicy: false,
QueueID: 99,
NodeName: "testnode", // nodename defaults to os.Hostname so we ignore for tests
NetfilterBug1766Fix: false,
NFTableName: "kube-network-policies",
},
}, {
name: "table name",
config: Config{
NodeName: "testnode", // nodename defaults to os.Hostname so we ignore for tests
QueueID: 99,
NFTableName: "kindnet-network-policies",
},
expected: Config{
FailOpen: false,
AdminNetworkPolicy: false,
BaselineAdminNetworkPolicy: false,
QueueID: 99,
NodeName: "testnode", // nodename defaults to os.Hostname so we ignore for tests
NetfilterBug1766Fix: false,
NFTableName: "kindnet-network-policies",
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
c := tt.config
if err := c.Defaults(); err != nil {
t.Errorf("Config.Defaults() error = %v", err)
}

if c != tt.expected {
t.Errorf("Config.Defaults() = %v, want %v", c, tt.expected)
}
})
}
}

0 comments on commit 26cd9e1

Please sign in to comment.