# HG changeset patch
# User Brian Neal <bgneal@gmail.com>
# Date 1338175258 18000
# Node ID d46635a9f088e777dd86b73b64b4ec7882d05b8a
# Parent  f4051bcac8c5197d1963cba967a549cc75bd4735
We can now decrypt an actual encoded message from WWII!

diff -r f4051bcac8c5 -r d46635a9f088 enigma/tests/test_enigma.py
--- a/enigma/tests/test_enigma.py	Sun May 27 21:53:47 2012 -0500
+++ b/enigma/tests/test_enigma.py	Sun May 27 22:20:58 2012 -0500
@@ -50,7 +50,7 @@
         self.assertEqual(plain_text, self.PLAIN_TEXT)
 
 
-class ActualCipherTestCase(unittest.TestCase):
+class ActualDecryptTestCase(unittest.TestCase):
     """This example taken from Dirk Rijmenants' simulator manual."""
 
     def setUp(self):
@@ -63,14 +63,25 @@
                 ring_settings=ring_settings,
                 plugboard_settings='AV BS CG DL FU HZ IN KM OW RX')
 
-    def test_decrypt(self):
+    def decrypt(self, start, enc_key, ciphertext, truth_data):
 
-        # retrieve the message key
-        self.machine.set_display('WXC')
-        msg_key = self.machine.process_text('KCH')
+        # remove spaces & Kenngruppen from the ciphertext
+        ciphertext = ciphertext.replace(' ', '')[5:]
 
-        # set the message key
+        # remove spaces from the truth data
+        truth_data = truth_data.replace(' ', '')
+
+        # decrypt the message key
+        self.machine.set_display(start)
+        msg_key = self.machine.process_text(enc_key)
+
+        # decrypt the cipher text with the unencrypted message key
         self.machine.set_display(msg_key)
+        plaintext = self.machine.process_text(ciphertext)
+
+        self.assertEqual(plaintext, truth_data)
+
+    def test_decrypt_part1(self):
 
         ciphertext = (
             'RFUGZ EDPUD NRGYS ZRCXN'
@@ -81,11 +92,43 @@
             'UBSTS LRNBZ SZWNR FXWFY'
             'SSXJZ VIJHI DISHP RKLKA'
             'YUPAD TXQSP INQMA TLPIF'
-            'SVKDA SCTAC DPBOP VHJK')
+            'SVKDA SCTAC DPBOP VHJK'
+        )
 
-        # remove spaces and the Kenngruppen
-        ciphertext = ciphertext.replace(' ', '')[5:]
-        plaintext = self.machine.process_text(ciphertext)
+        truth_data = (
+            'AUFKL XABTE ILUNG XVONX' 
+            'KURTI NOWAX KURTI NOWAX'
+            'NORDW ESTLX SEBEZ XSEBE'
+            'ZXUAF FLIEG ERSTR ASZER'
+            'IQTUN GXDUB ROWKI XDUBR'
+            'OWKIX OPOTS CHKAX OPOTS'
+            'CHKAX UMXEI NSAQT DREIN'
+            'ULLXU HRANG ETRET ENXAN'
+            'GRIFF XINFX RGTX'
+        )
 
-        print("'{}'\n\n'{}'\n\n'{}'\n\n".format(msg_key, ciphertext[:11],
-            plaintext[:11]))
+        self.decrypt('WXC', 'KCH', ciphertext, truth_data)
+
+    def test_decrypt_part2(self):
+
+        ciphertext = (
+            'FNJAU SFBWD NJUSE GQOBH'
+            'KRTAR EEZMW KPPRB XOHDR'
+            'OEQGB BGTQV PGVKB VVGBI'
+            'MHUSZ YDAJQ IROAX SSSNR'
+            'EHYGG RPISE ZBOVM QIEMM'
+            'ZCYSG QDGRE RVBIL EKXYQ'
+            'IRGIR QNRDN VRXCY YTNJR'
+        )
+
+        truth_data = (
+            'DREIG EHTLA NGSAM ABERS' 
+            'IQERV ORWAE RTSXE INSSI' 
+            'EBENN ULLSE QSXUH RXROE' 
+            'MXEIN SXINF RGTXD REIXA' 
+            'UFFLI EGERS TRASZ EMITA' 
+            'NFANG XEINS SEQSX KMXKM' 
+            'XOSTW XKAME NECXK'
+        )
+
+        self.decrypt('CRS', 'YPJ', ciphertext, truth_data)