add tests

crates/dagger-codegen/src/handlers/object.rs

@@ -19,8 +19,6 @@ impl Handler for Object {
             .ok_or(eyre::anyhow!("could not find name"))?;
         let description = render_description(t);
 
-        let input = rust::import("dagger_core", "Input");
-
         let fields = match t.fields.as_ref() {
             Some(i) => fields::render_fields(i)?,
             None => None,
@@ -40,8 +38,6 @@ impl Handler for Object {
             impl $name {
                 $(if fields.is_some() => $fields)
             }
-
-            impl $input for $name {}
         };
 
         Ok(out)
@@ -51,7 +47,8 @@ impl Handler for Object {
 #[cfg(test)]
 mod tests {
     use dagger_core::introspection::{
-        FullType, FullTypeFields, FullTypeFieldsType, TypeRef, __TypeKind,
+        FullType, FullTypeFields, FullTypeFieldsArgs, FullTypeFieldsType, InputValue, TypeRef,
+        __TypeKind,
     };
     use pretty_assertions::assert_eq;
 
@@ -88,21 +85,75 @@ mod tests {
             enum_values: None,
             possible_types: None,
         };
-        let expected = r#"use dagger_core::Input;
-
-
+        let expected = r#"
 /// A directory whose contents persists across sessions
 pub struct CacheVolume {}
 
 impl CacheVolume {
     pub fn id(
         &self,
-    ) -> Option<CacheID> {
+    ) -> CacheID {
+        let query = self.selection.select("id");
+
+        CacheID {
+            conn: self.conn.clone(),
+            proc: self.proc.clone(),
+            selection: query,
+        }
+
         todo!()
     }
 }
+"#;
+        let handler = Object {};
+        let obj = handler.render(&t).unwrap();
+        let actual = obj.to_file_string().unwrap();
 
-impl Input for CacheVolume {}
+        assert_eq!(actual, expected);
+    }
+
+    #[test]
+    fn can_render_query_container() {
+        let t: FullType = FullType {
+            kind: Some(__TypeKind::OBJECT),
+            name: Some("Query".into()),
+            description: None,
+            fields: Some(vec![FullTypeFields {
+                name: Some("container".into()),
+                description: Some("Loads a container from ID.\nNull ID returns an empty container (scratch).\nOptional platform argument initializes new containers to execute and publish as that platform. Platform defaults to that of the builder's host.".into()),
+                args: Some(vec![
+                    Some(
+                        FullTypeFieldsArgs
+                        {
+                        input_value: InputValue { name: "id".into(), description: None, type_: TypeRef { kind: Some(__TypeKind::SCALAR), name: Some("ContainerID".into()), of_type: None }, default_value: None } 
+                    }),
+                    Some(
+                        FullTypeFieldsArgs {
+                        input_value: InputValue {
+                            name: "platform".into(), description: None, type_: TypeRef { kind: Some(__TypeKind::SCALAR), name: Some("Platform".into()), of_type: None }, 
+                            default_value: None }
+                    })
+                ]),
+                type_: Some(FullTypeFieldsType {
+                    type_ref: TypeRef {
+                        kind: Some(__TypeKind::NON_NULL),
+                        name: None,
+                        of_type: Some(Box::new(TypeRef {
+                            kind: Some(__TypeKind::SCALAR),
+                            name: Some("CacheID".into()),
+                            of_type: None,
+                        })),
+                    },
+                }),
+                is_deprecated: Some(false),
+                deprecation_reason: None,
+            }]),
+            input_fields: None,
+            interfaces: None,
+            enum_values: None,
+            possible_types: None,
+        };
+        let expected = r#"
 "#;
         let handler = Object {};
         let obj = handler.render(&t).unwrap();