Add more tests

gravitational · Feb 21, 2025 · d70bd7f · d70bd7f
1 parent d560f1c
commit d70bd7f
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 1 deletion.
diff --git a/lib/services/local/desktops.go b/lib/services/local/desktops.go
@@ -39,7 +39,7 @@ func NewWindowsDesktopService(backend backend.Backend) *WindowsDesktopService {
 }
 
 func (s *WindowsDesktopService) getWindowsDesktop(ctx context.Context, name, hostID string) (types.WindowsDesktop, error) {
-	key := backend.ExactKey(windowsDesktopsPrefix, hostID, name)
+	key := backend.NewKey(windowsDesktopsPrefix, hostID, name)
 	item, err := s.Get(ctx, key)
 	if err != nil {
 		return nil, trace.Wrap(err)

diff --git a/lib/services/local/desktops_test.go b/lib/services/local/desktops_test.go
@@ -30,8 +30,60 @@ import (
 	"github.com/gravitational/teleport/api/types"
 	"github.com/gravitational/teleport/lib/backend"
 	"github.com/gravitational/teleport/lib/backend/memory"
+	"github.com/gravitational/trace"
 )
 
+func TestGetWindowsDesktops(t *testing.T) {
+	t.Parallel()
+
+	ctx := context.Background()
+	clock := clockwork.NewFakeClock()
+
+	bk, err := memory.New(memory.Config{
+		Context: ctx,
+		Clock:   clock,
+	})
+	require.NoError(t, err)
+
+	service := NewWindowsDesktopService(bk)
+
+	d1, err := types.NewWindowsDesktopV3("apple", nil, types.WindowsDesktopSpecV3{Addr: "_", HostID: "test-host-id-1"})
+	require.NoError(t, err)
+	require.NoError(t, service.CreateWindowsDesktop(ctx, d1))
+
+	d2, err := types.NewWindowsDesktopV3("apple", nil, types.WindowsDesktopSpecV3{Addr: "_", HostID: "test-host-id-2"})
+	require.NoError(t, err)
+	require.NoError(t, service.CreateWindowsDesktop(ctx, d2))
+
+	d3, err := types.NewWindowsDesktopV3("carrot", nil, types.WindowsDesktopSpecV3{Addr: "_", HostID: "test-host-id-2"})
+	require.NoError(t, err)
+	require.NoError(t, service.CreateWindowsDesktop(ctx, d3))
+
+	// search by name and host ID
+	result, err := service.GetWindowsDesktops(ctx, types.WindowsDesktopFilter{Name: "apple", HostID: "test-host-id-2"})
+	require.NoError(t, err)
+	require.Len(t, result, 1)
+	require.Equal(t, d1.GetName(), result[0].GetName())
+
+	// search by host ID
+	result, err = service.GetWindowsDesktops(ctx, types.WindowsDesktopFilter{HostID: "test-host-id-2"})
+	require.NoError(t, err)
+	require.Len(t, result, 2)
+	require.Equal(t, d2.GetName(), result[0].GetName())
+	require.Equal(t, d3.GetName(), result[1].GetName())
+
+	// no filter
+	result, err = service.GetWindowsDesktops(ctx, types.WindowsDesktopFilter{})
+	require.NoError(t, err)
+	require.Len(t, result, 3)
+
+	// non-matching filter
+	result, err = service.GetWindowsDesktops(ctx, types.WindowsDesktopFilter{Name: "foo", HostID: "bar"})
+	require.Error(t, err)
+	require.True(t, trace.IsNotFound(err))
+	require.Empty(t, result)
+}
+
 func TestListWindowsDesktops(t *testing.T) {
 	t.Parallel()