@platformos/platformos-language-server-common 0.0.7 → 0.0.9

package/src/renamed/handlers/AssetRenameHandler.spec.ts

@@ -10,7 +10,7 @@ import { RenameHandler } from '../RenameHandler';
 
 describe('Module: AssetRenameHandler', () => {
   const mockRoot = 'mock-fs:';
-  const findThemeRootURI = async () => mockRoot;
+  const findAppRootURI = async () => mockRoot;
   let capabilities: ClientCapabilities;
   let documentManager: DocumentManager;
   let handler: RenameHandler;
@@ -22,14 +22,14 @@ describe('Module: AssetRenameHandler', () => {
     capabilities = new ClientCapabilities();
     fs = new MockFileSystem(
       {
-        'assets/oldName.js': 'console.log("Hello, world!")',
-        'sections/section.liquid': `<script src="{{ 'oldName.js' | asset_url }}" defer></script>`,
-        'blocks/block.liquid': `{{ 'oldName.js' | asset_url | script_tag }} oldName.js`,
+        'app/assets/oldName.js': 'console.log("Hello, world!")',
+        'app/views/partials/section.liquid': `<script src="{{ 'oldName.js' | asset_url }}" defer></script>`,
+        'app/views/partials/block.liquid': `{{ 'oldName.js' | asset_url | script_tag }} oldName.js`,
       },
       mockRoot,
     );
     documentManager = new DocumentManager(fs);
-    handler = new RenameHandler(connection, capabilities, documentManager, findThemeRootURI);
+    handler = new RenameHandler(connection, capabilities, documentManager, findAppRootURI);
   });
 
   describe('when the client does not support workspace/applyEdit', () => {
@@ -45,8 +45,8 @@ describe('Module: AssetRenameHandler', () => {
       await handler.onDidRenameFiles({
         files: [
           {
-            oldUri: 'mock-fs:/assets/oldName.js',
-            newUri: 'mock-fs:/assets/newName.js',
+            oldUri: 'mock-fs:/app/assets/oldName.js',
+            newUri: 'mock-fs:/app/assets/newName.js',
           },
         ],
       });
@@ -67,8 +67,8 @@ describe('Module: AssetRenameHandler', () => {
       await handler.onDidRenameFiles({
         files: [
           {
-            oldUri: 'mock-fs:/assets/oldName.js',
-            newUri: 'mock-fs:/assets/newName.js',
+            oldUri: 'mock-fs:/app/assets/oldName.js',
+            newUri: 'mock-fs:/app/assets/newName.js',
           },
         ],
       });
@@ -90,7 +90,7 @@ describe('Module: AssetRenameHandler', () => {
           documentChanges: [
             {
               textDocument: {
-                uri: 'mock-fs:/sections/section.liquid',
+                uri: 'mock-fs:/app/views/partials/section.liquid',
                 version: null,
               },
               edits: [expectedTextEdit],
@@ -98,7 +98,7 @@ describe('Module: AssetRenameHandler', () => {
             },
             {
               textDocument: {
-                uri: 'mock-fs:/blocks/block.liquid',
+                uri: 'mock-fs:/app/views/partials/block.liquid',
                 version: null,
               },
               edits: [expectedTextEdit],
@@ -113,8 +113,8 @@ describe('Module: AssetRenameHandler', () => {
       await handler.onDidRenameFiles({
         files: [
           {
-            oldUri: 'mock-fs:/assets/oldName.js',
-            newUri: 'mock-fs:/assets/newName.js',
+            oldUri: 'mock-fs:/app/assets/oldName.js',
+            newUri: 'mock-fs:/app/assets/newName.js',
           },
         ],
       });
@@ -122,9 +122,9 @@ describe('Module: AssetRenameHandler', () => {
       const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
       const expectedFs = new MockFileSystem(
         {
-          'assets/oldName.js': 'console.log("Hello, world!")',
-          'sections/section.liquid': `<script src="{{ 'newName.js' | asset_url }}" defer></script>`,
-          'blocks/block.liquid': `{{ 'newName.js' | asset_url | script_tag }} oldName.js`,
+          'app/assets/oldName.js': 'console.log("Hello, world!")',
+          'app/views/partials/section.liquid': `<script src="{{ 'newName.js' | asset_url }}" defer></script>`,
+          'app/views/partials/block.liquid': `{{ 'newName.js' | asset_url | script_tag }} oldName.js`,
         },
         'mock-fs:',
       );
@@ -145,8 +145,8 @@ describe('Module: AssetRenameHandler', () => {
       await handler.onDidRenameFiles({
         files: [
           {
-            oldUri: 'mock-fs:/assets/oldName.js',
-            newUri: `mock-fs:/assets/newName.js.liquid`,
+            oldUri: 'mock-fs:/app/assets/oldName.js',
+            newUri: `mock-fs:/app/assets/newName.js.liquid`,
           },
         ],
       });
@@ -154,9 +154,9 @@ describe('Module: AssetRenameHandler', () => {
       const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
       const expectedFs = new MockFileSystem(
         {
-          'assets/oldName.js': 'console.log("Hello, world!")',
-          'sections/section.liquid': `<script src="{{ 'newName.js' | asset_url }}" defer></script>`,
-          'blocks/block.liquid': `{{ 'newName.js' | asset_url | script_tag }} oldName.js`,
+          'app/assets/oldName.js': 'console.log("Hello, world!")',
+          'app/views/partials/section.liquid': `<script src="{{ 'newName.js' | asset_url }}" defer></script>`,
+          'app/views/partials/block.liquid': `{{ 'newName.js' | asset_url | script_tag }} oldName.js`,
         },
         'mock-fs:',
       );
@@ -176,19 +176,19 @@ describe('Module: AssetRenameHandler', () => {
     it('handles .css.liquid files', async () => {
       fs = new MockFileSystem(
         {
-          'assets/oldName.css.liquid': 'body { color: red; }',
-          'sections/section.liquid': `{% echo 'oldName.css' | asset_url | stylesheet_tag %}`,
+          'app/assets/oldName.css.liquid': 'body { color: red; }',
+          'app/views/partials/section.liquid': `{% echo 'oldName.css' | asset_url | stylesheet_tag %}`,
         },
         'mock-fs:',
       );
       documentManager = new DocumentManager(fs);
-      handler = new RenameHandler(connection, capabilities, documentManager, findThemeRootURI);
+      handler = new RenameHandler(connection, capabilities, documentManager, findAppRootURI);
 
       await handler.onDidRenameFiles({
         files: [
           {
-            oldUri: 'mock-fs:/assets/oldName.css.liquid',
-            newUri: `mock-fs:/assets/newName.css.liquid`,
+            oldUri: 'mock-fs:/app/assets/oldName.css.liquid',
+            newUri: `mock-fs:/app/assets/newName.css.liquid`,
           },
         ],
       });
@@ -196,8 +196,8 @@ describe('Module: AssetRenameHandler', () => {
       const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
       const expectedFs = new MockFileSystem(
         {
-          'assets/newName.css.liquid': 'body { color: red; }',
-          'sections/section.liquid': `{% echo 'newName.css' | asset_url | stylesheet_tag %}`,
+          'app/assets/newName.css.liquid': 'body { color: red; }',
+          'app/views/partials/section.liquid': `{% echo 'newName.css' | asset_url | stylesheet_tag %}`,
         },
         'mock-fs:',
       );

package/src/renamed/handlers/AssetRenameHandler.ts

@@ -12,7 +12,7 @@ import { ClientCapabilities } from '../../ClientCapabilities';
 import { DocumentManager, isLiquidSourceCode } from '../../documents';
 import { assetName, isAsset } from '../../utils/uri';
 import { BaseRenameHandler } from '../BaseRenameHandler';
-import { FindThemeRootURI } from '../../internal-types';
+import { FindAppRootURI } from '../../internal-types';
 
 /**
  * The AssetRenameHandler will handle asset renames.
@@ -22,7 +22,7 @@ import { FindThemeRootURI } from '../../internal-types';
  *
  * We'll do that for `.(css|js).liquid` files as well
  *
- * We'll do this by visiting all the liquid files in the theme and looking for
+ * We'll do this by visiting all the liquid files in the app and looking for
  * string | asset_url Variable nodes that reference the old asset. We'll then create a
  * WorkspaceEdit that changes the references to the new asset.
  */
@@ -31,7 +31,7 @@ export class AssetRenameHandler implements BaseRenameHandler {
     private documentManager: DocumentManager,
     private connection: Connection,
     private capabilities: ClientCapabilities,
-    private findThemeRootURI: FindThemeRootURI,
+    private findAppRootURI: FindAppRootURI,
   ) {}
 
   async onDidRenameFiles(params: RenameFilesParams): Promise<void> {
@@ -44,11 +44,11 @@ export class AssetRenameHandler implements BaseRenameHandler {
     // Only preload if you have something to do (folder renames are not supported)
     if (relevantRenames.length !== 1) return;
     const rename = relevantRenames[0];
-    const rootUri = await this.findThemeRootURI(path.dirname(params.files[0].oldUri));
+    const rootUri = await this.findAppRootURI(path.dirname(params.files[0].oldUri));
     if (!rootUri) return;
     await this.documentManager.preload(rootUri);
-    const theme = this.documentManager.theme(rootUri, true);
-    const liquidSourceCodes = theme.filter(isLiquidSourceCode);
+    const app = this.documentManager.app(rootUri, true);
+    const liquidSourceCodes = app.filter(isLiquidSourceCode);
 
     const oldAssetName = assetName(rename.oldUri);
     const newAssetName = assetName(rename.newUri);

package/src/renamed/handlers/PartialRenameHandler.spec.ts

@@ -9,7 +9,7 @@ import { RenameHandler } from '../RenameHandler';
 
 describe('Module: PartialRenameHandler', () => {
   const mockRoot = 'mock-fs:';
-  const findThemeRootURI = async () => mockRoot;
+  const findAppRootURI = async () => mockRoot;
   let capabilities: ClientCapabilities;
   let documentManager: DocumentManager;
   let handler: RenameHandler;
@@ -21,15 +21,15 @@ describe('Module: PartialRenameHandler', () => {
     capabilities = new ClientCapabilities();
     fs = new MockFileSystem(
       {
-        'sections/section.liquid': `<div>{% render 'oldName', foo: 'bar' %}oldName</div>`,
-        'blocks/block.liquid': `<div>{% render 'oldName', foo: 'baz' %}</div>`,
+        'app/views/partials/page.liquid': `<div>{% render 'oldName', foo: 'bar' %}oldName</div>`,
+        'app/lib/component.liquid': `<div>{% render 'oldName', foo: 'baz' %}</div>`,
         'app/views/partials/oldName.liquid': `<div>oldName{%</div>`,
         'app/views/partials/other.liquid': `<div>{% render 'oldName' %}{% render 'other' %}</div>`,
       },
       mockRoot,
     );
     documentManager = new DocumentManager(fs);
-    handler = new RenameHandler(connection, capabilities, documentManager, findThemeRootURI);
+    handler = new RenameHandler(connection, capabilities, documentManager, findAppRootURI);
   });
 
   describe('when the client does not support workspace/applyEdit', () => {
@@ -90,7 +90,7 @@ describe('Module: PartialRenameHandler', () => {
           documentChanges: [
             {
               textDocument: {
-                uri: 'mock-fs:/sections/section.liquid',
+                uri: 'mock-fs:/app/views/partials/page.liquid',
                 version: null,
               },
               edits: [expectedTextEdit],
@@ -98,7 +98,7 @@ describe('Module: PartialRenameHandler', () => {
             },
             {
               textDocument: {
-                uri: 'mock-fs:/blocks/block.liquid',
+                uri: 'mock-fs:/app/views/partials/other.liquid',
                 version: null,
               },
               edits: [expectedTextEdit],
@@ -106,7 +106,7 @@ describe('Module: PartialRenameHandler', () => {
             },
             {
               textDocument: {
-                uri: 'mock-fs:/app/views/partials/other.liquid',
+                uri: 'mock-fs:/app/lib/component.liquid',
                 version: null,
               },
               edits: [expectedTextEdit],
@@ -130,8 +130,8 @@ describe('Module: PartialRenameHandler', () => {
       const params: ApplyWorkspaceEditParams = connection.spies.sendRequest.mock.calls[0][1];
       const expectedFs = new MockFileSystem(
         {
-          'sections/section.liquid': `<div>{% render 'newName', foo: 'bar' %}oldName</div>`,
-          'blocks/block.liquid': `<div>{% render 'newName', foo: 'baz' %}</div>`,
+          'app/views/partials/page.liquid': `<div>{% render 'newName', foo: 'bar' %}oldName</div>`,
+          'app/lib/component.liquid': `<div>{% render 'newName', foo: 'baz' %}</div>`,
           'app/views/partials/newName.liquid': `<div>oldName{%</div>`,
           'app/views/partials/other.liquid': `<div>{% render 'newName' %}{% render 'other' %}</div>`,
         },

package/src/renamed/handlers/PartialRenameHandler.ts

@@ -12,7 +12,7 @@ import { ClientCapabilities } from '../../ClientCapabilities';
 import { DocumentManager, isLiquidSourceCode } from '../../documents';
 import { isPartial, partialName } from '../../utils/uri';
 import { BaseRenameHandler } from '../BaseRenameHandler';
-import { FindThemeRootURI } from '../../internal-types';
+import { FindAppRootURI } from '../../internal-types';
 
 /**
  * The PartialRenameHandler will handle partial renames.
@@ -22,7 +22,7 @@ import { FindThemeRootURI } from '../../internal-types';
  *
  *   {% render 'oldName' %} -> {% render 'newName' %}
  *
- * We'll do this by visiting all the liquid files in the theme and looking for
+ * We'll do this by visiting all the liquid files in the app and looking for
  * render and include tags that reference the old partial. We'll then create a
  * WorkspaceEdit that changes the references to the new partial.
  */
@@ -31,7 +31,7 @@ export class PartialRenameHandler implements BaseRenameHandler {
     private documentManager: DocumentManager,
     private connection: Connection,
     private capabilities: ClientCapabilities,
-    private findThemeRootURI: FindThemeRootURI,
+    private findAppRootURI: FindAppRootURI,
   ) {}
 
   async onDidRenameFiles(params: RenameFilesParams): Promise<void> {
@@ -43,11 +43,11 @@ export class PartialRenameHandler implements BaseRenameHandler {
     // Only preload if you have something to do (folder renames are not supported)
     if (relevantRenames.length !== 1) return;
     const rename = relevantRenames[0];
-    const rootUri = await this.findThemeRootURI(path.dirname(params.files[0].oldUri));
+    const rootUri = await this.findAppRootURI(path.dirname(params.files[0].oldUri));
     if (!rootUri) return;
     await this.documentManager.preload(rootUri);
-    const theme = this.documentManager.theme(rootUri, true);
-    const liquidSourceCodes = theme.filter(isLiquidSourceCode);
+    const app = this.documentManager.app(rootUri, true);
+    const liquidSourceCodes = app.filter(isLiquidSourceCode);
     const oldPartialName = partialName(rename.oldUri);
     const newPartialName = partialName(rename.newUri);
     const editLabel = `Rename partial '${oldPartialName}' to '${newPartialName}'`;
@@ -73,13 +73,13 @@ export class PartialRenameHandler implements BaseRenameHandler {
           if (typeof node.markup === 'string') {
             return;
           }
-          const snippet = node.markup.snippet;
-          if (snippet.type === NodeTypes.String && snippet.value === oldPartialName) {
+          const partial = node.markup.partial;
+          if (partial.type === NodeTypes.String && partial.value === oldPartialName) {
             return {
               newText: `${newPartialName}`,
               range: Range.create(
-                textDocument.positionAt(snippet.position.start + 1), // +1 to skip the opening quote
-                textDocument.positionAt(snippet.position.end - 1), // -1 to skip the closing quote
+                textDocument.positionAt(partial.position.start + 1), // +1 to skip the opening quote
+                textDocument.positionAt(partial.position.end - 1), // -1 to skip the closing quote
               ),
             };
           }

package/src/server/{ThemeGraphManager.ts → AppGraphManager.ts}

@@ -1,13 +1,7 @@
-import {
-  path,
-  recursiveReadDirectory,
-  SectionSchema,
-  SourceCodeType,
-  ThemeBlockSchema,
-} from '@platformos/platformos-check-common';
+import { path, SourceCodeType } from '@platformos/platformos-check-common';
 import { AbstractFileSystem } from '@platformos/platformos-common';
 import {
-  buildThemeGraph,
+  buildAppGraph,
   getWebComponentMap,
   IDependencies as GraphDependencies,
   Location,
@@ -22,40 +16,36 @@ import {
   AugmentedLocation,
   AugmentedLocationWithExistence,
   AugmentedReference,
-  ThemeGraphDidUpdateNotification,
+  AppGraphDidUpdateNotification,
 } from '../types';
 import { debounce } from '../utils';
-import { FindThemeRootURI } from '../internal-types';
+import { FindAppRootURI } from '../internal-types';
 
-export class ThemeGraphManager {
-  graphs: Map<string, ReturnType<typeof buildThemeGraph>> = new Map();
+export class AppGraphManager {
+  graphs: Map<string, ReturnType<typeof buildAppGraph>> = new Map();
 
   constructor(
     private connection: Connection,
     private documentManager: DocumentManager,
     private fs: AbstractFileSystem,
-    private findThemeRootURI: FindThemeRootURI,
+    private findAppRootURI: FindAppRootURI,
   ) {}
 
-  async getThemeGraphForURI(uri: string) {
-    const rootUri = await this.findThemeRootURI(uri);
+  async getAppGraphForURI(uri: string) {
+    const rootUri = await this.findAppRootURI(uri);
     if (!rootUri) {
       return undefined;
     }
 
     if (!this.graphs.has(rootUri)) {
-      this.graphs.set(rootUri, this.buildThemeGraph(rootUri));
+      this.graphs.set(rootUri, this.buildAppGraph(rootUri));
     }
 
     return this.graphs.get(rootUri);
   }
 
-  async getReferences(
-    uri: string,
-    offset?: number,
-    { includeIndirect = true, includePreset = true } = {},
-  ) {
-    const graph = await this.getThemeGraphForURI(uri);
+  async getReferences(uri: string, offset?: number, { includeIndirect = true } = {}) {
+    const graph = await this.getAppGraphForURI(uri);
     if (!graph) return [];
 
     const module = graph.modules[uri];
@@ -64,7 +54,6 @@ export class ThemeGraphManager {
     const includedTypes: (AugmentedReference['type'] | undefined)[] = [
       'direct',
       includeIndirect ? 'indirect' : undefined,
-      includePreset ? 'preset' : undefined,
     ];
 
     const refs = module.references.filter((dep) => includedTypes.includes(dep.type));
@@ -84,31 +73,16 @@ export class ThemeGraphManager {
     );
   }
 
-  async getDependencies(
-    uri: string,
-    offset?: number,
-    { includeIndirect = true, includePreset = true } = {},
-  ) {
-    const graph = await this.getThemeGraphForURI(uri);
+  async getDependencies(uri: string, offset?: number, { includeIndirect = true } = {}) {
+    const graph = await this.getAppGraphForURI(uri);
     if (!graph) return [];
 
-    let module = graph.modules[uri];
-    if (!module) {
-      // If the module is not found, we might be dealing with dead code.
-      // dead code doesn't show up in the graph, but it might still have dependencies.
-      // So we're building a smaller graph with that file as entry point to figure
-      // out what it depends on.
-      const deadCodeGraph = await this.buildThemeGraph(graph.rootUri, [uri]);
-      module = deadCodeGraph.modules[uri];
-    }
-
-    // If the module is still not found, we return an empty array.
+    const module = graph.modules[uri];
     if (!module) return [];
 
     const includedTypes: (AugmentedReference['type'] | undefined)[] = [
       'direct',
       includeIndirect ? 'indirect' : undefined,
-      includePreset ? 'preset' : undefined,
     ];
 
     const deps = module.dependencies.filter((dep) => includedTypes.includes(dep.type)) ?? [];
@@ -148,33 +122,6 @@ export class ThemeGraphManager {
     };
   }
 
-  async deadCode(rootUri: string): Promise<string[]> {
-    const graph = await this.getThemeGraphForURI(rootUri);
-    if (!graph) return [];
-
-    const files = await recursiveReadDirectory(
-      this.fs,
-      rootUri,
-      ([uri]) =>
-        ['assets', 'blocks', 'layout', 'sections', 'snippets', 'templates'].some((dir) =>
-          uri.startsWith(path.join(rootUri, dir)),
-        ) &&
-        (uri.endsWith('.liquid') ||
-          uri.endsWith('.json') ||
-          uri.endsWith('.js') ||
-          uri.endsWith('.css')),
-    );
-
-    const unusedFiles = new Set<string>();
-    for (const file of files) {
-      if (!graph.modules[file]) {
-        unusedFiles.add(file);
-      }
-    }
-
-    return Array.from(unusedFiles).sort();
-  }
-
   public operationQueue: string[] = [];
 
   async rename(oldUri: string, newUri: string) {
@@ -203,23 +150,23 @@ export class ThemeGraphManager {
     if (operations.length === 0) return;
 
     const anyUri = operations[0];
-    const rootUri = await this.findThemeRootURI(anyUri);
+    const rootUri = await this.findAppRootURI(anyUri);
     if (!rootUri) return;
 
     const graph = await this.graphs.get(rootUri);
     if (!graph) return;
 
     this.graphs.delete(rootUri);
-    await this.getThemeGraphForURI(rootUri);
-    this.connection.sendNotification(ThemeGraphDidUpdateNotification.type, { uri: rootUri });
+    await this.getAppGraphForURI(rootUri);
+    this.connection.sendNotification(AppGraphDidUpdateNotification.type, { uri: rootUri });
   }, 500);
 
-  private buildThemeGraph = async (rootUri: string, entryPoints?: string[]) => {
+  private buildAppGraph = async (rootUri: string, entryPoints?: string[]) => {
     const { documentManager } = this;
     await documentManager.preload(rootUri);
 
     const dependencies = await this.graphDependencies(rootUri);
-    return buildThemeGraph(rootUri, dependencies, entryPoints);
+    return buildAppGraph(rootUri, dependencies, entryPoints);
   };
 
   private getSourceCode = async (uri: string) => {
@@ -236,27 +183,11 @@ export class ThemeGraphManager {
   }
 
   private async graphDependencies(rootUri: string): Promise<GraphDependencies> {
-    const { documentManager, fs, getSourceCode } = this;
+    const { fs, getSourceCode } = this;
     const webComponentDefs = await this.getWebComponentMap(rootUri);
     return {
       fs: fs,
       getSourceCode: getSourceCode,
-      async getBlockSchema(name: string) {
-        const blockUri = path.join(rootUri, 'blocks', `${name}.liquid`);
-        const doc = documentManager.get(blockUri);
-        if (!doc || doc.type !== SourceCodeType.LiquidHtml) {
-          return;
-        }
-        return (await doc.getSchema()) as ThemeBlockSchema;
-      },
-      async getSectionSchema(name) {
-        const sectionUri = path.join(rootUri, 'sections', `${name}.liquid`);
-        const doc = documentManager.get(sectionUri);
-        if (!doc || doc.type !== SourceCodeType.LiquidHtml) {
-          return;
-        }
-        return (await doc.getSchema()) as SectionSchema;
-      },
       getWebComponentDefinitionReference(customElementName: string) {
         return webComponentDefs.get(customElementName);
       },

package/src/server/safe.ts

@@ -14,7 +14,7 @@
  * @param defaultReturnValue
  *
  * @example
- * const getThemeBlockNames = safe(async function () { ... }, []);
+ * const getPartialNames = safe(async function () { ... }, []);
  */
 export const safe = <T extends (...args: any[]) => Promise<any>>(
   fn: T,

package/src/server/startServer.spec.ts

@@ -1,5 +1,5 @@
 import { allChecks, path } from '@platformos/platformos-check-common';
-import { MockFileSystem, MockTheme } from '@platformos/platformos-check-common/dist/test';
+import { MockFileSystem, MockApp } from '@platformos/platformos-check-common/dist/test';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import {
   DidChangeConfigurationNotification,
@@ -16,8 +16,8 @@ import { startServer } from './startServer';
 const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
 
 describe('Module: server', () => {
-  const mockRoot = path.normalize('browser:/theme');
-  const filePath = 'app/code.liquid';
+  const mockRoot = path.normalize('browser:/app');
+  const filePath = 'app/views/partials/code.liquid';
   const fileURI = path.join(mockRoot, filePath);
   const fileContents = `{% render 'foo' %}`;
   let checkOnChange: boolean | null = null;
@@ -25,7 +25,7 @@ describe('Module: server', () => {
   let checkOnOpen: boolean | null = null;
   let connection: MockConnection;
   let dependencies: ReturnType<typeof getDependencies>;
-  let fileTree: MockTheme;
+  let fileTree: MockApp;
   let logger: any;
 
   beforeEach(() => {
@@ -60,7 +60,7 @@ describe('Module: server', () => {
 
     fileTree = {
       '.pos': '',
-      'app/code.liquid': fileContents,
+      'app/views/partials/code.liquid': fileContents,
       '.git/test': 'test',
       'modules/test': 'test',
     };
@@ -375,25 +375,23 @@ describe('Module: server', () => {
     return flushAsync();
   }
 
-  function getDependencies(logger: any, fileTree: MockTheme): Dependencies {
-    const MissingTemplate = allChecks.filter((c) => c.meta.code === 'MissingTemplate');
+  function getDependencies(logger: any, fileTree: MockApp): Dependencies {
+    const MissingTemplate = allChecks.filter((c) => c.meta.code === 'MissingPartial');
 
     return {
       fs: new MockFileSystem(fileTree, mockRoot),
       log: logger,
       loadConfig: async () => ({
-        context: 'theme',
         settings: {},
         checks: MissingTemplate,
         rootUri: mockRoot,
       }),
-      themeDocset: {
+      platformosDocset: {
         graphQL: async () => null,
         filters: async () => [],
         objects: async () => [],
         liquidDrops: async () => [],
         tags: async () => [],
-        systemTranslations: async () => ({}),
       },
       jsonValidationSet: {
         schemas: async () => [],
@@ -403,11 +401,11 @@ describe('Module: server', () => {
 
   function missingTemplateDiagnostic() {
     return {
-      code: 'MissingTemplate',
+      code: 'MissingPartial',
       codeDescription: { href: expect.any(String) },
       message: "'foo' does not exist",
       severity: 1,
-      source: 'theme-check',
+      source: 'platformos-check',
       range: {
         start: {
           character: 10,