Python wait_for_successful_query示例

示例#1

文件： test_http.py 项目： mamh-mixed/saltstack-salt

def test_wait_for_without_interval():
    """
    Test for wait_for_successful_query waits for request_interval
    """

    query_mock = MagicMock(side_effect=[{"error": "error"}, {"result": True}])

    with patch.object(http, "query", query_mock):
        with patch("time.sleep", MagicMock()) as sleep_mock:
            assert http.wait_for_successful_query("url", status=200) == {"result": True}
            sleep_mock.assert_not_called()

示例#2

    def test_wait_for_with_interval(self):
        """
        Test for wait_for_successful_query waits for request_interval
        """

        query_mock = MagicMock(side_effect=[{"error": "error"}, {"result": True}])

        with patch.object(http, "query", query_mock):
            with patch("time.sleep", MagicMock()) as sleep_mock:
                self.assertEqual(
                    http.wait_for_successful_query(
                        "url", request_interval=1, status=200
                    ),
                    {"result": True},
                )
                sleep_mock.assert_called_once_with(1)

示例#3

    def test_wait_for_without_interval(self):
        '''
        Test for wait_for_successful_query waits for request_interval
        '''

        query_mock = MagicMock(side_effect=[{
            'error': 'error'
        }, {
            'result': True
        }])

        with patch.object(http, 'query', query_mock):
            with patch('time.sleep', MagicMock()) as sleep_mock:
                self.assertEqual(
                    http.wait_for_successful_query('url', status=200),
                    {'result': True})
                sleep_mock.assert_not_called()