Fix testcase

Author: owjs3901

libs/sheet/src/lib.rs

@@ -377,12 +377,13 @@ impl StyleSheet {
                                 st.style_order(),
                                 if !single_css { Some(filename) } else { None },
                             )
-                        } else if let Some(StyleProperty::ClassName(cls)) =
-                            style.extract(if !single_css { Some(filename) } else { None })
-                        {
-                            cls
                         } else {
-                            continue;
+                            match st.extract(if !single_css { Some(filename) } else { None }) {
+                                StyleProperty::ClassName(cls)
+                                | StyleProperty::Variable {
+                                    class_name: cls, ..
+                                } => cls,
+                            }
                         };
 
                     if self.add_property_with_layer(
@@ -2360,4 +2361,56 @@ mod tests {
         assert!(css.contains("width:1px"));
         assert!(!css.contains("width:2px"));
     }
+
+    #[test]
+    #[serial]
+    fn test_first_value_without_dollar_prefix_uses_raw_value() {
+        reset_class_map();
+        reset_file_map();
+        let mut sheet = StyleSheet::default();
+
+        let mut styles = FxHashSet::default();
+        // FirstValue resolution but value has no $ prefix — should use the raw value as-is
+        styles.insert(ExtractStyleValue::Static(
+            ExtractStaticStyle::new("width", "100px", 0, None)
+                .with_theme_token_resolution(ThemeTokenResolution::FirstValue),
+        ));
+
+        let (collected, _) = sheet.update_styles(&styles, "test.tsx", true);
+        assert!(collected);
+
+        let css = sheet.create_css(None, false);
+        assert!(css.contains("width:100px"));
+    }
+
+    #[test]
+    #[serial]
+    fn test_first_value_box_shadow_resolves_shadow_token() {
+        reset_class_map();
+        reset_file_map();
+        let mut sheet = StyleSheet::default();
+        let theme: Theme = serde_json::from_str(
+            r#"{
+                "shadow": {
+                    "default": {
+                        "card": ["0 1px 2px #0003", null, "0 4px 8px #0003"]
+                    }
+                }
+            }"#,
+        )
+        .unwrap();
+        sheet.set_theme(theme);
+
+        let mut styles = FxHashSet::default();
+        styles.insert(ExtractStyleValue::Static(
+            ExtractStaticStyle::new("box-shadow", "$card", 0, None)
+                .with_theme_token_resolution(ThemeTokenResolution::FirstValue),
+        ));
+
+        let (collected, _) = sheet.update_styles(&styles, "test.tsx", true);
+        assert!(collected);
+
+        let css = sheet.create_css(None, false);
+        assert!(css.contains("box-shadow:0 1px 2px #0003"));
+    }
 }

libs/sheet/src/theme.rs

@@ -2265,4 +2265,47 @@ mod tests {
             ]
         );
     }
+
+    #[test]
+    fn test_get_shadow_token_levels() {
+        let mut theme = Theme::default();
+        theme.add_shadow(
+            "default",
+            "sm",
+            vec![
+                Some("0 1px 2px rgba(0,0,0,.1)".to_string()),
+                None,
+                Some("0 2px 4px rgba(0,0,0,.2)".to_string()),
+            ],
+        );
+        theme.add_shadow(
+            "default",
+            "md",
+            vec![Some("0 4px 8px rgba(0,0,0,.1)".to_string())],
+        );
+
+        let levels = theme.get_shadow_token_levels();
+        assert_eq!(levels.get("sm").unwrap(), &vec![0u8, 2]);
+        assert_eq!(levels.get("md").unwrap(), &vec![0u8]);
+    }
+
+    #[test]
+    fn test_get_default_shadow_value() {
+        let mut theme = Theme::default();
+        theme.add_shadow(
+            "default",
+            "card",
+            vec![Some("0 1px 2px #0003".to_string()), None],
+        );
+
+        assert_eq!(
+            theme.get_default_shadow_value("card"),
+            Some("0 1px 2px #0003")
+        );
+        assert_eq!(theme.get_default_shadow_value("nonexistent"), None);
+
+        // No shadows at all
+        let empty = Theme::default();
+        assert_eq!(empty.get_default_shadow_value("card"), None);
+    }
 }