fix(resolveClassNames): Handle theme dereferencing in object keys (#3)

packages/treat/src/runtime.ts

@@ -57,7 +57,7 @@ export const resolveClassNames = (
           hasOwn.call(arg, key) &&
           (arg as { [classRef: string]: boolean })[key]
         ) {
-          classes.push(key);
+          classes.push(getClassName(themeRef, key));
         }
       }
     }

packages/treat/tests/runtime.test.js

@@ -0,0 +1,35 @@
+const { resolveClassNames } = require('treat');
+
+const themeRef = '_theme';
+
+describe('resolveClassNames', () => {
+  it('should handle dereference themed classes', () => {
+    expect(resolveClassNames(themeRef, '$someClass')).toBe('someClass_theme');
+  });
+
+  it('should handle regular classes', () => {
+    expect(resolveClassNames(themeRef, 'someClass')).toBe('someClass');
+  });
+
+  it('should handle arrays', () => {
+    const value = ['$someClass', 'someOtherClass'];
+
+    expect(resolveClassNames(themeRef, value)).toBe(
+      'someClass_theme someOtherClass',
+    );
+  });
+
+  it('should handle objects', () => {
+    const value = { ['$someClass']: true, someOtherClass: true };
+
+    expect(resolveClassNames(themeRef, value)).toBe(
+      'someClass_theme someOtherClass',
+    );
+  });
+
+  it('should handle objects with falsy values', () => {
+    const value = { $someClass: false, someOtherClass: true };
+
+    expect(resolveClassNames(themeRef, value)).toBe('someOtherClass');
+  });
+});