[tor-commits] [obfsproxy/master] Add scramblesuit unittests for the state module.

[asn at torproject.org]

commit 5333b1b650667409a7dc52cb646b2a822e3a2c80
Author: George Kadianakis <desnacked at riseup.net>
Date:   Mon Apr 14 15:01:01 2014 +0300

    Add scramblesuit unittests for the state module.
---
 obfsproxy/test/transports/test_scramblesuit.py |   47 ++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/obfsproxy/test/transports/test_scramblesuit.py b/obfsproxy/test/transports/test_scramblesuit.py
index 541d92e..df7f69b 100644
--- a/obfsproxy/test/transports/test_scramblesuit.py
+++ b/obfsproxy/test/transports/test_scramblesuit.py
@@ -19,6 +19,8 @@ import obfsproxy.transports.scramblesuit.mycrypto as mycrypto
 import obfsproxy.transports.scramblesuit.uniformdh as uniformdh
 import obfsproxy.transports.scramblesuit.scramblesuit as scramblesuit
 import obfsproxy.transports.scramblesuit.message as message
+import obfsproxy.transports.scramblesuit.state as state
+
 
 # Disable all logging as it would yield plenty of warning and error
 # messages.
@@ -226,6 +228,51 @@ class UtilTest( unittest.TestCase ):
         # Read file where we (hopefully) don't have permissions.
         self.failUnless(util.readFromFile("/etc/shadow") == None)
 
+class StateTest( unittest.TestCase ):
+
+    def setUp( self ):
+        const.STATE_LOCATION = "/tmp/"
+        self.stateFile = const.STATE_LOCATION + const.SERVER_STATE_FILE
+        self.state = state.State()
+
+    def tearDown( self ):
+        try:
+            os.unlink(self.stateFile)
+        except OSError:
+            pass
+
+    def test1_genState( self ):
+        self.state.genState()
+        self.failUnless(os.path.exists(self.stateFile))
+
+    def test2_loadState( self ):
+        # load() should create the state file if it doesn't exist yet.
+        self.failIf(os.path.exists(self.stateFile))
+        self.failUnless(isinstance(state.load(), state.State))
+        self.failUnless(os.path.exists(self.stateFile))
+
+    def test3_replay( self ):
+        key = "A" * const.HMAC_SHA256_128_LENGTH
+        self.state.genState()
+        self.state.registerKey(key)
+        self.failUnless(self.state.isReplayed(key))
+        self.failIf(self.state.isReplayed("B" * const.HMAC_SHA256_128_LENGTH))
+
+    def test4_ioerrorFail( self ):
+        def fake_open(name, mode):
+            raise IOError()
+        self.state.genState()
+
+        import __builtin__
+        real_open = __builtin__.open
+        __builtin__.open = fake_open
+
+        # Make state.load() fail
+        self.assertRaises(SystemExit, state.load)
+        # Make State.writeState() fail.
+        self.assertRaises(SystemExit, self.state.genState)
+
+        __builtin__.open = real_open
 
 class MockArgs( object ):
     uniformDHSecret = sharedSecret = ext_cookie_file = dest = None