feat: add "__argos__" class on root tag

argos-ci · Dec 13, 2023 · 3949b94 · 3949b94
1 parent f660148
commit 3949b94
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/packages/browser/src/stabilization.ts b/packages/browser/src/stabilization.ts
@@ -150,6 +150,14 @@ export function restoreElementPositions(document: Document) {
  });
 }
 
+export function addGlobalClass(document: Document, className: string) {
+ document.documentElement.classList.add(className);
+}
+
+export function removeGlobalClass(document: Document, className: string) {
+ document.documentElement.classList.remove(className);
+}
+
 export type SetupOptions = { fullPage?: boolean; argosCSS?: string };
 
 /**
@@ -159,6 +167,7 @@ export function setup(
  document: Document,
  { fullPage, argosCSS }: SetupOptions = {},
 ) {
+ addGlobalClass(document, "__argos__");
  injectGlobalStyles(document, GLOBAL_CSS, "argos-reset-style");
  if (argosCSS) {
  injectGlobalStyles(document, argosCSS, "argos-user-style");
@@ -178,6 +187,7 @@ export function teardown(
  document: Document,
  { fullPage, argosCSS }: SetupOptions = {},
 ) {
+ removeGlobalClass(document, "__argos__");
  removeGlobalStyles(document, "argos-reset-style");
  if (argosCSS) {
  removeGlobalStyles(document, "argos-user-style");

diff --git a/packages/playwright/fixtures/dummy.html b/packages/playwright/fixtures/dummy.html
@@ -44,6 +44,9 @@
  border: 1px solid black;
  overflow: scroll;
  }
+ .__argos__ h1 {
+ font-variant: small-caps;
+ }
  </style>
  </head>
  <body>