From 8007b11aa994a3f2734615fb7b4bef50fa2b092b Mon Sep 17 00:00:00 2001
From: Marco Salazar <marco@elementl.com>
Date: Mon, 20 Jan 2025 20:37:56 -0500
Subject: [PATCH] update mock import

---
 .../packages/ui-core/src/assets/__tests__/AssetTables.test.tsx | 2 +-
 .../packages/ui-core/src/selection/SelectionInputUtil.ts       | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/AssetTables.test.tsx b/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/AssetTables.test.tsx
index cf370f5df33e9..5e33f9441a04d 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/AssetTables.test.tsx
+++ b/js_modules/dagster-ui/packages/ui-core/src/assets/__tests__/AssetTables.test.tsx
@@ -33,7 +33,7 @@ const MOCKS = [
 // This file must be mocked because Jest can't handle `import.meta.url`.
 jest.mock('../../graph/asyncGraphLayout', () => ({}));
 
-jest.mock('shared/asset-selection/useAssetSelectionInput', () => {
+jest.mock('shared/asset-selection/input/useAssetSelectionInput', () => {
   const mock: typeof useAssetSelectionInput = ({
     assets,
     assetsLoading,
diff --git a/js_modules/dagster-ui/packages/ui-core/src/selection/SelectionInputUtil.ts b/js_modules/dagster-ui/packages/ui-core/src/selection/SelectionInputUtil.ts
index 565bb3aaacf4b..1b55df25c2a75 100644
--- a/js_modules/dagster-ui/packages/ui-core/src/selection/SelectionInputUtil.ts
+++ b/js_modules/dagster-ui/packages/ui-core/src/selection/SelectionInputUtil.ts
@@ -1,4 +1,5 @@
 import {ParserRuleContext} from 'antlr4ts';
+import {AttributeValueContext} from './generated/SelectionAutoCompleteParser';
 
 export const removeQuotesFromString = (value: string) => {
   if (value.length > 1 && value[0] === '"' && value[value.length - 1] === '"') {
@@ -17,6 +18,8 @@ export function getValueNodeValue(ctx: ParserRuleContext | null) {
       return ctx.text.slice(0, -1);
     case 'QuotedStringValueContext':
       return ctx.text.slice(1, -1);
+    case 'AttributeValueContext':
+      return getValueNodeValue((ctx as AttributeValueContext).value());
     default:
       return ctx?.text ?? '';
   }