@alloy-js/python 0.4.0-dev.3 → 0.4.0-dev.6

package/dist/dev/test/memberexpressions.test.js

@@ -0,0 +1,1668 @@
+import { createComponent as _$createComponent, memo as _$memo, createIntrinsic as _$createIntrinsic } from "@alloy-js/core/jsx-runtime";
+import { code, memberRefkey, refkey } from "@alloy-js/core";
+import { d } from "@alloy-js/core/testing";
+import { describe, expect, it } from "vitest";
+import * as py from "../src/index.js";
+import { ClassDeclaration, FunctionDeclaration, SourceFile, VariableDeclaration } from "../src/index.js";
+import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+it("renders basic member expression with dot notation", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "obj"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 22,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "property"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 23,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 21,
+    columnNumber: 7
+  })])).toBe(d`
+    obj.property
+  `);
+});
+it("renders basic member expression with key", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 35,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        key: 12345
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 36,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 37,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 34,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[12345].foo-bar
+  `);
+});
+it("renders basic member expression with key with reference", () => {
+  const rk1 = refkey();
+  expect(toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return [_$createComponent(py.VariableDeclaration, {
+        name: "test1",
+        refkey: rk1,
+        initializer: 1
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 50,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 52,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: rk1
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 53,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 54,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 51,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 49,
+    columnNumber: 7
+  })])).toBe(d`
+    test1 = 1
+    arr[test1].foo-bar
+  `);
+});
+it("renders basic member expression with keys", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 68,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        keys: [1, 2]
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 69,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 70,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 67,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[1, 2].foo-bar
+  `);
+});
+it("renders basic member expression with keys with references", () => {
+  const rk1 = refkey();
+  const rk2 = refkey();
+  expect(toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return [_$createComponent(py.VariableDeclaration, {
+        name: "test1",
+        refkey: rk1,
+        initializer: 1
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 84,
+        columnNumber: 9
+      }), _$createComponent(py.VariableDeclaration, {
+        name: "test2",
+        refkey: rk2,
+        initializer: 2
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 85,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 87,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            keys: [rk1, rk2]
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 88,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 89,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 86,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 83,
+    columnNumber: 7
+  })])).toBe(d`
+    test1 = 1
+    test2 = 2
+    arr[test1, test2].foo-bar
+  `);
+});
+it("renders basic member expression with slice - 1", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 104,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        slice: {
+          start: 1,
+          stop: 3,
+          step: 2
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 105,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 106,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 103,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[1:3:2].foo-bar
+  `);
+});
+it("renders basic member expression with slice - 1 with references", () => {
+  const rk1 = refkey();
+  const rk2 = refkey();
+  const rk3 = refkey();
+  expect(toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return [_$createComponent(py.VariableDeclaration, {
+        name: "test1",
+        refkey: rk1,
+        initializer: 1
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 121,
+        columnNumber: 9
+      }), _$createComponent(py.VariableDeclaration, {
+        name: "test2",
+        refkey: rk2,
+        initializer: 3
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 122,
+        columnNumber: 9
+      }), _$createComponent(py.VariableDeclaration, {
+        name: "test3",
+        refkey: rk3,
+        initializer: 2
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 123,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 125,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {
+              start: rk1,
+              stop: rk2,
+              step: rk3
+            }
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 126,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 129,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 124,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 120,
+    columnNumber: 7
+  })])).toBe(d`
+    test1 = 1
+    test2 = 3
+    test3 = 2
+    arr[test1:test2:test3].foo-bar
+  `);
+});
+it("renders basic member expression with slice - 2", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 145,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        slice: {
+          stop: 3
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 146,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 147,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 144,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[:3].foo-bar
+  `);
+});
+it("renders basic member expression with slice - 3", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 159,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        slice: {
+          step: 2
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 160,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 161,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 158,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[::2].foo-bar
+  `);
+});
+it("renders basic member expression with slice - 4", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 173,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        slice: {
+          start: 1
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 174,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 175,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 172,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[1:].foo-bar
+  `);
+});
+it("renders basic member expression with slice - 5", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "arr"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 187,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        slice: {
+          start: 1,
+          step: 2
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 188,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "foo-bar"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 189,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 186,
+    columnNumber: 7
+  })])).toBe(d`
+    arr[1::2].foo-bar
+  `);
+});
+it("renders basic member expression with an expression index - 1", () => {
+  const xRefkey = refkey();
+  expect(toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return [_$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 1,
+        refkey: xRefkey
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 202,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 204,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: code`${xRefkey} + 1`
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 205,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 206,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 203,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 201,
+    columnNumber: 7
+  })])).toBe(d`
+    x = 1
+    arr[x + 1].foo-bar
+  `);
+});
+it("renders basic member expression with an expression index - 2", () => {
+  expect(toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 221,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: code`"foo" + 1`
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 222,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 223,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 220,
+        columnNumber: 9
+      });
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 219,
+    columnNumber: 7
+  })])).toBe(d`
+    arr["foo" + 1].foo-bar
+  `);
+});
+it("throws an error for invalid identifiers with quotes", () => {
+  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "obj"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 236,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: `property-"name"`
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 237,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 235,
+    columnNumber: 7
+  })])).toThrowError(/Invalid identifier: property-"name"/);
+});
+it("supports multiple levels of nesting", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "a"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 247,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "b"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 248,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "c"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 249,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "d"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 250,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 246,
+    columnNumber: 7
+  })])).toBe(d`
+    a.b.c.d
+  `);
+});
+it("handles a mix of dot and bracket notation", () => {
+  expect(toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "obj"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 262,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "normalProp"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 263,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        key: "special-prop"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 264,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        key: "123"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 265,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 261,
+    columnNumber: 7
+  })])).toBe(d`
+    obj.normalProp["special-prop"]["123"]
+  `);
+});
+it("throws an error when providing conflicting part props", () => {
+  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "obj"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 277,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        id: "property",
+        args: [1, 2]
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 278,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 276,
+    columnNumber: 7
+  })])).toThrowError(`Only one of args, id can be used for a MemberExpression part at a time`);
+});
+it("throws an error when providing an empty slice", () => {
+  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "obj"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 290,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        slice: {}
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 291,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 289,
+    columnNumber: 7
+  })])).toThrowError(`MemberExpression.Part: slice object cannot be empty`);
+});
+it("throws an error when providing an empty array", () => {
+  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+    get children() {
+      return [_$createComponent(py.MemberExpression.Part, {
+        id: "obj"
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 301,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression.Part, {
+        keys: []
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 302,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 300,
+    columnNumber: 7
+  })])).toThrowError(`MemberExpression.Part: keys array cannot be empty`);
+});
+it("takes children for the id part", () => {
+  expect(toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return [_$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "child1"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 313,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "child1"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 314,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 312,
+        columnNumber: 9
+      }), _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "child1"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 317,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "child2"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 318,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            args: true
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 319,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "child3"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 320,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            args: true
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 321,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: code`"foo" + 1`
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 322,
+            columnNumber: 11
+          }), _$createComponent(py.MemberExpression.Part, {
+            args: true
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 323,
+            columnNumber: 11
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 316,
+        columnNumber: 9
+      })];
+    }
+  }, {
+    fileName: import.meta.url,
+    lineNumber: 311,
+    columnNumber: 7
+  })])).toBe(d`
+    child1["child1"]
+    child1["child2"]()["child3"]()["foo" + 1]()
+  `);
+});
+describe("with refkeys", () => {
+  it("handles symbols correctly", () => {
+    const rk1 = refkey();
+    const rk2 = refkey();
+    expect(toSourceText([_$createComponent(py.StatementList, {
+      get children() {
+        return [_$createComponent(py.VariableDeclaration, {
+          name: "test1",
+          refkey: rk1,
+          initializer: 1
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 340,
+          columnNumber: 11
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "test1",
+          refkey: rk2,
+          initializer: 2
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 341,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              refkey: rk1
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 343,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              refkey: rk2
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 344,
+              columnNumber: 13
+            })];
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 342,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 339,
+      columnNumber: 9
+    })])).toBe(d`
+        test1 = 1
+        test1_2_test = 2
+        test1.test1_2_test
+      `);
+  });
+  it("Correctly resolves member expressions from 2 different classes", () => {
+    const model1Ref = refkey();
+    const model2Ref = refkey();
+    const classMethod1Ref = refkey();
+    const classMethod2Ref = refkey();
+    const v1Rk = refkey();
+    const v2Rk = refkey();
+    const template = _$createComponent(py.StatementList, {
+      get children() {
+        return [_$createComponent(ClassDeclaration, {
+          name: "Model1",
+          refkey: model1Ref,
+          get children() {
+            return _$createComponent(VariableDeclaration, {
+              name: "foo",
+              refkey: classMethod1Ref,
+              type: "str",
+              omitNone: true,
+              instanceVariable: true
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 365,
+              columnNumber: 11
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 364,
+          columnNumber: 9
+        }), _$createComponent(ClassDeclaration, {
+          name: "Model2",
+          refkey: model2Ref,
+          get children() {
+            return _$createComponent(VariableDeclaration, {
+              name: "bar",
+              refkey: classMethod2Ref,
+              type: "str",
+              omitNone: true,
+              instanceVariable: true
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 374,
+              columnNumber: 11
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 373,
+          columnNumber: 9
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "model1_instance",
+          refkey: v1Rk,
+          type: model1Ref,
+          get initializer() {
+            return _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: model1Ref
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 388,
+                  columnNumber: 15
+                }), _$createComponent(py.MemberExpression.Part, {
+                  args: true
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 389,
+                  columnNumber: 15
+                })];
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 387,
+              columnNumber: 13
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 382,
+          columnNumber: 9
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "model2_instance",
+          refkey: v2Rk,
+          type: model2Ref,
+          get initializer() {
+            return _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: model2Ref
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 399,
+                  columnNumber: 15
+                }), _$createComponent(py.MemberExpression.Part, {
+                  args: true
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 400,
+                  columnNumber: 15
+                })];
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 398,
+              columnNumber: 13
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 393,
+          columnNumber: 9
+        }), [_$memo(() => memberRefkey(v1Rk, classMethod1Ref), false, "memberRefkey(\u2026)")], [_$memo(() => memberRefkey(v2Rk, classMethod2Ref), false, "memberRefkey(\u2026)")]];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 363,
+      columnNumber: 7
+    });
+    expect(toSourceText([template])).toBe(d`
+      class Model1:
+          foo: str
+
+      class Model2:
+          bar: str
+
+      model1_instance: Model1 = Model1()
+      model2_instance: Model2 = Model2()
+      model1_instance.foo
+      model2_instance.bar
+    `);
+  });
+  it("handles optional parameters correctly", () => {
+    const fooRef = refkey();
+    const modelRef = refkey();
+    const parameters = [{
+      name: "foo",
+      default: null,
+      refkey: fooRef,
+      type: modelRef
+    }];
+    const messageRef = refkey();
+    const template = _$createComponent(py.StatementList, {
+      get children() {
+        return [_$createComponent(ClassDeclaration, {
+          name: "Model",
+          refkey: modelRef,
+          get children() {
+            return _$createComponent(VariableDeclaration, {
+              name: "bar",
+              get refkey() {
+                return refkey();
+              },
+              type: "str",
+              omitNone: true
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 438,
+              columnNumber: 11
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 437,
+          columnNumber: 9
+        }), _$createComponent(FunctionDeclaration, {
+          name: "fooFunction",
+          parameters: parameters,
+          get children() {
+            return _$createComponent(py.StatementList, {
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  name: "message",
+                  refkey: messageRef,
+                  get initializer() {
+                    return _$createComponent(py.MemberExpression, {
+                      get children() {
+                        return [_$createComponent(py.MemberExpression.Part, {
+                          refkey: fooRef
+                        }, {
+                          fileName: import.meta.url,
+                          lineNumber: 452,
+                          columnNumber: 19
+                        }), _$createComponent(py.MemberExpression.Part, {
+                          id: "bar"
+                        }, {
+                          fileName: import.meta.url,
+                          lineNumber: 453,
+                          columnNumber: 19
+                        })];
+                      }
+                    }, {
+                      fileName: import.meta.url,
+                      lineNumber: 451,
+                      columnNumber: 17
+                    });
+                  }
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 447,
+                  columnNumber: 13
+                }), ["print(", messageRef, ")"]];
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 446,
+              columnNumber: 11
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 445,
+          columnNumber: 9
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 436,
+      columnNumber: 7
+    });
+    expect(toSourceText([template])).toBe(d`
+      class Model:
+          bar: str
+
+      def foo_function(foo: Model = None):
+          message = foo.bar
+          print(message)
+
+    `);
+  });
+  it("handles class member correctly", () => {
+    const classRefkey = refkey();
+    const interfaceRefkey = refkey();
+    const interfaceMemberRefkey = refkey();
+    const classMethodRefkey = refkey();
+    const classMemberRefkey = refkey();
+    const instanceRefkey = refkey();
+    expect(toSourceText([_$createComponent(py.StatementList, {
+      get children() {
+        return [_$createComponent(ClassDeclaration, {
+          name: "Bar",
+          refkey: interfaceRefkey,
+          get children() {
+            return _$createComponent(VariableDeclaration, {
+              name: "prop1",
+              refkey: interfaceMemberRefkey,
+              type: "str",
+              omitNone: true
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 485,
+              columnNumber: 13
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 484,
+          columnNumber: 11
+        }), _$createComponent(ClassDeclaration, {
+          name: "Foo",
+          refkey: classRefkey,
+          get children() {
+            return [_$createComponent(VariableDeclaration, {
+              name: "test1",
+              refkey: classMemberRefkey,
+              type: interfaceRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 493,
+              columnNumber: 13
+            }), _$createIntrinsic("br", {}, {
+              fileName: import.meta.url,
+              lineNumber: 498,
+              columnNumber: 13
+            }), _$createComponent(FunctionDeclaration, {
+              name: "testMethod",
+              parameters: [],
+              refkey: classMethodRefkey,
+              returnType: interfaceRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 499,
+              columnNumber: 13
+            })];
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 492,
+          columnNumber: 11
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "inst",
+          refkey: instanceRefkey,
+          initializer: code`${classRefkey}()`
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 506,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              refkey: instanceRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 512,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              refkey: classMemberRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 513,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              refkey: interfaceMemberRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 514,
+              columnNumber: 13
+            })];
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 511,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              refkey: instanceRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 517,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              refkey: classMethodRefkey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 518,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: []
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 519,
+              columnNumber: 13
+            })];
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 516,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 483,
+      columnNumber: 9
+    })])).toBe(d`
+       class Bar:
+           prop1: str
+
+       class Foo:
+           test1: Bar = None
+           def test_method() -> Bar:
+               pass
+
+
+       inst = Foo()
+       inst.test1.prop1
+       inst.test_method()
+    `);
+  });
+  it("handles late resolved refkeys correctly", () => {
+    const rk1 = refkey();
+    const rk2 = refkey();
+    expect(toSourceText([_$createComponent(py.StatementList, {
+      get children() {
+        return [_$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              refkey: rk1
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 546,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              refkey: rk2
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 547,
+              columnNumber: 13
+            })];
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 545,
+          columnNumber: 11
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "test1",
+          refkey: rk1,
+          initializer: 1
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 549,
+          columnNumber: 11
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "test1",
+          refkey: rk2,
+          initializer: 2
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 550,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 544,
+      columnNumber: 9
+    })])).toBe(d`
+      test1.test1_2_test
+      test1 = 1
+      test1_2_test = 2
+    `);
+  });
+  it("creates a full reference to the first refkey", () => {
+    const rk1 = refkey();
+    const res = toSourceTextMultiple([_$createComponent(SourceFile, {
+      path: "source.py",
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "importMe",
+          refkey: rk1
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 564,
+          columnNumber: 9
+        });
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 563,
+      columnNumber: 7
+    }), _$createComponent(SourceFile, {
+      path: "index.py",
+      get children() {
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: rk1
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 569,
+                  columnNumber: 13
+                }), _$createComponent(py.MemberExpression.Part, {
+                  id: "foo"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 570,
+                  columnNumber: 13
+                })];
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 568,
+              columnNumber: 11
+            });
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 567,
+          columnNumber: 9
+        });
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 566,
+      columnNumber: 7
+    })]);
+    assertFileContents(res, {
+      "index.py": d`
+        from source import import_me
+
+        import_me.foo
+      `
+    });
+  });
+});
+describe("with function calls", () => {
+  it("handles simple function calls correctly", () => {
+    expect(toSourceText([_$createComponent(py.MemberExpression, {
+      get children() {
+        return [_$createComponent(py.MemberExpression.Part, {
+          id: "myFunction"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 591,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          args: [1, 2]
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 592,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 590,
+      columnNumber: 9
+    })])).toBe(d`
+      myFunction(1, 2)
+    `);
+  });
+  it("handles function calls with references correctly", () => {
+    const xRefkey = refkey();
+    const yRefkey = refkey();
+    expect(toSourceText([_$createComponent(py.StatementList, {
+      get children() {
+        return [_$createComponent(py.VariableDeclaration, {
+          name: "x",
+          initializer: 1,
+          refkey: xRefkey
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 606,
+          columnNumber: 11
+        }), _$createComponent(py.VariableDeclaration, {
+          name: "y",
+          initializer: 2,
+          refkey: yRefkey
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 607,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              id: "myFunction"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 609,
+              columnNumber: 13
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: [xRefkey, yRefkey]
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 610,
+              columnNumber: 13
+            })];
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 608,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 605,
+      columnNumber: 9
+    })])).toBe(d`
+      x = 1
+      y = 2
+      myFunction(x, y)
+    `);
+  });
+  it("handles method calls correctly", () => {
+    expect(toSourceText([_$createComponent(py.MemberExpression, {
+      get children() {
+        return [_$createComponent(py.MemberExpression.Part, {
+          id: "method1"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 625,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          args: [1, 2]
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 626,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          args: []
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 627,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          id: "method2"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 628,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          args: []
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 629,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          id: "prop"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 630,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 624,
+      columnNumber: 9
+    })])).toBe(d`
+      method1(1, 2)().method2().prop
+    `);
+  });
+  it("handles function calls correctly", () => {
+    expect(toSourceText([_$createComponent(py.MemberExpression, {
+      get children() {
+        return [_$createComponent(py.MemberExpression.Part, {
+          id: "myFunction"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 642,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          args: [1, 2]
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 643,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          id: "prop"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 644,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 641,
+      columnNumber: 9
+    })])).toBe(d`
+      myFunction(1, 2).prop
+    `);
+  });
+  it("handles function calls correctly", () => {
+    expect(toSourceText([_$createComponent(py.MemberExpression, {
+      get children() {
+        return [_$createComponent(py.MemberExpression.Part, {
+          id: "myFunction"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 656,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          args: [1, 2]
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 657,
+          columnNumber: 11
+        }), _$createComponent(py.MemberExpression.Part, {
+          id: "prop"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 658,
+          columnNumber: 11
+        })];
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 655,
+      columnNumber: 9
+    })])).toBe(d`
+      myFunction(1, 2).prop
+    `);
+  });
+});
+describe("formatting", () => {
+  describe("simple chains", () => {
+    it("just dots", () => {
+      expect(toSourceText([_$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "four"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 674,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "four"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 675,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "two"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 676,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "two"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 677,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "two"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 678,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "two"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 679,
+            columnNumber: 15
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 673,
+        columnNumber: 13
+      })], {
+        printOptions: {
+          printWidth: 14
+        }
+      })).toBe(d`
+        four.four \\
+            .two.two \\
+            .two.two
+      `);
+    });
+    it("bracket breaks", () => {
+      expect(toSourceText([_$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "obj"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 696,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "property-name"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 697,
+            columnNumber: 15
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "prop"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 698,
+            columnNumber: 15
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 695,
+        columnNumber: 13
+      })], {
+        printOptions: {
+          printWidth: 12
+        }
+      })).toBe(d`
+        obj[
+            "property-name"
+        ].prop
+      `);
+    });
+  });
+  describe("call chains", () => {
+    it("renders multiple calls on the same line when there are no breaks and they fit", () => {
+      expect(toSourceText([_$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "z"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 716,
+            columnNumber: 13
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "object"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 717,
+            columnNumber: 13
+          }), _$createComponent(py.MemberExpression.Part, {
+            args: true
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 718,
+            columnNumber: 13
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "partial"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 719,
+            columnNumber: 13
+          }), _$createComponent(py.MemberExpression.Part, {
+            args: true
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 720,
+            columnNumber: 13
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "optional"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 721,
+            columnNumber: 13
+          }), _$createComponent(py.MemberExpression.Part, {
+            args: true
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 722,
+            columnNumber: 13
+          })];
+        }
+      }, {
+        fileName: import.meta.url,
+        lineNumber: 715,
+        columnNumber: 11
+      })])).toBe(d`
+        z.object().partial().optional()
+      `);
+    });
+  });
+});
+//# sourceMappingURL=memberexpressions.test.js.map