From 9add33a56ee189d4b2c5da00c8e43ceaee7e917c Mon Sep 17 00:00:00 2001
From: Torsten Dittmann <torsten.dittmann@googlemail.com>
Date: Thu, 13 Oct 2022 14:45:51 +0200
Subject: [PATCH] fix: tests for code snippet

---
 src/lib/components/code.svelte     | 4 +++-
 tests/unit/components/code.test.ts | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/lib/components/code.svelte b/src/lib/components/code.svelte
index 24d6c99e9..79af698ce 100644
--- a/src/lib/components/code.svelte
+++ b/src/lib/components/code.svelte
@@ -25,7 +25,9 @@
 </script>
 
 <section class="box u-overflow-hidden u-min-height-100">
-    <div class="u-position-absolute u-inset-inline-end-8 u-inset-block-start-8 u-flex u-gap-8">
+    <div
+        data-testid="controls"
+        class="u-position-absolute u-inset-inline-end-8 u-inset-block-start-8 u-flex u-gap-8">
         {#if label}
             <Pill>{label}</Pill>
         {/if}
diff --git a/tests/unit/components/code.test.ts b/tests/unit/components/code.test.ts
index a23ce8f49..7edda4f5e 100644
--- a/tests/unit/components/code.test.ts
+++ b/tests/unit/components/code.test.ts
@@ -4,12 +4,12 @@ import { vi } from 'vitest';
 import { Code } from '../../../src/lib/components';
 
 test('default props', async () => {
-    const { container } = render(Code, {
+    const { getByTestId } = render(Code, {
         code: 'console.log("test");',
         language: 'js'
     });
 
-    expect(container.querySelector('.controls').children.length).toEqual(0);
+    expect(getByTestId('controls').children.length).toEqual(0);
 });
 
 test('shows label', async () => {