From f13d70aa75f15cc29ab398fadbb86d1eddbb8807 Mon Sep 17 00:00:00 2001
From: Miguel Garnacho Velez <migarve55@gmail.com>
Date: Mon, 22 Apr 2019 18:00:21 +0200
Subject: [PATCH] Changed tests

---
 src/app/message/message.component.spec.ts | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/app/message/message.component.spec.ts b/src/app/message/message.component.spec.ts
index 56dbe18..df9b278 100644
--- a/src/app/message/message.component.spec.ts
+++ b/src/app/message/message.component.spec.ts
@@ -5,6 +5,7 @@ import { ChatService } from '../services/chat.service';
 import { ToastrModule } from 'ngx-toastr';
 import {By} from '@angular/platform-browser';
 import { BrowserAnimationsModule } from '@angular/platform-browser/animations';
+import { ChatMessage } from '../models/chat-message.model';
 
 describe('MessageComponent', () => {
   let component: MessageComponent;
@@ -22,6 +23,8 @@ describe('MessageComponent', () => {
   beforeEach(() => {
     fixture = TestBed.createComponent(MessageComponent);
     component = fixture.componentInstance;
+    component.chatMessage = new ChatMessage("Test 1", "Test msg", new Date());
+    component.chatMessage.webId = "";
     fixture.detectChanges();
   });
 
@@ -29,17 +32,15 @@ describe('MessageComponent', () => {
     expect(component).toBeTruthy();
   });
 
-  it('should remove message (case a)', () => {
-    component.messageContent = '';
+  it('should not remove a message that is not yours', () => {
+    component.isOwnMessage = false;
     component.removeMessage();
     expect(By.css('#toastr-container'));
   });
 
-  it('should remove message (case b)', () => {
-    component.messageContent = 'a';
+  it('should remove message', () => {
     component.isOwnMessage = true;
     component.removeMessage();
-    expect(By.css('#toastr-container'));
-    expect(component.messageContent === '');
+    expect(!By.css('#toastr-container'));
   });
 });