コード例 #1
0
    def setUp(self):
        # Call the super setup
        super(EventManagerQuerySetTest, self).setUp()

        # Create a query set reference
        self.queryset = EventQuerySet()

        # Create a manager reference
        self.manager = EventManager()
コード例 #2
0
    def setUp(self):
        # Call the super setup
        super(EventManagerQuerySetTest, self).setUp()

        # Create a query set reference
        self.queryset = EventQuerySet()

        # Create a manager reference
        self.manager = EventManager()
コード例 #3
0
class EventManagerQuerySetTest(TestCase):
    def setUp(self):
        # Call the super setup
        super(EventManagerQuerySetTest, self).setUp()

        # Create a query set reference
        self.queryset = EventQuerySet()

        # Create a manager reference
        self.manager = EventManager()

    def test_mark_seen(self):
        event = G(Event, context={})
        medium = G(Medium)
        Event.objects.mark_seen(medium)
        self.assertEqual(EventSeen.objects.count(), 1)
        self.assertTrue(
            EventSeen.objects.filter(event=event, medium=medium).exists())

    @patch('entity_event.context_loader.load_contexts_and_renderers',
           spec_set=True)
    def test_load_contexts_and_renderers(self,
                                         mock_load_contexts_and_renderers):
        e = G(Event, context={})
        medium = G(Medium)
        Event.objects.load_contexts_and_renderers(medium)
        self.assertEquals(mock_load_contexts_and_renderers.call_count, 1)
        self.assertEquals(
            list(mock_load_contexts_and_renderers.call_args_list[0][0][0]),
            [e])
        self.assertEquals(
            mock_load_contexts_and_renderers.call_args_list[0][0][1], [medium])

    @patch.object(EventManager, 'get_queryset', autospec=True)
    def test_cache_related(self, mock_get_queryset):
        # Setup some mock return values
        mock_get_queryset.return_value = Mock(EventQuerySet(), autospec=True)

        # Call the method
        self.manager.cache_related()

        # Assert that we called get queryset
        mock_get_queryset.assert_called_once_with(self.manager)
コード例 #4
0
class EventManagerQuerySetTest(TestCase):
    def setUp(self):
        # Call the super setup
        super(EventManagerQuerySetTest, self).setUp()

        # Create a query set reference
        self.queryset = EventQuerySet()

        # Create a manager reference
        self.manager = EventManager()

    def test_mark_seen(self):
        event = G(Event, context={})
        medium = G(Medium)
        Event.objects.mark_seen(medium)
        self.assertEqual(EventSeen.objects.count(), 1)
        self.assertTrue(EventSeen.objects.filter(event=event, medium=medium).exists())

    @patch('entity_event.context_loader.load_contexts_and_renderers', spec_set=True)
    def test_load_contexts_and_renderers(self, mock_load_contexts_and_renderers):
        e = G(Event, context={})
        medium = G(Medium)
        Event.objects.load_contexts_and_renderers(medium)
        self.assertEquals(mock_load_contexts_and_renderers.call_count, 1)
        self.assertEquals(list(mock_load_contexts_and_renderers.call_args_list[0][0][0]), [e])
        self.assertEquals(mock_load_contexts_and_renderers.call_args_list[0][0][1], [medium])

    @patch.object(EventManager, 'get_queryset', autospec=True)
    def test_cache_related(self, mock_get_queryset):
        # Setup some mock return values
        mock_get_queryset.return_value = Mock(EventQuerySet(), autospec=True)

        # Call the method
        self.manager.cache_related()

        # Assert that we called get queryset
        mock_get_queryset.assert_called_once_with(self.manager)