feat: enhance session deletion message and add tests for session creation and listing

Author: therealjohn

cli/azd/extensions/azure.ai.agents/internal/cmd/session.go

@@ -424,7 +424,10 @@ The isolation key is derived from the Entra token by default.`,
 				)
 			}
 
-			fmt.Printf("Session %q deleted.\n", sessionID)
+			fmt.Printf(
+				"Session %q deleted from agent %q.\n",
+				sessionID, sc.agentName,
+			)
 			return nil
 		},
 	}

cli/azd/extensions/azure.ai.agents/internal/pkg/agents/agent_api/operations_test.go

@@ -4,7 +4,9 @@
 package agent_api
 
 import (
+	"io"
 	"net/http"
+	"strings"
 	"testing"
 
 	"github.com/Azure/azure-sdk-for-go/sdk/azcore/policy"
@@ -86,3 +88,87 @@ func TestGetSession_404ReturnsError(t *testing.T) {
 	)
 	require.Error(t, err, "404 should be an error from GetSession")
 }
+
+// fakeBodyTransport returns a canned status code and JSON body.
+type fakeBodyTransport struct {
+	statusCode int
+	body       string
+}
+
+func (f *fakeBodyTransport) Do(req *http.Request) (*http.Response, error) {
+	return &http.Response{
+		StatusCode: f.statusCode,
+		Header:     http.Header{"Content-Type": {"application/json"}},
+		Body:       io.NopCloser(strings.NewReader(f.body)),
+		Request:    req,
+	}, nil
+}
+
+func TestCreateSession_Returns201WithBody(t *testing.T) {
+	body := `{
+		"agent_session_id": "sess-new",
+		"version_indicator": {"type": "version_ref", "agent_version": "3"},
+		"status": "running",
+		"created_at": 1700000000,
+		"last_accessed_at": 1700000100,
+		"expires_at": 1700086400
+	}`
+
+	client := newTestClient(
+		"https://test.example.com/api/projects/proj",
+		&fakeBodyTransport{
+			statusCode: http.StatusCreated,
+			body:       body,
+		},
+	)
+
+	session, err := client.CreateSession(
+		t.Context(), "my-agent", "",
+		&CreateAgentSessionRequest{
+			VersionIndicator: &VersionIndicator{
+				Type:         "version_ref",
+				AgentVersion: "3",
+			},
+		},
+		"2025-11-15-preview",
+	)
+
+	require.NoError(t, err)
+	require.Equal(t, "sess-new", session.AgentSessionID)
+	require.Equal(t, "3", session.VersionIndicator.AgentVersion)
+	require.Equal(t, AgentSessionStatus("running"), session.Status)
+}
+
+func TestListSessions_Returns200WithPagination(t *testing.T) {
+	body := `{
+		"data": [
+			{
+				"agent_session_id": "sess-1",
+				"version_indicator": {"type": "version_ref", "agent_version": "2"},
+				"status": "running",
+				"created_at": 1700000000,
+				"last_accessed_at": 1700000100,
+				"expires_at": 1700086400
+			}
+		],
+		"pagination_token": "next-page-abc"
+	}`
+
+	client := newTestClient(
+		"https://test.example.com/api/projects/proj",
+		&fakeBodyTransport{
+			statusCode: http.StatusOK,
+			body:       body,
+		},
+	)
+
+	result, err := client.ListSessions(
+		t.Context(), "my-agent", nil, nil, "2025-11-15-preview",
+	)
+
+	require.NoError(t, err)
+	require.Len(t, result.Data, 1)
+	require.Equal(t, "sess-1", result.Data[0].AgentSessionID)
+	require.NotNil(t, result.PaginationToken)
+	require.Equal(t, "next-page-abc", *result.PaginationToken)
+}