Question

我的解决方案基于：

我有一个可以实例化的类，它可以写入文件。我正在尝试测试它，但我在模仿open()时遇到了问题。我使用以下代码作为最小的代码，可以

import os
import unittest
from unittest.mock import mock_open, patch

__author__ = 'drews'


class MockPathExists(object):
    def __init__(self, return_value):
        self.received_args = None
        self.return_value = return_value

    def __call__(self, *args, **kwargs):
        self.received_args = args
        return self.return_value


class WriteData:
    def __init__(self, dir, name='World'):
        self.name = name
        self.dir = dir

    def dump(self):
        if os.path.exists(self.dir):
            with open('{0}/output.text'.format(self.dir), 'w+') as fp:
                fp.write('Hello, {0}!'.format(self.name))


class TestListWindowsPasswords(unittest.TestCase):
    def setUp(self):
        self._orig_pathexists = os.path.exists
        os.path.exists = MockPathExists(True)

    def test_dump(self):
        m = mock_open()
        with patch.object(WriteData, 'open', m, create=True):
            data_writer = WriteData(
                dir='/my/path/not/exists',
                name='Foo'
            )
            data_writer.dump()

        self.assertEqual(os.path.exists.received_args[0], '/my/path/not/exists/output.text')
        m.assert_called_once_with('/my/path/not/exists/output.text', 'w+')
        handle = m()
        handle.write.assert_called_once_with('Hello, Foo!')



    def tearDown(self):
        os.path.exists = self._orig_pathexists

当我运行它时，我收到以下错误：

Error
Traceback (most recent call last):
  File "/Users/drews/Development/tool/tests/test_mockopen.py", line 41, in test_dump
    data_writer.dump()
  File "/Users/drews/Development/tool/tests/test_mockopen.py", line 25, in dump
    with open('{0}/output.text'.format(self.dir), 'w+') as fp:
FileNotFoundError: [Errno 2] No such file or directory: '/my/path/not/exists/output.text'

我如何模拟open（），以便它只返回一个file_pointer，并且根本不尝试与文件系统交互？

Answer 1

使用mock_open模拟module.open（或module，WriteData =包含import builtins class TestListWindowsPasswords(unittest.TestCase): def setUp(self): self._orig_pathexists = os.path.exists os.path.exists = MockPathExists(True) def test_dump(self): m = mock_open() with patch('builtins.open', unittest.mock.mock_open()) as m: data_writer = WriteData( dir='/my/path/not/exists', name='Foo' ) data_writer.dump() self.assertEqual(os.path.exists.received_args[0], '/my/path/not/exists') # fixed m.assert_called_once_with('/my/path/not/exists/output.text', 'w+') handle = m() handle.write.assert_called_once_with('Hello, Foo!')的模块名称）：

<?php


    error_reporting(0);
    $con = mysql_connect('localhost','root','') or die('could not connect to database');

    mysql_select_db('addscloud');

    @session_start();

    if(isset($_POST['submit']))
    {
    $query = "SELECT * FROM  tbl_reg where reg_mobile = '".$_POST['reg_mobile']."' and reg_pass = '".$_POST['reg_pass']."' ";

    var_dump($query);

    $result=mysql_query($query);
    $row=mysql_fetch_array($result);


    if($row)
    {
    $_SESSION['var']=2; 
    header('Location:registration_user.php');
    }

    else
    {
    $_SESSION['msg']="invalid user Name or Password";
    header('Location:login_form.php');
    }

    }


    ?>

 <?php
  if(isset($_SESSION['msg']))
 {
 echo $_SESSION['msg'];
  unset($_SESSION['msg']);
  }
  ?>

Answer 2

您可以使用__enter__ magic方法来模拟哪个：

from unittest.mock import patch, MagicMock, call, mock_open

@patch('os')
@patch('builtins.open', new_callable=mock_open())
def test_dump(self, mock_open_file, mock_os):
    data_writer = WriteData(dir='/my/path/not/exists', name='Foo')

    mock_os.path.exists.assert_called_once_with('/my/path/not/exists')
    mock_open_file.assert_called_once_with('/my/path/not/exists/output.text', 'w+')
    mock_open_file.return_value.__enter__().write.assert_called_once_with('Hello, Foo!')

希望这会有所帮助！

如何模拟open（...）。write（）而不会出现“No such file or directory”错误？

2 个答案: