Python mock.create_autospec函数代码示例

def test_mongo_date_range_query():
    self = create_autospec(TickStore)
    self._collection = create_autospec(Collection)
    self._symbol_query.return_value = {"sy": {"$in" : ["s1" , "s2"]}}
    self._collection.aggregate.return_value = iter([{"_id": "s1", "start": dt(2014, 1, 1, 0, 0, tzinfo=mktz())},
                                                    {"_id": "s2", "start": dt(2014, 1, 1, 12, 0, tzinfo=mktz())}])

    self._collection.find_one.side_effect = [
        {'e': dt(2014, 1, 1, 15, 0, tzinfo=mktz())},
        {'e': dt(2014, 1, 2, 12, 0, tzinfo=mktz())}]

    query = TickStore._mongo_date_range_query(self, 'sym', DateRange(dt(2014, 1, 2, 0, 0, tzinfo=mktz()),
                                                                     dt(2014, 1, 3, 0, 0, tzinfo=mktz())))

    assert self._collection.aggregate.call_args_list == [call([
     {"$match": {"s": {"$lte": dt(2014, 1, 2, 0, 0, tzinfo=mktz())}, "sy": {"$in" : ["s1" , "s2"]}}},
     {"$project": {"_id": 0, "s": 1, "sy": 1}},
     {"$group": {"_id": "$sy", "start": {"$max": "$s"}}},
     {"$sort": {"start": 1}}])]

    assert self._collection.find_one.call_args_list == [
        call({'sy': 's1', 's': dt(2014, 1, 1, 0, 0, tzinfo=mktz())}, {'e': 1}),
        call({'sy': 's2', 's': dt(2014, 1, 1, 12, 0, tzinfo=mktz())}, {'e': 1})]

    assert query == {'s': {'$gte': dt(2014, 1, 1, 12, 0, tzinfo=mktz()), '$lte': dt(2014, 1, 3, 0, 0, tzinfo=mktz())}}

 def test_handle_exception_timeout(self, mock_print_function):
     mock_request = mock.create_autospec(Request).return_value
     mock_request.url = "http://mockurl.mock"
     mock_exception = mock.create_autospec(exceptions.Timeout).return_value
     mock_exception.__str__.return_value = "Timeout reached"
     self.request_sender.handle_request_exception(mock_request, mock_exception)
     self.assertEquals(mock_print_function.call_count, 2)

    def setUp(self):
        self.vlan_id_gen = 'gen_id'
        self.name_gen = 'gen_name'
        self.spec = Mock()
        self.vm = Mock()
        self.pv_service = Mock()
        self.pv_service.find_by_uuid = Mock(self.vm)
        self.si = Mock()
        self.vm_uuid = 'uuid'
        self.vlan_id = 100
        self.spec_type = Mock()

        vnic_device_mapper = create_autospec(spec=VNicDeviceMapper)
        vnic_device_mapper.vnic = create_autospec(spec=vim.vm.device.VirtualEthernetCard)
        vnic_device_mapper.vnic.macAddress = 'AA-BB'
        vnic_device_mapper.vnic.deviceInfo = Mock()
        vnic_device_mapper.vnic.deviceInfo.label = 'AA-BB'
        vnic_device_mapper.network = Mock()
        vnic_device_mapper.network.name = 'the network'
        vnic_device_mapper.network.key = 'keyyyyyey'
        vnic_device_mapper.requested_vnic = 'requested'

        self.dv_connector = Mock()
        self.dv_connector.connect_by_mapping = Mock(return_value=[vnic_device_mapper])
        self.dv_port_name_gen = Mock()
        self.vlan_spec_factory = Mock()
        self.vlan_id_range_parser = Mock()
        self.vlan_id_range_parser.parse_vlan_id = Mock(return_value=self.vlan_id_gen)
        self.dv_port_name_gen.generate_port_group_name = Mock(return_value=self.name_gen)
        self.vlan_spec_factory.get_vlan_spec = Mock(return_value=self.spec)

 def test_handle_exception_connection_error(self, mock_print_function):
     mock_request = mock.create_autospec(Request).return_value
     mock_request.url = "htt;//mockurl.mock"
     mock_exception = mock.create_autospec(exceptions.ConnectionError).return_value
     mock_exception.__str__.return_value = "DNS failure"
     self.request_sender.handle_request_exception(mock_request, mock_exception)
     self.assertEquals(mock_print_function.call_count, 2)

 def test_handle_exception_urlrequired(self, mock_print_function):
     mock_request = mock.create_autospec(Request).return_value
     mock_request.url = "htt;//mockurl.mock"
     mock_exception = mock.create_autospec(exceptions.URLRequired).return_value
     mock_exception.__str__.return_value = "Invalid url"
     self.request_sender.handle_request_exception(mock_request, mock_exception)
     self.assertEquals(mock_print_function.call_count, 2)

