fleet/orbit/pkg/execuser/execuser_linux_test.go
Lucas Manuel Rodriguez c81117b9b2
Add Fleet Desktop support for Wayland display sessions (#25998)
For #19043.

See the versions and distributions tested during development on the QA
notes of #19043.

---

- [X] Changes file added for user-visible changes in `changes/`,
`orbit/changes/` or `ee/fleetd-chrome/changes`.
See [Changes
files](https://github.com/fleetdm/fleet/blob/main/docs/Contributing/Committing-Changes.md#changes-files)
for more information.
- [x] A detailed QA plan exists on the associated ticket (if it isn't
there, work with the product group's QA engineer to add it)
- [X] Manual QA for all new/changed functionality
- For Orbit and Fleet Desktop changes:
- [X] Orbit runs on macOS, Linux and Windows. Check if the orbit
feature/bugfix should only apply to one platform (`runtime.GOOS`).
- [X] Manual QA must be performed in the three main OSs, macOS, Windows
and Linux.
- [X] Auto-update manual QA, from released version of component to new
version (see [tools/tuf/test](../tools/tuf/test/README.md)).
2025-02-05 14:00:13 -03:00

67 lines
1.2 KiB
Go

package execuser
import (
"bytes"
"testing"
"github.com/stretchr/testify/require"
)
func TestParseWhoOutputForDisplay(t *testing.T) {
testCases := []struct {
name string
output string
user string
expectedDisplay string
expectedErr bool
}{
{
"Ubuntu 22.04.2 (X11)",
`foo :0 2024-05-14 17:34 (:0)`,
"foo",
":0",
false,
},
{
"Ubuntu 22.04.2 (X11) - user not listed",
`foo :0 2024-05-14 17:34 (:0)`,
"bar",
"",
true,
},
{
"Ubuntu 24.04 (X11)",
`foo seat0 2024-05-14 17:42 (login screen)
foo :1 2024-05-14 17:42 (:1)`,
"foo",
":1",
false,
},
{
"Ubuntu 24.04 (Wayland) - DISPLAY not found",
`foo seat0 2024-05-14 18:11 (login screen)
foo tty2 2024-05-14 18:11 (tty2)`,
"foo",
"",
true,
},
{
"Empty",
``,
"foo",
"",
true,
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
display, err := parseWhoOutputForDisplay(bytes.NewReader([]byte(tc.output)), tc.user)
require.Equal(t, tc.expectedDisplay, display)
if tc.expectedErr {
require.Error(t, err)
} else {
require.NoError(t, err)
}
})
}
}