te')); return $arr; } /* 遍历用户所有主题 * @param $uid 用户ID * @param int $page 页数 * @param int $pagesize 每页记录条数 * @param bool $desc 排序方式 TRUE降序 FALSE升序 * @param string $key 返回的数组用那一列的值作为 key * @param array $col 查询哪些列 */ function thread_tid_find_by_uid($uid, $page = 1, $pagesize = 1000, $desc = TRUE, $key = 'tid', $col = array()) { if (empty($uid)) return array(); $orderby = TRUE == $desc ? -1 : 1; $arr = thread_tid__find($cond = array('uid' => $uid), array('tid' => $orderby), $page, $pagesize, $key, $col); return $arr; } // 遍历栏目下tid 支持数组 $fid = array(1,2,3) function thread_tid_find_by_fid($fid, $page = 1, $pagesize = 1000, $desc = TRUE) { if (empty($fid)) return array(); $orderby = TRUE == $desc ? -1 : 1; $arr = thread_tid__find($cond = array('fid' => $fid), array('tid' => $orderby), $page, $pagesize, 'tid', array('tid', 'verify_date')); return $arr; } function thread_tid_delete($tid) { if (empty($tid)) return FALSE; $r = thread_tid__delete(array('tid' => $tid)); return $r; } function thread_tid_count() { $n = thread_tid__count(); return $n; } // 统计用户主题数 大数量下严谨使用非主键统计 function thread_uid_count($uid) { $n = thread_tid__count(array('uid' => $uid)); return $n; } // 统计栏目主题数 大数量下严谨使用非主键统计 function thread_fid_count($fid) { $n = thread_tid__count(array('fid' => $fid)); return $n; } ?>mocking - Overriding mock pytest - Stack Overflow
最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

mocking - Overriding mock pytest - Stack Overflow

programmeradmin3浏览0评论

I am trying to test different authorization scenarios, one that simulates success, and another that simulates failure. I am trying to do this without creating independent test files for success and failure.

def mock_failed_authorization(instance_id=None,

crn=None,

f_instance=None,

resource_group_id=None,

f_resource_group=None):

def decorator(f):

@wraps(f)

def g(*args, **kwargs):

  if crn:

    pass

  if instance_id:

    pass

  if f_instance:

    pass

  if resource_group_id:

    pass

  if f_resource_group:

    pass

  return 'Role not found', 403

return g

return decorator

def mock_successful_authorization(

instance_id=None,

crn=None,

f_instance=None,

resource_group_id=None,

f_resource_group=None):

def decorator(f):

@wraps(f)

def g(*args, **kwargs):

  if crn:

    pass

  if instance_id:

    pass

  if f_instance:

    pass

  if resource_group_id:

    pass

  if f_resource_group:

    pass

  return f(*args, **kwargs)

return g

return decorator

@pytest.fixture

def patched_authorization_failure(mocker):

Patch the authorization function to simulate failure behavior

mocker.patch(

'some_module.middleware.bar.func',

mocker.MagicMock(side_effect=mock_failed_authorization),

)

@pytest.fixture

def patched_authorization_success(mocker):

Patch the authorization function to simulate success behavior

mocker.patch(

'some_module.middleware.bar.func',

mocker.MagicMock(side_effect=mock_successfu`l_authorization),

)

@pytest.mark.usefixtures('patched_authorization_success')

@responses.activate

def func_1():

//logic

assert resp.status_int == 200

@pytest.mark.usefixtures("patched_authorization_failure")

@responses.activate

def func_2():

  //logic

  assert resp.status_int == 403`

Here, when I run this test file, func_1 asserts correctly(200==200), but patched_authorization_failure fails to override the patch created by patched_authorization success, and func_2 asserts 200==403.

I have tried resetting the mock after each test, and also used a temporary variable to store some_module.middleware.bar.func and attempt to reset it after the test, but nothing seems to resolve this issue.

Similarly, when I place func_2 above func_1, func_1 fails to assert(403==200). Is there a way to undo this patch? I have tried using unittest.mock.patch as well, but to no avail.

发布评论

评论列表(0)

  1. 暂无评论