Python SubscriptionQuerySet Examples

Example #1

File: models_tests.py Project: travistruett/django-entity-event

class SubscriptionQuerySetTest(TestCase):
    """
    Test the subscription query set class
    """

    def setUp(self):
        # Call super
        super(SubscriptionQuerySetTest, self).setUp()

        # Create a query set to use
        self.queryset = SubscriptionQuerySet()

    @patch.object(SubscriptionQuerySet, 'select_related', autospec=True)
    def test_cache_related(self, mock_select_related):
        # Call the method
        self.queryset.cache_related()

        # Assert that we called select related with the correct args
        mock_select_related.assert_called_once_with(
            self.queryset,
            'medium',
            'source',
            'entity',
            'sub_entity_kind'
        )

Example #2

File: models_tests.py Project: travistruett/django-entity-event

class SubscriptionQuerySetTest(TestCase):
    """
    Test the subscription query set class
    """
    def setUp(self):
        # Call super
        super(SubscriptionQuerySetTest, self).setUp()

        # Create a query set to use
        self.queryset = SubscriptionQuerySet()

    @patch.object(SubscriptionQuerySet, 'select_related', autospec=True)
    def test_cache_related(self, mock_select_related):
        # Call the method
        self.queryset.cache_related()

        # Assert that we called select related with the correct args
        mock_select_related.assert_called_once_with(self.queryset, 'medium',
                                                    'source', 'entity',
                                                    'sub_entity_kind')

Example #3

File: models_tests.py Project: travistruett/django-entity-event

    def setUp(self):
        # Call super
        super(SubscriptionQuerySetTest, self).setUp()

        # Create a query set to use
        self.queryset = SubscriptionQuerySet()

Example #4

File: models_tests.py Project: travistruett/django-entity-event

    def setUp(self):
        # Call super
        super(SubscriptionQuerySetTest, self).setUp()

        # Create a query set to use
        self.queryset = SubscriptionQuerySet()