From 2d21de00fb94ec76e69ba28925f09acf5a8745f0 Mon Sep 17 00:00:00 2001
From: Abhishek <abhishekbanthia@me.com>
Date: Fri, 2 Jul 2021 15:19:56 -0500
Subject: [PATCH] Add tests for copy functionality.

---
 .../ClockerUITests/CopyToClipboardTests.swift | 53 +++++++++++++++++++
 Clocker/Panel/UI/TimezoneCellView.swift       |  1 -
 2 files changed, 53 insertions(+), 1 deletion(-)
 create mode 100644 Clocker/ClockerUITests/CopyToClipboardTests.swift

diff --git a/Clocker/ClockerUITests/CopyToClipboardTests.swift b/Clocker/ClockerUITests/CopyToClipboardTests.swift
new file mode 100644
index 0000000..96b5323
--- /dev/null
+++ b/Clocker/ClockerUITests/CopyToClipboardTests.swift
@@ -0,0 +1,53 @@
+// Copyright © 2015 Abhishek Banthia
+
+import XCTest
+
+class CopyToClipboardTests: XCTestCase {
+    var app: XCUIApplication!
+    
+    override func setUp() {
+        continueAfterFailure = false
+        app = XCUIApplication()
+        app.launch()
+
+        if app.tables["FloatingTableView"].exists == false {
+            app.tapMenubarIcon()
+            app.buttons["FloatingPin"].click()
+        }
+    }
+
+    override func tearDownWithError() throws {
+        app = nil
+    }
+
+    func testFullCopy() throws {
+        let cellCount = app.tables["FloatingTableView"].cells.count
+        var clipboardValue = String()
+        for cellIndex in 0..<cellCount {
+            let cell = app.tables["FloatingTableView"].cells.element(boundBy: cellIndex)
+            let customLabel = cell.staticTexts["CustomNameLabelForCell"].value ?? "Nil Custom Label"
+            let time = cell.staticTexts["ActualTime"].value ?? "Nil Value"
+            clipboardValue.append(contentsOf: "\(customLabel) - \(time)\n")
+        }
+        
+        app.buttons["Share"].click()
+        app/*@START_MENU_TOKEN@*/.menuItems["Copy All Times"]/*[[".dialogs[\"Clocker Panel\"]",".buttons[\"Share\"]",".menus.menuItems[\"Copy All Times\"]",".menuItems[\"Copy All Times\"]"],[[[-1,3],[-1,2],[-1,1,2],[-1,0,1]],[[-1,3],[-1,2],[-1,1,2]],[[-1,3],[-1,2]]],[0]]@END_MENU_TOKEN@*/.click()
+
+        let clipboard = NSPasteboard.general.string(forType: .string)
+        XCTAssert(clipboardValue == clipboard)
+    }
+    
+    func testIndividualTimezoneCopy() {
+        let cell = app.tables["FloatingTableView"].cells.firstMatch
+        let customLabel = cell.staticTexts["CustomNameLabelForCell"].value ?? "Nil Custom Label"
+        let time = cell.staticTexts["ActualTime"].value ?? "Nil Value"
+        let expectedValue = "\(customLabel) - \(time)"
+
+        // Tap to copy!
+        cell.tap()
+        
+        let clipboard = NSPasteboard.general.string(forType: .string) ?? "Empty Pasteboard"
+        XCTAssert(expectedValue == clipboard, "Clipboard value (\(clipboard)) doesn't match expected result")
+    }
+
+}
diff --git a/Clocker/Panel/UI/TimezoneCellView.swift b/Clocker/Panel/UI/TimezoneCellView.swift
index 2eaf28c..f315bb8 100644
--- a/Clocker/Panel/UI/TimezoneCellView.swift
+++ b/Clocker/Panel/UI/TimezoneCellView.swift
@@ -199,7 +199,6 @@ class TimezoneCellView: NSTableCellView {
 
     override func mouseDown(with _: NSEvent) {
         // Text is copied in the following format: Chicago - 1625185925
-        // TODO: Write tests for copy functionality
         let clipboardCopy = "\(customName.stringValue) - \(time.stringValue)"
         let pasteboard = NSPasteboard.general
         pasteboard.declareTypes([.string], owner: nil)