Fixed use of deprecated to satisfy function

unexpectedjs · Dec 26, 2018 · c47022e · c47022e
1 parent 93789a0
commit c47022e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -707,15 +707,15 @@ module.exports = {
       '<DOMElement> to only have (class|classes) <array|string>',
       (expect, subject, value) =>
         expect(subject, 'to have attributes', {
-          class: className => {
+          class: expect.it(className => {
             const actualClasses = getClassNamesFromAttributeValue(className);
             if (typeof value === 'string') {
               value = getClassNamesFromAttributeValue(value);
             }
             return bubbleError(() =>
               expect(actualClasses.sort(), 'to equal', value.sort())
             );
-          }
+          })
         })
     );
 
@@ -1104,6 +1104,13 @@ module.exports = {
                     expect(attrs.style, 'to satisfy', expectedStyleObj)
                   );
                 }
+              } else if (
+                expect.findTypeOf(expectedAttributeValue).is('expect.it')
+              ) {
+                expect.context.thisObject = subject;
+                return bubbleError(() =>
+                  expectedAttributeValue(attributeValue, expect.context)
+                );
               } else {
                 return bubbleError(() =>
                   expect(attributeValue, 'to satisfy', expectedAttributeValue)