[pypy-svn] r46965 - in pypy/dist/pypy/module/md5: . test

[arigo at codespeak.net]

Author: arigo
Date: Thu Sep 27 17:18:47 2007
New Revision: 46965

Added:
   pypy/dist/pypy/module/md5/   (props changed)
   pypy/dist/pypy/module/md5/__init__.py   (contents, props changed)
   pypy/dist/pypy/module/md5/interp_md5.py   (contents, props changed)
   pypy/dist/pypy/module/md5/test/   (props changed)
   pypy/dist/pypy/module/md5/test/test_md5.py   (contents, props changed)
Log:
An interp-level md5 module for PyPy.


Added: pypy/dist/pypy/module/md5/__init__.py
==============================================================================
--- (empty file)
+++ pypy/dist/pypy/module/md5/__init__.py	Thu Sep 27 17:18:47 2007
@@ -0,0 +1,29 @@
+
+"""
+Mixed-module definition for the md5 module.
+Note that there is also a pure Python implementation in pypy/lib/md5.py;
+the present mixed-module version of md5 takes precedence if it is enabled.
+"""
+
+from pypy.interpreter.mixedmodule import MixedModule
+
+
+class Module(MixedModule):
+    """\
+This module implements the interface to RSA's MD5 message digest
+algorithm (see also Internet RFC 1321). Its use is quite
+straightforward: use new() to create an md5 object. You can now feed
+this object with arbitrary strings using the update() method, and at any
+point you can ask it for the digest (a strong kind of 128-bit checksum,
+a.k.a. ``fingerprint'') of the concatenation of the strings fed to it so
+far using the digest() method."""
+
+    interpleveldefs = {
+        'md5': 'interp_md5.W_MD5',
+        'new': 'interp_md5.W_MD5',
+        'MD5Type': 'interp_md5.W_MD5',
+        'digest_size': 'space.wrap(16)',
+        }
+
+    appleveldefs = {
+        }

Added: pypy/dist/pypy/module/md5/interp_md5.py
==============================================================================
--- (empty file)
+++ pypy/dist/pypy/module/md5/interp_md5.py	Thu Sep 27 17:18:47 2007
@@ -0,0 +1,47 @@
+from pypy.rlib import rmd5
+from pypy.interpreter.baseobjspace import Wrappable
+from pypy.interpreter.typedef import TypeDef
+from pypy.interpreter.gateway import interp2app, ObjSpace, W_Root
+
+
+class W_MD5(Wrappable, rmd5.RMD5):
+    """
+    A subclass of RMD5 that can be exposed to app-level.
+    """
+
+    def __init__(self, space, initialdata=''):
+        self.space = space
+        self._init()
+        self.update(initialdata)
+
+    def digest_w(self):
+        return self.space.wrap(self.digest())
+
+    def hexdigest_w(self):
+        return self.space.wrap(self.hexdigest())
+
+    def copy_w(self):
+        return self.space.wrap(self.copy())
+
+
+def W_MD5___new__(space, w_subtype, initialdata=''):
+    """
+    Create a new md5 object and call its initializer.
+    """
+    w_md5 = space.allocate_instance(W_MD5, w_subtype)
+    md5 = space.interp_w(W_MD5, w_md5)
+    W_MD5.__init__(md5, space, initialdata)
+    return w_md5
+W_MD5___new__.unwrap_spec = [ObjSpace, W_Root, str]
+
+
+W_MD5.typedef = TypeDef(
+    'MD5Type',
+    __new__   = interp2app(W_MD5___new__),
+    update    = interp2app(W_MD5.update, unwrap_spec=['self', str]),
+    digest    = interp2app(W_MD5.digest_w),
+    hexdigest = interp2app(W_MD5.hexdigest_w),
+    copy      = interp2app(W_MD5.copy_w),
+    __doc__   = """md5(arg) -> return new md5 object.
+
+If arg is present, the method call update(arg) is made.""")

Added: pypy/dist/pypy/module/md5/test/test_md5.py
==============================================================================
--- (empty file)
+++ pypy/dist/pypy/module/md5/test/test_md5.py	Thu Sep 27 17:18:47 2007
@@ -0,0 +1,80 @@
+"""
+Tests for the md5 module implemented at interp-level in pypy/module/md5.
+"""
+
+import py
+from pypy.conftest import gettestobjspace
+
+
+class AppTestMD5(object):
+
+    def setup_class(cls):
+        """
+        Create a space with the md5 module and import it for use by the
+        tests.
+        """
+        cls.space = gettestobjspace(usemodules=['md5'])
+        cls.w_md5 = cls.space.appexec([], """():
+            import md5
+            return md5
+        """)
+
+
+    def test_digest_size(self):
+        """
+        md5.digest_size should be 16.
+        """
+        assert self.md5.digest_size == 16
+
+
+    def test_MD5Type(self):
+        """
+        Test the two ways to construct an md5 object.
+        """
+        md5 = self.md5
+        d = md5.md5()
+        assert isinstance(d, md5.MD5Type)
+        d = md5.new()
+        assert isinstance(d, md5.MD5Type)
+
+
+    def test_md5object(self):
+        """
+        Feed example strings into a md5 object and check the digest and
+        hexdigest.
+        """
+        md5 = self.md5
+        cases = (
+          ("",
+           "d41d8cd98f00b204e9800998ecf8427e"),
+          ("a",
+           "0cc175b9c0f1b6a831c399e269772661"),
+          ("abc",
+           "900150983cd24fb0d6963f7d28e17f72"),
+          ("message digest",
+           "f96b697d7cb7938d525a2f31aaf161d0"),
+          ("abcdefghijklmnopqrstuvwxyz",
+           "c3fcd3d76192e4007dfb496cca67e13b"),
+          ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789",
+           "d174ab98d277d9f5a5611c2c9f419d9f"),
+          ("1234567890"*8,
+           "57edf4a22be3c955ac49da2e2107b67a"),
+        )
+        for input, expected in cases:
+            d = md5.new(input)
+            assert d.hexdigest() == expected
+            assert d.digest() == expected.decode('hex')
+
+
+    def test_copy(self):
+        """
+        Test the copy() method.
+        """
+        md5 = self.md5
+        d1 = md5.md5()
+        d1.update("abcde")
+        d2 = d1.copy()
+        d2.update("fgh")
+        d1.update("jkl")
+        assert d1.hexdigest() == 'e570e7110ecef72fcb772a9c05d03373'
+        assert d2.hexdigest() == 'e8dc4081b13434b45189a720b77b6818'