Add support for pure annotations in class nodes' children (#14)

* add support for pure annotations class expressions

* Update src/index.js

Co-authored-by: Mateusz Burzyński <mateuszburzynski@gmail.com>

* tweak tests

---------

Co-authored-by: Mateusz Burzyński <mateuszburzynski@gmail.com>

Author: fubhy

src/index.js

@@ -63,7 +63,8 @@ const isInAssignmentContext = path => {
 
     if (
       parentPath.isVariableDeclaration() ||
-      parentPath.isAssignmentExpression()
+      parentPath.isAssignmentExpression() ||
+      parentPath.isClass()
     ) {
       return true
     }

test/fixtures/annotate/class-declaration-dynamic-extends/code.js

@@ -0,0 +1,8 @@
+const base = bar =>
+  class {
+    foo() {
+      return bar
+    }
+  }
+
+class MyClass extends base() {}

test/fixtures/annotate/class-declaration-dynamic-extends/output.js

@@ -0,0 +1,6 @@
+const base = bar => class {
+  foo() {
+    return bar;
+  }
+};
+class MyClass extends /*#__PURE__*/base() {}

test/fixtures/annotate/class-expression-dynamic-extends/code.js

@@ -0,0 +1,8 @@
+const base = bar =>
+  class {
+    foo() {
+      return bar
+    }
+  }
+
+const MyClass = class extends base() {}

test/fixtures/annotate/class-expression-dynamic-extends/output.js

@@ -0,0 +1,6 @@
+const base = bar => class {
+  foo() {
+    return bar;
+  }
+};
+const MyClass = class extends /*#__PURE__*/base() {};