Function TestIsIntrospectionPreservesClassification

core/telemetry_test.go:260–305

(t *testing.T)

}

func TestIsIntrospectionPreservesClassification(t *testing.T) {
	cache, err := dagql.NewCache(t.Context(), "", nil, nil)
	require.NoError(t, err)
	ctx := dagql.ContextWithCache(t.Context(), cache)

	functionFrame := testResultCall("function", dagql.String(""), nil)
	functionFrame.Type = dagql.NewResultCallType((&Function{}).Type())

	tests := []struct {
		name  string
		frame *dagql.ResultCall
		want  bool
	}{
		{
			name:  "root currentTypeDefs",
			frame: testResultCall("currentTypeDefs", dagql.String(""), nil),
			want:  true,
		},
		{
			name:  "root plain field",
			frame: testResultCall("plain", dagql.String(""), nil),
			want:  false,
		},
		{
			name:  "function builder field",
			frame: testResultCall("withArg", dagql.String(""), functionFrame),
			want:  true,
		},
		{
			name:  "descendant of introspection root",
			frame: testResultCall("name", dagql.String(""), testResultCall("currentTypeDefs", dagql.String(""), nil)),
			want:  true,
		},
		{
			name:  "descendant of plain root",
			frame: testResultCall("name", dagql.String(""), testResultCall("plain", dagql.String(""), nil)),
			want:  false,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			require.Equal(t, tc.want, isIntrospection(ctx, tc.frame))
		})
	}
}

type telemetryTestNoopResolver struct{}