add convenience includesContent audits

embroider-build · Sep 28, 2024 · f14f3b8 · f14f3b8
1 parent ff427a1
commit f14f3b8
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/test-packages/support/audit-assertions.ts b/test-packages/support/audit-assertions.ts
@@ -180,6 +180,42 @@ export class ExpectModule {
     });
   }
 
+  doesNotIncludeContent(src: string, message?: string) {
+    if (!this.module) {
+      this.emitMissingModule();
+      return;
+    }
+    if (this.module.type === 'unparseable') {
+      this.emitUnparsableModule(message);
+      return;
+    }
+
+    this.expectAudit.assert.pushResult({
+      result: !this.module.content.includes(src),
+      actual: this.module.content,
+      expected: true,
+      message: message ?? `Expected ${this.inputName} to not contain ${src}`,
+    });
+  }
+
+  includesContent(src: string, message?: string) {
+    if (!this.module) {
+      this.emitMissingModule();
+      return;
+    }
+    if (this.module.type === 'unparseable') {
+      this.emitUnparsableModule(message);
+      return;
+    }
+
+    this.expectAudit.assert.pushResult({
+      result: this.module.content.includes(src),
+      actual: this.module.content,
+      expected: true,
+      message: message ?? `Expected ${this.inputName} to contain ${src}`,
+    });
+  }
+
   private emitUnparsableModule(message?: string) {
     this.expectAudit.assert.pushResult({
       result: false,
@@ -327,6 +363,8 @@ class EmptyExpectModule implements PublicAPI<ExpectModule> {
   codeEquals() {}
   codeContains() {}
   withContents() {}
+  doesNotIncludeContent() {}
+  includesContent() {}
 
   resolves(): PublicAPI<ExpectResolution> {
     return new EmptyExpectResolution() as PublicAPI<ExpectResolution>;