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

Add rot13 test for Rotor

Browse source 
This test checks the modulo functionality.

This change also refactors how tests are set up. Each test is responsible for creating its own rotor.
This commit is contained in:
Eryn Wells 2015-07-18 10:47:14 -07:00
parent ccc1244e0c
commit f997e84639
1 changed files with 12 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

22
EnigmaTests/EnigmaTests.swift
View file

@ -10,19 +10,21 @@ import XCTest
@testable import Enigma
class RotorTests: XCTestCase {
let alphaSeries = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
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) {
let rotor = try! Enigma.Rotor(series: rotorSeries)
for (plainCharacter, cipherCharacter) in zip(alphaSeries.characters, rotorSeries.characters) {
XCTAssertEqual(try! rotor.encode(plainCharacter), cipherCharacter)
}
}
func testThatRotorCanDoRot13() {
let rot13Series = "NOPQRSTUVWXYZABCDEFGHIJKLM"
let rotor = try! Enigma.Rotor(series: alphaSeries)
rotor.advance(13)
for (plainCharacter, cipherCharacter) in zip(alphaSeries.characters, rot13Series.characters) {
XCTAssertEqual(try! rotor.encode(plainCharacter), cipherCharacter)
}
}