def test_evaluate_initializer_with_fallback_calls_fallback():
    init = mock.create_autospec(Initializer())
    fallback = mock.create_autospec(Initializer())
    fallback.side_effect = lambda x: np.array(1)
    init.side_effect = InitializationError
    evaluate_initializer(init, (7, 5), fallback)
    fallback.assert_called_once_with((7, 5))

 def test_handle_exception_baseexception(self, mock_print_function):
     mock_request = mock.create_autospec(Request).return_value
     mock_request.url = "http://mockurl.mock"
     mock_exception = mock.create_autospec(BaseException).return_value
     mock_exception.__str__.return_value = "BaseException message"
     self.request_sender.handle_request_exception(mock_request, mock_exception)
     self.assertEquals(mock_print_function.call_count, 2)

 def setup_resource(self):
     self.mcp = mock.create_autospec(mcp.MasterControlProgram)
     self.resource = www.ConfigResource(self.mcp)
     self.controller = self.resource.controller = mock.create_autospec(
         controller.ConfigController)
     with mock.patch('tron.api.www.respond', autospec=True) as self.respond:
         yield

    def test_create_empty(self):
        # Create a minimal fake GAPIC with a dummy response.
        from google.cloud.firestore_v1beta1.document import DocumentReference
        from google.cloud.firestore_v1beta1.document import DocumentSnapshot
        firestore_api = mock.Mock(spec=['commit'])
        document_reference = mock.create_autospec(DocumentReference)
        snapshot = mock.create_autospec(DocumentSnapshot)
        snapshot.exists = True
        document_reference.get.return_value = snapshot
        commit_response = mock.Mock(
            write_results=[document_reference],
            get=[snapshot],
            spec=['write_results'])
        firestore_api.commit.return_value = commit_response

        # Attach the fake GAPIC to a real client.
        client = _make_client('dignity')
        client._firestore_api_internal = firestore_api
        client.get_all = mock.MagicMock()
        client.get_all.exists.return_value = True

        # Actually make a document and call create().
        document = self._make_one('foo', 'twelve', client=client)
        document_data = {}
        write_result = document.create(document_data)
        self.assertTrue(write_result.get().exists)

 def test_job_runs_from_state(self):
     state_data = [
         dict(
             run_num=i,
             job_name="thename",
             run_time="sometime",
             start_time="start_time",
             end_time="sometime",
             cleanup_run=None,
             runs=[],
         ) for i in range(3, -1, -1)
     ]
     action_graph = mock.create_autospec(actiongraph.ActionGraph)
     output_path = mock.create_autospec(filehandler.OutputPath)
     context = mock.Mock()
     node_pool = mock.create_autospec(node.NodePool)
     runs = jobrun.job_runs_from_state(
         state_data,
         action_graph,
         output_path,
         context,
         node_pool,
     )
     assert len(runs) == 4
     assert all([type(job) == jobrun.JobRun for job in runs])

 def setup_jobrun(self):
     self.action_graph = mock.create_autospec(actiongraph.ActionGraph)
     self.run_time = datetime.datetime(2012, 3, 14, 15, 9, 26)
     self.path = ['base', 'path']
     self.output_path = mock.create_autospec(filehandler.OutputPath)
     self.node_pool = mock.create_autospec(node.NodePool)
     self.action_run_state_data = [{
         'job_run_id': 'thejobname.22',
         'action_name': 'blingaction',
         'state': 'succeeded',
         'run_time': 'sometime',
         'start_time': 'sometime',
         'end_time': 'sometime',
         'command': 'doit',
         'node_name': 'thenode',
     }]
     self.state_data = {
         'job_name': 'thejobname',
         'run_num': 22,
         'run_time': self.run_time,
         'node_name': 'thebox',
         'end_time': 'the_end',
         'start_time': 'start_time',
         'runs': self.action_run_state_data,
         'cleanup_run': None,
         'manual': True,
     }
     self.context = mock.Mock()

    def test_acquired(self):
        """
        Test the acquiring primitives
        """
        self.locker._sequence = '4'
        retval = ('/_locks/test_lock/4', None)
        self.locker._get_locker = mock.create_autospec(
            self.locker._get_locker, return_value=retval)
        self.assertTrue(self.locker._acquired())
        self.assertTrue(self.locker.is_taken)
        retval = ('/_locks/test_lock/1', '/_locks/test_lock/4')
        self.locker._get_locker = mock.MagicMock(return_value=retval)
        self.assertFalse(self.locker._acquired(blocking=False))
        self.assertFalse(self.locker.is_taken)
        d = {
            u'action': u'delete',
            u'node': {
                u'modifiedIndex': 190,
                u'key': u'/_locks/test_lock/1',
                u'value': self.locker.uuid
            }
        }
        self._mock_api(200, d)
        returns = [('/_locks/test_lock/1', '/_locks/test_lock/4'),  ('/_locks/test_lock/4', None)]

        def side_effect():
            return returns.pop()

        self.locker._get_locker = mock.create_autospec(
            self.locker._get_locker, side_effect=side_effect)
        self.assertTrue(self.locker._acquired())

 def create_blank_response(cls, code, msg):
   # TODO(wfarner): Don't use a mock here.
   response = create_autospec(spec=Response, instance=True)
   response.responseCode = code
   response.result = create_autospec(spec=Result, instance=True)
   response.details = [ResponseDetail(message=msg)]
   return response

