Attach/detach console iframe on addon enabled/disabled

1 files changed, 5 insertions, 1 deletions

diff --git a/test/content/operators/impls/ToggleAddonOperator.test.ts b/test/content/operators/impls/ToggleAddonOperator.test.ts
index 6026eb1..1f291b5 100644
--- a/test/content/operators/impls/ToggleAddonOperator.test.ts
+++ b/test/content/operators/impls/ToggleAddonOperator.test.ts
@@ -2,23 +2,27 @@ import { expect } from "chai";
 import ToggleAddonOperator from "../../../../src/content/operators/impls/ToggleAddonOperator";
 import MockAddonIndicatorClient from "../../mock/MockAddonIndicatorClient";
 import MockAddonEnabledRepository from "../../mock/MockAddonEnabledRepository";
+import MockConsoleFramePresenter from "./MockConsoleFramePresenter";
 
 describe("ToggleAddonOperator", () => {
   describe("#run", () => {
     it("toggles addon-enabled state", async () => {
       const client = new MockAddonIndicatorClient(true);
       const repository = new MockAddonEnabledRepository(true);
-      const sut = new ToggleAddonOperator(client, repository);
+      const presenter = new MockConsoleFramePresenter(true);
+      const sut = new ToggleAddonOperator(client, repository, presenter);
 
       await sut.run();
 
       expect(client.enabled).to.be.false;
       expect(repository.enabled).to.be.false;
+      expect(presenter.attached).to.be.false;
 
       await sut.run();
 
       expect(client.enabled).to.be.true;
       expect(repository.enabled).to.be.true;
+      expect(presenter.attached).to.be.true;
     });
   });
 });