From b98c4876434ec76a43cab471968cd3687da695ca Mon Sep 17 00:00:00 2001
From: geegaset <geegaset@gmail.com>
Date: Tue, 23 Aug 2016 11:48:13 +0100
Subject: [PATCH] TabInputButton accessibilityId introduced (#206)

* TabInputButton accessibilityId introduced

* TabInputButton accessibilityId introduced
---
 .../Source/Input/Photos/PhotosChatInputItem.swift            | 2 +-
 ChattoAdditions/Source/Input/TabInputButton.swift            | 5 ++++-
 ChattoAdditions/Source/Input/Text/TextChatInputItem.swift    | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/ChattoAdditions/Source/Input/Photos/PhotosChatInputItem.swift b/ChattoAdditions/Source/Input/Photos/PhotosChatInputItem.swift
index 59ac57a..daea340 100644
--- a/ChattoAdditions/Source/Input/Photos/PhotosChatInputItem.swift
+++ b/ChattoAdditions/Source/Input/Photos/PhotosChatInputItem.swift
@@ -56,7 +56,7 @@ public class PhotosChatInputItem: ChatInputItemProtocol {
     }
 
     lazy private var internalTabView: UIButton = {
-        return TabInputButton.makeInputButton(withAppearance: self.buttonAppearance)
+        return TabInputButton.makeInputButton(withAppearance: self.buttonAppearance, accessibilityID: "photos.chat.input.view")
     }()
 
     lazy var photosInputView: PhotosInputViewProtocol = {
diff --git a/ChattoAdditions/Source/Input/TabInputButton.swift b/ChattoAdditions/Source/Input/TabInputButton.swift
index aa19345..8ebb61c 100644
--- a/ChattoAdditions/Source/Input/TabInputButton.swift
+++ b/ChattoAdditions/Source/Input/TabInputButton.swift
@@ -36,13 +36,16 @@ public struct TabInputButtonAppearance {
 
 public class TabInputButton: UIButton {
 
-    static public func makeInputButton(withAppearance appearance: TabInputButtonAppearance) -> TabInputButton {
+    static public func makeInputButton(withAppearance appearance: TabInputButtonAppearance, accessibilityID: String? = nil) -> TabInputButton {
         let images = appearance.images
         let button = TabInputButton(type: .Custom)
         button.exclusiveTouch = true
         images.forEach { (state, image) in
             button.setImage(image, forState: state.controlState)
         }
+        if let accessibilityIdentifier = accessibilityID {
+            button.accessibilityIdentifier = accessibilityIdentifier
+        }
         button.size = appearance.size
         return button
     }
diff --git a/ChattoAdditions/Source/Input/Text/TextChatInputItem.swift b/ChattoAdditions/Source/Input/Text/TextChatInputItem.swift
index 55f896a..9e6d1db 100644
--- a/ChattoAdditions/Source/Input/Text/TextChatInputItem.swift
+++ b/ChattoAdditions/Source/Input/Text/TextChatInputItem.swift
@@ -43,7 +43,7 @@ public class TextChatInputItem {
     }
 
     lazy private var internalTabView: TabInputButton = {
-        return TabInputButton.makeInputButton(withAppearance: self.buttonAppearance)
+        return TabInputButton.makeInputButton(withAppearance: self.buttonAppearance, accessibilityID: "text.chat.input.view")
     }()
 
     public var selected = false {