From 31dd466c6b1538215ddce659f26dfe50a60706fc Mon Sep 17 00:00:00 2001
From: Jannis Mattheis <contact@jmattheis.de>
Date: Fri, 26 Oct 2018 21:32:11 +0200
Subject: [PATCH] Fix messages test

---
 ui/src/message/Message.tsx   | 4 ++--
 ui/src/tests/message.test.ts | 6 +++++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/ui/src/message/Message.tsx b/ui/src/message/Message.tsx
index 55cf976..b5c426c 100644
--- a/ui/src/message/Message.tsx
+++ b/ui/src/message/Message.tsx
@@ -80,8 +80,8 @@ class Message extends React.PureComponent<IProps & Style> {
                             <Typography variant="body1" className="date">
                                 <TimeAgo date={date} />
                             </Typography>
-                            <IconButton onClick={fDelete} className={classes.trash}>
-                                <Delete className="delete" />
+                            <IconButton onClick={fDelete} className={`${classes.trash} delete`}>
+                                <Delete />
                             </IconButton>
                         </div>
                         <Typography component="p" className="content">
diff --git a/ui/src/tests/message.test.ts b/ui/src/tests/message.test.ts
index 190b431..7e76c1b 100644
--- a/ui/src/tests/message.test.ts
+++ b/ui/src/tests/message.test.ts
@@ -201,7 +201,11 @@ describe('Messages', () => {
     });
     it('deletes a windows message', async () => {
         await navigate('Windows');
-        await page.click('#messages span:nth-of-type(2) .message .delete');
+        await page.evaluate(() =>
+            (document.querySelectorAll(
+                '#messages .message .delete'
+            )[1] as HTMLButtonElement).click()
+        );
         await expectMessages({
             all: [linux2, windows3, backup1, linux1, windows1],
             windows: [windows3, windows1],