Updated string test to catch both types of strings...

DamonOehlman · Oct 17, 2011 · ba519c0 · ba519c0
1 parent 8b3d5bc
commit ba519c0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/classtweak.js b/classtweak.js
@@ -66,7 +66,7 @@ function classtweak(elements, initAction) {
     } // tweak
 
     // check the elements
-    if (typeof elements == 'string') {
+    if (typeof elements == 'string' || elements instanceof String) {
         elements = document.querySelectorAll(elements);
     }
     else if (! Array.isArray(elements)) {

diff --git a/test/specs/queryselector.js b/test/specs/queryselector.js
@@ -42,4 +42,12 @@ describe('query selector tweaks', function() {
         expect(elements[1].className).toEqual('');
         expect(elements[2].className).toEqual('');
     });
+
+    it('can handle string objects', function() {
+        classtweak('section[data-route]', '+bounce');
+        classtweak(new String('section'), '!bounce');
+        expect(elements[0].className).toEqual('bounce');
+        expect(elements[1].className).toEqual('');
+        expect(elements[2].className).toEqual('');
+    });
 });