def test_get_verbose_status_of_marathon_app_column_alignment():
    fake_app = mock.create_autospec(marathon.models.app.MarathonApp)
    fake_app.version = '2015-01-15T05:30:49.862Z'
    fake_app.id = '/fake--service'

    fake_task1 = mock.create_autospec(marathon.models.app.MarathonTask)
    fake_task1.id = 'fake_task1_id'
    fake_task1.host = 'fake_deployed_host'
    fake_task1.ports = [6666]
    fake_task1.staged_at = datetime.datetime.fromtimestamp(0)
    fake_task1.health_check_results = []

    fake_task2 = mock.create_autospec(marathon.models.app.MarathonTask)
    fake_task2.id = 'fake_task2_id'
    fake_task2.host = 'fake_deployed_host_with_a_really_long_name'
    fake_task2.ports = [6666]
    fake_task2.staged_at = datetime.datetime.fromtimestamp(0)
    fake_task2.health_check_results = []

    fake_app.tasks = [fake_task1, fake_task2]
    tasks, out = marathon_serviceinit.get_verbose_status_of_marathon_app(fake_app)
    (_, _, _, headers_line, task1_line, task2_line) = out.split('\n')
    assert headers_line.index('Host deployed to') == task1_line.index('fake_deployed_host')
    assert headers_line.index('Host deployed to') == task2_line.index('fake_deployed_host_with_a_really_long_name')
    assert headers_line.index('Deployed at what localtime') == task1_line.index('1970-01-01T00:00')
    assert headers_line.index('Deployed at what localtime') == task2_line.index('1970-01-01T00:00')

def test_invoke__triggers_max_errors():
    with open('imhotep/fixtures/10line.diff') as f:
        ten_diff = f.read()
    reporter = mock.create_autospec(PRReporter)
    tool = mock.create_autospec(Tool)
    manager = mock.create_autospec(RepoManager)
    tool.get_configs.side_effect = AttributeError
    tool.invoke.return_value = {
        'f1.txt': {
            '1': 'there was an error',
            '2': 'there was an error',
            '3': 'there was an error',
            '4': 'there was an error',
            '5': 'there was an error',
            '6': 'there was an error',
            '7': 'there was an error',
            '8': 'there was an error',
            '9': 'there was an error',
        }
    }
    manager.clone_repo.return_value.diff_commit.return_value = ten_diff
    manager.clone_repo.return_value.tools = [tool]
    imhotep = Imhotep(
        pr_number=1,
        repo_manager=manager,
        commit_info=mock.Mock(),
    )
    imhotep.invoke(reporter=reporter, max_errors=2)

    assert reporter.report_line.call_count == 2
    assert reporter.post_comment.called