code quality

apps/website/ui/Badge.tsx

@@ -14,6 +14,7 @@ export interface BadgeProps {
   color?: string;
   borderColor?: string;
   transform?: 'none' | 'capitalize' | 'uppercase' | 'lowercase' | string;
+  'data-testid'?: string;
 }
 
 /** @internal */
@@ -31,7 +32,8 @@ export const Badge = ({
   bg,
   color,
   borderColor,
-  transform
+  transform,
+  'data-testid': dataTestId
 }: InternalBadgeProps) => {
   const variantClasses = {
     primary: 'bg-[var(--ui-color-intent-primary)] text-white',
@@ -76,7 +78,7 @@ export const Badge = ({
   ) : children;
 
   return (
-    <Box as="span" className={classes} style={style}>
+    <Box data-testid={dataTestId} as="span" className={classes} style={style}>
       {content}
     </Box>
   );