rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Style

Browse Source 
Signed-off-by: Bryan Frimin <bryan@probo.com>
This commit is contained in:
Bryan Frimin
2026-06-08 09:16:47 +02:00
parent 49d9a96355
commit f5a73c7be1
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
e2e/console/agent_run_test.go
View File

@@ -64,7 +64,7 @@ func seedAgentRun(t *testing.T, organizationID gid.GID, seed agentRunSeed) gid.G
id, tenant_id, organization_id, start_agent_name, status, id, tenant_id, organization_id, start_agent_name, status,
input_messages, error_message, started_at, created_at, updated_at input_messages, error_message, started_at, created_at, updated_at
) VALUES ( ) VALUES (
$1, $2, $3, $4, $5, $6, $7, $8, $9, $9 $1, $2, $3, $4, $5, $6::jsonb, $7, $8, $9, $9
) )
`, `,
id, id,
@@ -72,7 +72,7 @@ func seedAgentRun(t *testing.T, organizationID gid.GID, seed agentRunSeed) gid.G
organizationID, organizationID,
seed.agentName, seed.agentName,
seed.status, seed.status,
[]byte(`[]`), "[]",
seed.errorMessage, seed.errorMessage,
seed.startedAt, seed.startedAt,
seed.createdAt, seed.createdAt,
@@ -135,7 +135,7 @@ type agentRunNode struct {
} }
type agentRunConnectionResult struct { type agentRunConnectionResult struct {
Node struct { Node *struct {
AgentRuns struct { AgentRuns struct {
TotalCount int `json:"totalCount"` TotalCount int `json:"totalCount"`
Edges []struct { Edges []struct {
@@ -172,6 +172,7 @@ func TestAgentRun_List(t *testing.T) {
"orgId": owner.GetOrganizationID().String(), "orgId": owner.GetOrganizationID().String(),
}, &result) }, &result)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, result.Node, "organization node should resolve")
assert.Equal(t, 2, result.Node.AgentRuns.TotalCount) assert.Equal(t, 2, result.Node.AgentRuns.TotalCount)
require.Len(t, result.Node.AgentRuns.Edges, 2) require.Len(t, result.Node.AgentRuns.Edges, 2)
@@ -210,6 +211,7 @@ func TestAgentRun_ListEmpty(t *testing.T) {
"orgId": owner.GetOrganizationID().String(), "orgId": owner.GetOrganizationID().String(),
}, &result) }, &result)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, result.Node, "organization node should resolve")
assert.Equal(t, 0, result.Node.AgentRuns.TotalCount) assert.Equal(t, 0, result.Node.AgentRuns.TotalCount)
assert.Empty(t, result.Node.AgentRuns.Edges) assert.Empty(t, result.Node.AgentRuns.Edges)
@@ -236,6 +238,7 @@ func TestAgentRun_Ordering(t *testing.T) {
"orderBy": map[string]any{"direction": "ASC", "field": "CREATED_AT"}, "orderBy": map[string]any{"direction": "ASC", "field": "CREATED_AT"},
}, &result) }, &result)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, result.Node, "organization node should resolve")
require.Len(t, result.Node.AgentRuns.Edges, 3) require.Len(t, result.Node.AgentRuns.Edges, 3)
assert.Equal(t, oldestID.String(), result.Node.AgentRuns.Edges[0].Node.ID) assert.Equal(t, oldestID.String(), result.Node.AgentRuns.Edges[0].Node.ID)
@@ -253,6 +256,7 @@ func TestAgentRun_Ordering(t *testing.T) {
"orderBy": map[string]any{"direction": "DESC", "field": "CREATED_AT"}, "orderBy": map[string]any{"direction": "DESC", "field": "CREATED_AT"},
}, &result) }, &result)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, result.Node, "organization node should resolve")
require.Len(t, result.Node.AgentRuns.Edges, 3) require.Len(t, result.Node.AgentRuns.Edges, 3)
assert.Equal(t, newestID.String(), result.Node.AgentRuns.Edges[0].Node.ID) assert.Equal(t, newestID.String(), result.Node.AgentRuns.Edges[0].Node.ID)
@@ -301,6 +305,7 @@ func TestAgentRun_Pagination(t *testing.T) {
"first": 2, "first": 2,
}, &firstPage) }, &firstPage)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, firstPage.Node, "organization node should resolve")
assert.Equal(t, 3, firstPage.Node.AgentRuns.TotalCount) assert.Equal(t, 3, firstPage.Node.AgentRuns.TotalCount)
testutil.AssertFirstPage(t, len(firstPage.Node.AgentRuns.Edges), firstPage.Node.AgentRuns.PageInfo, 2, true) testutil.AssertFirstPage(t, len(firstPage.Node.AgentRuns.Edges), firstPage.Node.AgentRuns.PageInfo, 2, true)
@@ -314,6 +319,7 @@ func TestAgentRun_Pagination(t *testing.T) {
"after": *firstPage.Node.AgentRuns.PageInfo.EndCursor, "after": *firstPage.Node.AgentRuns.PageInfo.EndCursor,
}, &secondPage) }, &secondPage)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, secondPage.Node, "organization node should resolve")
testutil.AssertLastPage(t, len(secondPage.Node.AgentRuns.Edges), secondPage.Node.AgentRuns.PageInfo, 1, true) testutil.AssertLastPage(t, len(secondPage.Node.AgentRuns.Edges), secondPage.Node.AgentRuns.PageInfo, 1, true)
@@ -415,6 +421,7 @@ func TestAgentRun_RBAC(t *testing.T) {
"orgId": member.GetOrganizationID().String(), "orgId": member.GetOrganizationID().String(),
}, &listResult) }, &listResult)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, listResult.Node, "organization node should resolve")
assert.Equal(t, 1, listResult.Node.AgentRuns.TotalCount) assert.Equal(t, 1, listResult.Node.AgentRuns.TotalCount)
var getResult struct { var getResult struct {
@@ -473,6 +480,7 @@ func TestAgentRun_TenantIsolation(t *testing.T) {
"orgId": org2Owner.GetOrganizationID().String(), "orgId": org2Owner.GetOrganizationID().String(),
}, &result) }, &result)
require.NoError(t, err) require.NoError(t, err)
require.NotNil(t, result.Node, "organization node should resolve")
assert.Equal(t, 0, result.Node.AgentRuns.TotalCount) assert.Equal(t, 0, result.Node.AgentRuns.TotalCount)
assert.Empty(t, result.Node.AgentRuns.Edges) assert.Empty(t, result.Node.AgentRuns.Edges)