mirror of
https://github.com/juanfont/headscale.git
synced 2025-09-20 17:53:11 +02:00
fix ssh unit test
Signed-off-by: Kristoffer Dalby <kristoffer@tailscale.com>
This commit is contained in:
parent
64560bbf6e
commit
d8adb728c0
@ -3539,6 +3539,11 @@ func Test_getFilteredByACLPeers(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestSSHRules(t *testing.T) {
|
func TestSSHRules(t *testing.T) {
|
||||||
|
users := []types.User{
|
||||||
|
{
|
||||||
|
Name: "user1",
|
||||||
|
},
|
||||||
|
}
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
node types.Node
|
node types.Node
|
||||||
@ -3552,18 +3557,14 @@ func TestSSHRules(t *testing.T) {
|
|||||||
Hostname: "testnodes",
|
Hostname: "testnodes",
|
||||||
IPv4: iap("100.64.99.42"),
|
IPv4: iap("100.64.99.42"),
|
||||||
UserID: 0,
|
UserID: 0,
|
||||||
User: types.User{
|
User: users[0],
|
||||||
Name: "user1",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
peers: types.Nodes{
|
peers: types.Nodes{
|
||||||
&types.Node{
|
&types.Node{
|
||||||
Hostname: "testnodes2",
|
Hostname: "testnodes2",
|
||||||
IPv4: iap("100.64.0.1"),
|
IPv4: iap("100.64.0.1"),
|
||||||
UserID: 0,
|
UserID: 0,
|
||||||
User: types.User{
|
User: users[0],
|
||||||
Name: "user1",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
pol: ACLPolicy{
|
pol: ACLPolicy{
|
||||||
@ -3676,18 +3677,14 @@ func TestSSHRules(t *testing.T) {
|
|||||||
Hostname: "testnodes",
|
Hostname: "testnodes",
|
||||||
IPv4: iap("100.64.0.1"),
|
IPv4: iap("100.64.0.1"),
|
||||||
UserID: 0,
|
UserID: 0,
|
||||||
User: types.User{
|
User: users[0],
|
||||||
Name: "user1",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
peers: types.Nodes{
|
peers: types.Nodes{
|
||||||
&types.Node{
|
&types.Node{
|
||||||
Hostname: "testnodes2",
|
Hostname: "testnodes2",
|
||||||
IPv4: iap("100.64.99.42"),
|
IPv4: iap("100.64.99.42"),
|
||||||
UserID: 0,
|
UserID: 0,
|
||||||
User: types.User{
|
User: users[0],
|
||||||
Name: "user1",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
pol: ACLPolicy{
|
pol: ACLPolicy{
|
||||||
@ -3725,7 +3722,7 @@ func TestSSHRules(t *testing.T) {
|
|||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
got, err := tt.pol.CompileSSHPolicy(&tt.node, []types.User{}, tt.peers)
|
got, err := tt.pol.CompileSSHPolicy(&tt.node, users, tt.peers)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
if diff := cmp.Diff(tt.want, got); diff != "" {
|
if diff := cmp.Diff(tt.want, got); diff != "" {
|
||||||
|
Loading…
Reference in New Issue
Block a user