add cache evict status

Signed-off-by: Paul Dittamo <[email protected]>
flyteorg · Mar 12, 2024 · e9b4b25 · e9b4b25
1 parent eb7fe2b
commit e9b4b25
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/packages/oss-console/src/components/Executions/CacheStatus.tsx b/packages/oss-console/src/components/Executions/CacheStatus.tsx
@@ -3,6 +3,7 @@ import { type SvgIconProps } from '@mui/material/SvgIcon';
 import Tooltip from '@mui/material/Tooltip';
 import Typography from '@mui/material/Typography';
 import CachedOutlined from '@mui/icons-material/CachedOutlined';
+import DeleteForeverOutlinedIcon from '@mui/icons-material/DeleteForeverOutlined';
 import ErrorOutlined from '@mui/icons-material/ErrorOutlined';
 import InfoOutlined from '@mui/icons-material/InfoOutlined';
 import SmsFailedOutlinedIcon from '@mui/icons-material/SmsFailedOutlined';
@@ -55,6 +56,9 @@ const NodeExecutionCacheStatusIcon: React.ComponentType<
     case CatalogCacheStatus.MAP_CACHE: {
       return <MapCacheIcon {...svgIconProps} ref={ref} data-testid="cache-icon" />;
     }
+    case CatalogCacheStatus.CACHE_EVICTED: {
+      return <DeleteForeverOutlinedIcon {...svgIconProps} ref={ref} data-testid="cache-icon" />;
+    }
     default: {
       assertNever(status as never);
       // @ts-ignore

diff --git a/packages/oss-console/src/components/Executions/test/CacheStatus.test.tsx b/packages/oss-console/src/components/Executions/test/CacheStatus.test.tsx
@@ -88,6 +88,7 @@ describe('Executions > CacheStatus', () => {
       ${CatalogCacheStatus.CACHE_MISS}           | ${cacheStatusMessages[CatalogCacheStatus.CACHE_MISS]}
       ${CatalogCacheStatus.CACHE_POPULATED}      | ${cacheStatusMessages[CatalogCacheStatus.CACHE_POPULATED]}
       ${CatalogCacheStatus.CACHE_PUT_FAILURE}    | ${cacheStatusMessages[CatalogCacheStatus.CACHE_PUT_FAILURE]}
+      ${CatalogCacheStatus.CACHE_EVICTED}        | ${cacheStatusMessages[CatalogCacheStatus.CACHE_EVICTED]}
     `('for each case', ({ cacheStatus, expected }) => {
       it(`renders correct text ${expected} for status ${cacheStatus}`, async () => {
         const { queryByText } = renderComponent({ cacheStatus });

diff --git a/packages/oss-console/src/components/flytegraph/__stories__/CustomNodes.stories.tsx b/packages/oss-console/src/components/flytegraph/__stories__/CustomNodes.stories.tsx
@@ -69,6 +69,7 @@ const CACHE_STATUSES = [
   { status: CatalogCacheStatus.CACHE_MISS, text: 'cache miss' },
   { status: CatalogCacheStatus.CACHE_POPULATED, text: 'cache populated' },
   { status: CatalogCacheStatus.CACHE_PUT_FAILURE, text: 'cache put failure' },
+  { status: CatalogCacheStatus.CACHE_EVICTED, text: 'cache evicted' },
 ];
 
 stories.add('Task Node by Cache Status', () => (