From cd1fadf6c6878fbdf13a34a009cc35d53c43f91f Mon Sep 17 00:00:00 2001
From: Tapan Prakash <tapanprakasht@gmail.com>
Date: Mon, 9 Oct 2017 20:43:13 +0530
Subject: [PATCH] Adding test case for textColor when dataSource is nil

---
 Tests/MessagesDisplayDataSourceTests.swift | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Tests/MessagesDisplayDataSourceTests.swift b/Tests/MessagesDisplayDataSourceTests.swift
index 0dddd146..ef801e8c 100644
--- a/Tests/MessagesDisplayDataSourceTests.swift
+++ b/Tests/MessagesDisplayDataSourceTests.swift
@@ -56,6 +56,12 @@ class MessagesDisplayDelegateTests: XCTestCase {
         XCTAssertEqual(testClass.textColor(for: testClass.messageList[0], at: IndexPath(item: 0, section: 0), in: testClass.messagesCollectionView), UIColor.white)
         XCTAssertEqual(testClass.textColor(for: testClass.messageList[1], at: IndexPath(item: 1, section: 0), in: testClass.messagesCollectionView), UIColor.darkText)
     }
+  
+    func testMessageTextColorWhenDataSourceIsNil() {
+        testClass.messagesCollectionView.messagesDataSource = nil
+        XCTAssertEqual(testClass.textColor(for: testClass.messageList[0], at: IndexPath(item: 0, section: 0), in: testClass.messagesCollectionView), UIColor.darkText)
+        XCTAssertEqual(testClass.textColor(for: testClass.messageList[1], at: IndexPath(item: 1, section: 0), in: testClass.messagesCollectionView), UIColor.darkText)
+    }
     
     func testBackGroundColorDefaultState() {
         XCTAssertEqual(testClass.backgroundColor(for: testClass.messageList[0], at:  IndexPath(item: 0, section: 0), in: testClass.messagesCollectionView), UIColor.outgoingGreen)