mirror of
https://github.com/juanfont/headscale.git
synced 2025-10-28 10:51:44 +01:00
scenario: create users in alphabetical order
Signed-off-by: Kristoffer Dalby <kristoffer@tailscale.com>
This commit is contained in:
parent
0be4c859e9
commit
a61ccc32cc
@ -22,6 +22,7 @@ import (
|
|||||||
"github.com/samber/lo"
|
"github.com/samber/lo"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
|
xmaps "golang.org/x/exp/maps"
|
||||||
"golang.org/x/sync/errgroup"
|
"golang.org/x/sync/errgroup"
|
||||||
"tailscale.com/envknob"
|
"tailscale.com/envknob"
|
||||||
)
|
)
|
||||||
@ -512,23 +513,26 @@ func (s *Scenario) CreateHeadscaleEnv(
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
for userName, clientCount := range users {
|
usernames := xmaps.Keys(users)
|
||||||
err = s.CreateUser(userName)
|
sort.Strings(usernames)
|
||||||
|
for _, username := range usernames {
|
||||||
|
clientCount := users[username]
|
||||||
|
err = s.CreateUser(username)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = s.CreateTailscaleNodesInUser(userName, "all", clientCount, tsOpts...)
|
err = s.CreateTailscaleNodesInUser(username, "all", clientCount, tsOpts...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
key, err := s.CreatePreAuthKey(userName, true, false)
|
key, err := s.CreatePreAuthKey(username, true, false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = s.RunTailscaleUp(userName, headscale.GetEndpoint(), key.GetKey())
|
err = s.RunTailscaleUp(username, headscale.GetEndpoint(), key.GetKey())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user