Python factory.create_resource函数代码示例

    def test_token_guessing(self):
        """Test guessing the resource's token."""
        r = create_resource(self.item_payload, '')
        self.assertTrue('resource_token' not in r.fields)
        for field in self.item_payload['resource_token']:
            self.assertTrue(field in r.fields)

        r = create_resource(self.count_payload, '')
        self.assertTrue('count' in r.fields)

    def test_no_token_guessing(self):
        """Test constructing without guessing the resource token."""
        r = create_resource(self.item_payload, '', guess_token=False)
        self.assertTrue('resource_token' in r.fields)
        self.assertTrue('field1' not in r.fields)
        self.assertTrue('field1' in r.fields['resource_token'])

        r = create_resource(self.list_payload, '', guess_token=False)
        self.assertTrue('resource_token' in r.fields)

 def test_resource_specific_base_class(self):
     """Test constructing a resource with a specific base class."""
     r = create_resource(self.root_payload, '')
     self.assertFalse(isinstance(r, RootResource))
     r = create_resource(
         self.root_payload,
         '',
         mime_type='application/vnd.reviewboard.org.root+json')
     self.assertTrue(isinstance(r, RootResource))

    def setUp(self):
        self.transport = MockSyncTransport()
        self.item_resource = create_resource(self.item_payload, '')
        self.item = SyncTransportItemResource(self.transport,
                                              self.item_resource)

        self.list_resource = create_resource(self.list_payload, '')
        self.list = SyncTransportListResource(self.transport,
                                              self.list_resource)

    def test_no_token_guessing(self):
        """Testing constructing without guessing the resource token"""
        r = create_resource(self.transport, self.item_payload, '',
                            guess_token=False)
        self.assertTrue('resource_token' in r)
        self.assertTrue('field1' not in r)
        self.assertTrue('field1' in r.resource_token)

        r = create_resource(self.transport, self.list_payload, '',
                            guess_token=False)
        self.assertTrue('resource_token' in r)

def _create_resource_for_field(parent_resource, field_payload,
                               mimetype, item_mimetype=None, url=None):
    """Create a resource instance based on field data.

    This will construct a resource instance for the payload of a field,
    using the given mimetype to identify it. This is intended for use with
    expanded resources or items in lists.

    Args:
        parent_resource (Resource):
            The resource containing the field payload.

        field_payload (dict):
            The field payload to use as the new resource's payload.

        mimetype (unicode):
            The mimetype of the resource.

        item_mimetype (unicode, optional):
            The mimetype of any items in the resource, if this resource
            represents a list.

        url (unicode, optional):
            The URL of the resource, if one is available.
    """
    # We need to import this here to avoid circular imports.
    from rbtools.api.factory import create_resource

    return create_resource(transport=parent_resource._transport,
                           payload=field_payload,
                           url=url,
                           mime_type=mimetype,
                           item_mime_type=item_mimetype,
                           guess_token=False)

    def test_list_resource_links(self):
        """Testing link resource link generation."""
        r = create_resource(self.transport, self.list_payload, '')

        self.assertTrue(hasattr(r, 'get_self'))
        self.assertTrue(callable(r.get_self))
        request = r.get_self()
        self.assertTrue(isinstance(request, HttpRequest))
        self.assertEqual(request.method, 'GET')
        self.assertEqual(request.url,
                         self.list_payload['links']['self']['href'])

        self.assertTrue(hasattr(r, 'create'))
        self.assertTrue(callable(r.create))
        request = r.create()
        self.assertTrue(isinstance(request, HttpRequest))
        self.assertEqual(request.method, 'POST')
        self.assertEqual(request.url,
                         self.list_payload['links']['create']['href'])

        self.assertTrue(hasattr(r, 'get_other_link'))
        self.assertTrue(callable(r.get_other_link))
        request = r.get_other_link()
        self.assertTrue(isinstance(request, HttpRequest))
        self.assertEqual(request.method, 'GET')
        self.assertEqual(request.url,
                         self.list_payload['links']['other_link']['href'])

        self.assertFalse(hasattr(r, 'update'))
        self.assertFalse(hasattr(r, 'delete'))

 def test_item_construction(self):
     """Testing constructing an item resource."""
     r = create_resource(self.transport, self.item_payload, '')
     self.assertTrue(isinstance(r, ItemResource))
     self.assertEqual(r.field1,
                      self.item_payload['resource_token']['field1'])
     self.assertEqual(r.field2,
                      self.item_payload['resource_token']['field2'])

 def test_item_construction(self):
     """Test constructing an item resource."""
     r = create_resource(self.item_payload, '')
     self.assertTrue(isinstance(r, ResourceItem))
     self.assertEqual(r.fields['field1'],
                      self.item_payload['resource_token']['field1'])
     self.assertEqual(r.fields['field2'],
                      self.item_payload['resource_token']['field2'])

    def test_resource_dict_field(self):
        """Testing access of a dictionary field."""
        r = create_resource(self.transport, self.item_payload, '')

        field = r.nested_field

        self.assertTrue(isinstance(field, ResourceDictField))
        self.assertEqual(
            field.nested1,
            self.item_payload['resource_token']['nested_field']['nested1'])

    def test_getattr_with_expanded_item_resource(self):
        """Testing ItemResource.__getattr__ with field as expanded item
        resource
        """
        r = create_resource(transport=self.transport,
                            payload=self.expanded_item_payload,
                            url='')

        self.assertIsInstance(r['item1'], ExpandedItemResource)
        self.assertIsInstance(r['item2'], ResourceDictField)
        self.assertIsInstance(r['other-item'], ResourceDictField)

    def test_root_resource_templates(self):
        """Test generation of methods for the root resource uri templates."""
        r = create_resource(
            self.root_payload,
            '',
            mime_type='application/vnd.reviewboard.org.root+json')

        for template_name in self.root_payload['uri_templates']:
            method_name = "get_%s" % template_name
            self.assertTrue(hasattr(r, method_name))
            self.assertTrue(callable(getattr(r, method_name)))

    def test_resource_dict_field_iteration(self):
        """Testing iterating sub-fields of a dictionary field."""
        r = create_resource(self.transport, self.item_payload, '')

        field = r.nested_field
        iterated_fields = set(f for f in field.iterfields())
        nested_fields = set(
            f for f in self.item_payload['resource_token']['nested_field'])

        self.assertEqual(set(),
                         nested_fields.symmetric_difference(iterated_fields))

    def test_iteritems_with_expanded_item_resource(self):
        """Testing ItemResource.iteritems with field as expanded item resource
        """
        r = create_resource(transport=self.transport,
                            payload=self.expanded_item_payload,
                            url='')
        items = dict(r.iteritems())

        self.assertIsInstance(items['item1'], ExpandedItemResource)
        self.assertIsInstance(items['item2'], ResourceDictField)
        self.assertIsInstance(items['other-item'], ResourceDictField)

    def test_list_resource_list(self):
        """Testing list resource lists."""
        r = create_resource(self.transport, self.list_payload, '')
        self.assertEqual(r.num_items, len(self.list_payload['resource_token']))
        self.assertEqual(r.total_results, self.list_payload['total_results'])

        for index in range(r.num_items):
            for field in r[index].iterfields():
                self.assertEqual(
                    r[index][field],
                    self.list_payload['resource_token'][index][field])