From 31774fbc078de4b53305cc355fbba16678d0029d Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Sat, 18 Jul 2015 10:39:53 -0700
Subject: [PATCH] Basic unit test for unadvanced Rotor

---
 EnigmaTests/EnigmaTests.swift | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/EnigmaTests/EnigmaTests.swift b/EnigmaTests/EnigmaTests.swift
index 0a39ecf..147b7ed 100644
--- a/EnigmaTests/EnigmaTests.swift
+++ b/EnigmaTests/EnigmaTests.swift
@@ -7,6 +7,8 @@
 //
 
 import XCTest
+@testable import Enigma
+
 
 class EnigmaTests: XCTestCase {
     
@@ -33,3 +35,23 @@ class EnigmaTests: XCTestCase {
     }
     
 }
+
+
+class RotorTests: XCTestCase {
+    let rotorSeries = "EKMFLGDQVZNTOWYHXUSPAIBRCJ"
+    var rotor: Enigma.Rotor!
+
+    override func setUp() {
+        rotor = try! Enigma.Rotor(series: rotorSeries)
+    }
+
+    override func tearDown() {
+        rotor = nil
+    }
+
+    func testThatUnadvancedSubstitutionWorks() {
+        for (plainCharacter, cipherCharacter) in zip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".characters, rotorSeries.characters) {
+            XCTAssertEqual(try! rotor.encode(plainCharacter), cipherCharacter)
+        }
+    }
+}
\ No newline at end of file