eryn/enigma-machine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Clean up components

Browse source
This commit is contained in:
Eryn Wells 2015-07-18 22:59:58 -07:00
parent cb437b7d96
commit b7798d2dcd
2 changed files with 62 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

20
EnigmaTests/EnigmaTests.swift
View file

@ -15,33 +15,43 @@ class RotorTests: XCTestCase {
let rotorSeries = "EKMFLGDQVZNTOWYHXUSPAIBRCJ"
func testThatUnadvancedSubstitutionWorks() {
let rotor = try! Enigma.Rotor(series: rotorSeries)
let rotor = try! Rotor(series: rotorSeries)
for (plainCharacter, cipherCharacter) in zip(alphaSeries.characters, rotorSeries.characters) {
XCTAssertEqual(try! rotor.encode(plainCharacter), cipherCharacter)
}
}
func testThatRotorCanDoRot13() {
func testThatRotorCanAdvanceToRot13() {
let rot13Series = "NOPQRSTUVWXYZABCDEFGHIJKLM"
let rotor = try! Enigma.Rotor(series: alphaSeries)
let rotor = try! Rotor(series: alphaSeries)
rotor.advance(13)
for (plainCharacter, cipherCharacter) in zip(alphaSeries.characters, rot13Series.characters) {
XCTAssertEqual(try! rotor.encode(plainCharacter), cipherCharacter)
}
}
func testThatRotorCanSetToRot13() {
let rot13Series = "NOPQRSTUVWXYZABCDEFGHIJKLM"
let rotor = try! Rotor(series: alphaSeries)
rotor.position = 13
for (plainCharacter, cipherCharacter) in zip(alphaSeries.characters, rot13Series.characters) {
XCTAssertEqual(try! rotor.encode(plainCharacter), cipherCharacter)
}
}
}
class PlugboardTests: XCTestCase {
func testThatEmptyPlugboardPassesThroughAllCharacters() {
let plugboard = try! Enigma.Plugboard()
let plugboard = Plugboard()
for c in alphaSeries.characters {
XCTAssertEqual(try! plugboard.encode(c), c)
}
}
func testThatPlugboardPairsAreBidirectional() {
let plugboard = try! Enigma.Plugboard(pairs: [("A", "H")])
let plugboard = Plugboard()
plugboard.addPlug("A", b: "H")
XCTAssertEqual(try! plugboard.encode("A"), "H")
XCTAssertEqual(try! plugboard.encode("H"), "A")
}