Basic unit test for unadvanced Rotor

This commit is contained in:
Eryn Wells 2015-07-18 10:39:53 -07:00
parent 56ae8cc5bb
commit 31774fbc07

View file

@ -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)
}
}
}