jayjay
/
k9s
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

receive sets.String as parameter and improve tests.

mine
Gustavo Paiva 2019-10-14 01:27:44 -03:00
parent a7bfe0e638
commit 616b068c0d
2 changed files with 33 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
internal/resource/no.go
View File

@ -3,10 +3,11 @@ package resource
import (
"strings"
"k8s.io/apimachinery/pkg/util/sets"
"github.com/derailed/k9s/internal/k8s"
"github.com/rs/zerolog/log"
v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/util/sets"
mv1beta1 "k8s.io/metrics/pkg/apis/metrics/v1beta1"
)
@ -143,12 +144,13 @@ func (r *Node) Fields(ns string) Row {
sta := make([]string, 10)
r.status(no.Status, no.Spec.Unschedulable, sta)
ro := r.findNodeRoles(no)
ro := sets.NewString()
r.findNodeRoles(no, &ro)
return append(ff,
no.Name,
join(sta, ","),
join(ro, ","),
join(ro.List(), ","),
no.Status.NodeInfo.KubeletVersion,
no.Status.NodeInfo.KernelVersion,
iIP,
@ -202,8 +204,7 @@ func gatherNodeMX(no *v1.Node, mx *mv1beta1.NodeMetrics) (c metric, a metric, p
return
}
func (_ *Node) findNodeRoles(no *v1.Node) []string {
roles := sets.NewString()
func (_ *Node) findNodeRoles(no *v1.Node, roles *sets.String) []string {
for k, v := range no.Labels {
switch {
case strings.HasPrefix(k, labelNodeRolePrefix):
@ -215,6 +216,10 @@ func (_ *Node) findNodeRoles(no *v1.Node) []string {
}
}
if roles.Len() == 0 {
roles.Insert(MissingValue)
}
return roles.List()
}

27
internal/resource/no_int_test.go
View File

@ -3,6 +3,8 @@ package resource
import (
"testing"
"k8s.io/apimachinery/pkg/util/sets"
"github.com/stretchr/testify/assert"
v1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@ -43,7 +45,7 @@ func TestNodeRoles(t *testing.T) {
node: v1.Node{
ObjectMeta: metav1.ObjectMeta{
Labels: map[string]string{
"node-role.kubernetes.io/master": "true",
"kubernetes.io/role": "master",
"node-role.kubernetes.io/worker": "true",
},
},
@ -56,18 +58,35 @@ func TestNodeRoles(t *testing.T) {
ObjectMeta: metav1.ObjectMeta{
Labels: map[string]string{
"node-role.kubernetes.io/worker": "true",
"node-role.kubernetes.io/master": "true",
"kubernetes.io/role": "master",
},
},
},
roles: []string{"master", "worker"},
},
{
node: v1.Node{
ObjectMeta: metav1.ObjectMeta{
Labels: map[string]string{
"kubernetes.io/role": "worker",
},
},
},
roles: []string{"worker"},
},
{
node: v1.Node{},
roles: []string{"<none>"},
},
}
no := NewNode(nil)
for _, u := range uu {
roles := no.findNodeRoles(&u.node)
assert.Equal(t, u.roles, roles)
roles := sets.NewString()
no.findNodeRoles(&u.node, &roles)
assert.Equal(t, u.roles, roles.List())
}
}