def test_calculate_md5sum(self): open_mock = mock.MagicMock(return_value=FakeFile('fake file content')) file_path = '/tmp/file' with mock.patch('__builtin__.open', open_mock): self.assertEquals(calculate_md5sum(file_path), '199df6f47108545693b5c9cb5344bf13') open_mock.assert_called_once_with(file_path, 'rb')
def test_calculate_md5sum(self): open_mock = mock.MagicMock(return_value=FakeFile('fake file content')) file_path = '/tmp/file' with mock.patch('__builtin__.open', open_mock): self.assertEqual( calculate_md5sum(file_path), '199df6f47108545693b5c9cb5344bf13') open_mock.assert_called_once_with(file_path, 'rb')
def _check_checksum(self): """Calculate checksum and compare it with `self.checksum` :raises: errors.WrongChecksum """ calculated_checksum = calculate_md5sum(self.dst_path) if calculated_checksum != self.checksum: raise errors.WrongChecksum( u'File "{0}" has wrong checkum, actual ' 'checksum "{1}" expected checksum "{2}"'.format( self.dst_path, calculated_checksum, self.checksum))