From c113c8e962f1da2b9e098f68c940fd1d4b03db20 Mon Sep 17 00:00:00 2001
From: Matthew Chen <matthew@signal.org>
Date: Fri, 14 Dec 2018 10:01:51 -0500
Subject: [PATCH] Add DeviceNamesTest.

---
 .../tests/Util/DeviceNamesTest.swift          | 66 +++++++++++++++++++
 .../tests/Util/JobQueueTest.swift             |  5 --
 2 files changed, 66 insertions(+), 5 deletions(-)
 create mode 100644 SignalServiceKit/tests/Util/DeviceNamesTest.swift

diff --git a/SignalServiceKit/tests/Util/DeviceNamesTest.swift b/SignalServiceKit/tests/Util/DeviceNamesTest.swift
new file mode 100644
index 000000000..2e8f8bf73
--- /dev/null
+++ b/SignalServiceKit/tests/Util/DeviceNamesTest.swift
@@ -0,0 +1,66 @@
+//
+//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
+//
+
+import Foundation
+import XCTest
+import Curve25519Kit
+
+@testable import SignalServiceKit
+
+class DeviceNamesTest: SSKBaseTestSwift {
+
+    override func setUp() {
+        super.setUp()
+    }
+
+    override func tearDown() {
+        super.tearDown()
+    }
+
+    // MARK: 
+
+    func testNotEncrypted() {
+
+        let identityKeyPair = Curve25519.generateKeyPair()
+
+        let plaintext = "alice"
+        guard let plaintextData = plaintext.data(using: .utf8) else {
+            XCTFail("Could not convert text to UTF-8.")
+            return
+        }
+
+        do {
+            _ = try DeviceNames.decryptDeviceName(input: plaintextData,
+                                                      identityKeyPair: identityKeyPair)
+            XCTFail("Unexpectedly did not throw error.")
+        } catch {
+            // Failure is expected.
+        }
+    }
+
+    func testSimple() {
+
+        let identityKeyPair = Curve25519.generateKeyPair()
+
+        let plaintext = "alice"
+        let encrypted: Data
+        do {
+            encrypted = try DeviceNames.encryptDeviceName(plaintext: plaintext,
+                                                      identityKeyPair: identityKeyPair)
+        } catch {
+            XCTFail("Failed with error: \(error)")
+            return
+        }
+
+        let decrypted: String
+        do {
+            decrypted = try DeviceNames.decryptDeviceName(input: encrypted,
+                                                      identityKeyPair: identityKeyPair)
+        } catch {
+            XCTFail("Failed with error: \(error)")
+            return
+        }
+        XCTAssertEqual(plaintext, decrypted)
+    }
+}
diff --git a/SignalServiceKit/tests/Util/JobQueueTest.swift b/SignalServiceKit/tests/Util/JobQueueTest.swift
index a76c1c484..8516f8c08 100644
--- a/SignalServiceKit/tests/Util/JobQueueTest.swift
+++ b/SignalServiceKit/tests/Util/JobQueueTest.swift
@@ -3,11 +3,6 @@
 //
 
 import Foundation
-
-//
-//  Copyright (c) 2018 Open Whisper Systems. All rights reserved.
-//
-
 import XCTest
 
 @testable import SignalServiceKit