(t *testing.T)
| 362 | } |
| 363 | |
| 364 | func TestModuleObjectConvertToSDKInputRewritesStoredResults(t *testing.T) { |
| 365 | t.Parallel() |
| 366 | |
| 367 | ctx := t.Context() |
| 368 | cacheIface, err := dagql.NewCache(ctx, "", nil, nil) |
| 369 | assert.NilError(t, err) |
| 370 | sc := cacheIface |
| 371 | root := &Query{} |
| 372 | testSrv := &moduleObjectTestServer{ |
| 373 | mockServer: &mockServer{}, |
| 374 | cache: sc, |
| 375 | root: root, |
| 376 | } |
| 377 | root.Server = testSrv |
| 378 | dag := newCoreDagqlServerForTest(t, root) |
| 379 | testSrv.dag = dag |
| 380 | ctx = dagql.ContextWithCache(ctx, sc) |
| 381 | ctx = ContextWithQuery(ctx, root) |
| 382 | ctx = engine.ContextWithClientMetadata(ctx, &engine.ClientMetadata{ |
| 383 | ClientID: "module-object-rewrite-client", |
| 384 | SessionID: "module-object-rewrite-session", |
| 385 | }) |
| 386 | installModuleObjectTestModuleClass(dag) |
| 387 | installTypeDefTestClasses(dag) |
| 388 | |
| 389 | childObjDef := NewObjectTypeDef("Child", "", nil) |
| 390 | childObjDefRes := newTypeDefDetachedResult(t, dag, "moduleObjectRewriteChildObj", childObjDef) |
| 391 | parentObjDef := NewObjectTypeDef("Parent", "", nil) |
| 392 | childTypeDef := newTypeDefDetachedResult(t, dag, "moduleObjectRewriteChildTypeDef", (&TypeDef{}).WithObjectTypeDef(childObjDefRes)) |
| 393 | parentObjDef.Fields = append(parentObjDef.Fields, newTypeDefDetachedResult(t, dag, "moduleObjectRewriteChildField", NewFieldTypeDef("child", childTypeDef, "", nil))) |
| 394 | parentObjDefRes := newTypeDefDetachedResult(t, dag, "moduleObjectRewriteParentObj", parentObjDef) |
| 395 | mod := &Module{ |
| 396 | NameField: "test", |
| 397 | Deps: NewSchemaBuilder(nil, nil), |
| 398 | ObjectDefs: dagql.ObjectResultArray[*TypeDef]{ |
| 399 | newTypeDefDetachedResult(t, dag, "moduleObjectRewriteChildTopTypeDef", (&TypeDef{}).WithObjectTypeDef(childObjDefRes)), |
| 400 | newTypeDefDetachedResult(t, dag, "moduleObjectRewriteParentTopTypeDef", (&TypeDef{}).WithObjectTypeDef(parentObjDefRes)), |
| 401 | }, |
| 402 | } |
| 403 | modRes, err := dagql.NewObjectResultForCall(mod, dag, moduleObjectTestSyntheticCall("moduleObjectRewriteStoredResultsModule", mod)) |
| 404 | assert.NilError(t, err) |
| 405 | parentType := &ModuleObjectType{ |
| 406 | typeDef: parentObjDef, |
| 407 | mod: modRes, |
| 408 | } |
| 409 | parentCall := moduleObjectTestSyntheticCall("moduleObjectParent", &ModuleObject{TypeDef: parentObjDef}) |
| 410 | ctx = dagql.ContextWithCall(ctx, parentCall) |
| 411 | |
| 412 | childDetached, err := dagql.NewResultForCall(&ModuleObject{ |
| 413 | Module: modRes, |
| 414 | TypeDef: childObjDef, |
| 415 | Fields: map[string]any{ |
| 416 | "name": "child", |
| 417 | }, |
| 418 | }, moduleObjectTestSyntheticCall("moduleObjectChild", &ModuleObject{TypeDef: childObjDef})) |
| 419 | assert.NilError(t, err) |
| 420 | child, err := sc.AttachResult(ctx, "test-session", dag, childDetached) |
| 421 | assert.NilError(t, err) |
nothing calls this directly
no test coverage detected