1 files changed, 8 insertions, 3 deletions
diff --git a/tests/drivers/storage_analysis_test.js b/tests/drivers/storage_analysis_test.js
index da967f5..774fa89 100644
--- a/tests/drivers/storage_analysis_test.js
+++ b/tests/drivers/storage_analysis_test.js
@@ -26,7 +26,7 @@ function check(text, needle) {
     return { diagnostics, warnings: diagnostics.filter(d => d.severity === 2), hover: replies.find(m => m.id === 2)?.result?.contents?.value };
 }
 const prefix = 'unchecked fn main() {\n';
-for (const [type, length] of [['int', 3], ['u8', 3], ['fixed []int', 3], ['[]u8', 3]]) {
+for (const [type, length] of [['int', 3], ['u8', 3], ['fixed int[?]', 3], ['u8[?]', 3]]) {
     const r = check(prefix + ` let p = new(${type}, ${length});\n let alias = p;\n return alias[${length}];\n}\n`, 'alias[');
     assert.strictEqual(r.warnings.length, 1, JSON.stringify(r.diagnostics));
     assert.match(r.warnings[0].message, /out-of-bounds.*index 3.*length 3/);
@@ -56,16 +56,21 @@ const safe = [
  prefix + ' let p = new(int, 50); return p[300 as u8]; }',
  prefix + ' let p = new(int, 3); let q = p; q[2] = 7; free(p); return 0; }',
  prefix + ' let p = new(3); free(p); p = new(3); return p[0]; }',
- 'fn unknown(int* p) {}\n' + prefix + ' let p = new(int, 3); unknown(p); return p[9]; }',
+ 'unchecked fn unknown(int* p) {}\n' + prefix + ' let p = new(int, 3); unknown(p); return p[9]; }',
  'fn maybe(int n) { return n; }\n' + prefix + ' let p = new(int, 3); return p[maybe(9)]; }',
  prefix + ' let p = new(int, 3); if (1) free(p); return p[0]; }',
  prefix + ' let p = new(int, 3); while (0) free(p); return p[0]; }',
  prefix + ' let p = new(int, 3); let f = fn() { free(p); return 0; }; return p[0]; }',
- prefix + ' let p = new([]int, 0); push(p, 9); return p[0]; }',
+ prefix + ' let p = new(int[?], 0); push(p, 9); return p[0]; }',
 ];
 for (const source of safe) assert.deepStrictEqual(check(source).warnings, [], source);
+const preserved = check('int read(int* p) { return p[0]; }\n' + prefix + ' let p = new(int, 3); read(p); return p[9]; }');
+assert.match(preserved.warnings[0]?.message || '', /index 9.*length 3/);
 const unknown = check('int read(int* p) { return p[0]; }\nfn main() { int n = 1; return read(&n); }\n', 'p[');
 assert.match(unknown.hover, /size: unknown.*lifetime: unknown/);
 const stack = check('fn main() { let a[3]; return a[2]; }\n', 'a[');
 assert.match(stack.hover, /size: 3 elements.*lifetime: stack storage/);
+const replacement = check('fn main() {\n let owner = new(1), replacement = new(2);\n free(owner); owner = replacement;\n return owner[0];\n}\n', 'replacement;');
+assert.ok(!replacement.diagnostics.some(d => d.severity === 1), JSON.stringify(replacement.diagnostics));
+assert.match(replacement.hover, /moved: replaces a released owner in `owner` on line 3/);
 console.log('ok   definite bounds, free and scope lifetimes, alias hovers, and conservative unknown cases');