Python mommy.make_recipe函数代码示例

    def test_create_block_redirects_if_no_blocktypes_available(self):
        """
        Test that create block form redirects if trying to create a
        block with an event type that the user already has
        """
        event_type = mommy.make_recipe('booking.event_type_PC')
        block_type_pc5 = mommy.make_recipe(
            'booking.blocktype5', event_type=event_type
        )
        block_type_pc10 = mommy.make_recipe(
            'booking.blocktype5', event_type=event_type
        )
        mommy.make_recipe(
            'booking.block', user=self.user, block_type=block_type_pc5,
            paid=True
        )
        # create form with a different blocktype, but one with the same event
        # type as the user's booked block
        form_data = {'block_type': block_type_pc10}

        url = reverse('booking:add_block')
        request = self.factory.post(url, form_data)
        self._set_session(self.user, request)

        form_data = {'block_type': block_type_pc10.id}
        form = BlockCreateForm(data=form_data)
        form.full_clean()
        view = setup_view(BlockCreateView, request)
        resp = view.form_valid(view, form)

        self.assertEqual(resp.status_code, 302)

    def test_image_operations_as_authorised_users(self):
        # create a campaign & deployment that ONLY bill can see
        bills_campaign = mommy.make_one("catamidb.Campaign", id=3, short_name="cp__1")
        bills_deployment = mommy.make_recipe("catamidb.Deployment3", id=3, campaign=bills_campaign)
        bills_image = mommy.make_recipe("catamidb.Image3", id=3, deployment=bills_deployment)
        assign("view_campaign", self.user_bill, bills_campaign)

        # check that bill can see via the API
        response = self.bill_api_client.get(self.image_url, format="json")
        self.assertValidJSONResponse(response)
        self.assertEqual(len(self.deserialize(response)["objects"]), 3)

        # check that bill can get to the object itself
        response = self.bill_api_client.get(self.image_url + "3/", format="json")
        self.assertValidJSONResponse(response)

        # check that bob can not see - now we know tastypie API has correct
        # permission validation
        response = self.bob_api_client.get(self.image_url, format="json")
        self.assertValidJSONResponse(response)
        self.assertEqual(len(self.deserialize(response)["objects"]), 2)

        # check bob can NOT get to the hidden object
        response = self.bob_api_client.get(self.image_url + "3/", format="json")
        self.assertHttpUnauthorized(response)

        # check that anonymous can see public ones as well
        response = self.anon_api_client.get(self.image_url, format="json")
        self.assertValidJSONResponse(response)
        self.assertEqual(len(self.deserialize(response)["objects"]), 2)

        # check anonymous can NOT get to the hidden object
        response = self.anon_api_client.get(self.image_url + "3/", format="json")
        self.assertHttpUnauthorized(response)

 def test_not_assignments_where_not_student(self):
     testuser = mommy.make(settings.AUTH_USER_MODEL)
     mommy.make_recipe('devilry.apps.core.assignment_activeperiod_start')
     mockresponse = self.mock_http200_getrequest_htmls(requestuser=testuser)
     self.assertEqual(
             0,
             self.__get_assignment_count(selector=mockresponse.selector))

 def testContent(self):
     mommy.make_recipe('apps.front.user')
     mommy.make_recipe('apps.lecturers.lecturer')
     login(self)
     response = self.client.get('/dozenten/')
     self.assertContains(response, '<h1>Unsere Dozenten</h1>')
     self.assertContains(response, 'David<br />Krakaduku')

 def test_grouplist_status_waiting_for_feedback_sanity(self):
     testuser = mommy.make(settings.AUTH_USER_MODEL)
     testperiod = mommy.make_recipe('devilry.apps.core.period_active')
     testgroup = mommy.make('core.AssignmentGroup',
                            parentnode=mommy.make_recipe('devilry.apps.core.assignment_activeperiod_start',
                                                         parentnode=testperiod))
     mommy.make('core.Candidate',
                relatedstudent__user=testuser,
                assignment_group=testgroup)
     devilry_group_mommy_factories.feedbackset_first_attempt_published(
             group=testgroup, grading_points=3)
     devilry_group_mommy_factories.feedbackset_new_attempt_unpublished(
             group=testgroup,
             deadline_datetime=timezone.now() - timedelta(days=2))
     mockresponse = self.mock_http200_getrequest_htmls(
             requestuser=testuser, cradmin_role=testperiod)
     self.assertFalse(
             mockresponse.selector.exists(
                     '.devilry-cradmin-groupitemvalue '
                     '.devilry-cradmin-groupitemvalue-grade'))
     self.assertEqual(
             'Status: waiting for feedback',
             mockresponse.selector.one(
                     '.devilry-cradmin-groupitemvalue '
                     '.devilry-cradmin-groupitemvalue-status').alltext_normalized)

    def test_status_choice_filter(self):
        open_bookings = mommy.make_recipe(
            'booking.booking', event=self.event, status='OPEN', _quantity=5
            )
        cancelled_bookings = mommy.make_recipe(
            'booking.booking',
            event=self.event,
            status='CANCELLED',
            _quantity=5
            )
        resp = self._get_response(
            self.staff_user, self.event.slug, status_choice='ALL'
        )
        # bookings: open - 5 plus 2 created in setup, cancelled = 5 (12)
        # also shows forms for available spaces (16 max, 9 spaces)
        self.assertEquals(len(resp.context_data['formset'].forms), 21)

        resp = self._get_response(
            self.staff_user, self.event.slug, status_choice='OPEN'
        )
        # 5 open plus 2 created in setup, plus empty forms for available
        # spaces to max participants 16
        forms = resp.context_data['formset'].forms
        self.assertEquals(len(forms), 16)
        self.assertEquals(
            set([form.instance.status for form in forms]), {'OPEN'}
            )

        resp = self._get_response(
            self.staff_user, self.event.slug, status_choice='CANCELLED'
        )
        forms = resp.context_data['formset'].forms
        # 5 cancelled plus empty forms for 9 available spaces
        self.assertEquals(len(forms), 14)

    def test_events_excluded_from_form_for_instructors(
            self, mock_tz, mock_date
    ):
        mock_tz.now.return_value = datetime(
            year=2015, month=9, day=7, hour=10, tzinfo=timezone.utc
        )
        mock_date.today.return_value = date(year=2015, month=9, day=7)
        mommy.make_recipe(
            'booking.future_EV',
            date=datetime(
                year=2015, month=9, day=7,
                hour=18, minute=0, tzinfo=timezone.utc
            ),
            _quantity=3
        )
        pole_classes = mommy.make_recipe(
            'booking.future_PC',
            date=datetime(
                year=2015, month=9, day=7,
                hour=18, minute=0, tzinfo=timezone.utc
            ),
            _quantity=3
        )
        resp = self._get_response(self.instructor_user)

        form = resp.context_data['form']
        self.assertEqual(len(form.events), 3)
        self.assertEqual(
            sorted([ev.id for ev in form.events]),
            sorted([ev.id for ev in pole_classes])
        )

 def test_create_existing_booking_txn_with_txn_id(self):
     """
     if the existing transaction is already associated with a paypal
     transaction_id, we do need to create a new transaction, with new
     invoice number with incremented counter
     """
     user = mommy.make_recipe('flex_bookings.user', username="testuser")
     booking = mommy.make_recipe(
         'flex_bookings.booking', user=user, event__name='test event'
     )
     booking_txn = helpers.create_booking_paypal_transaction(user, booking)
     self.assertEqual(booking_txn.booking, booking)
     self.assertEqual(
         booking_txn.invoice_id,
         'testuser-te-{}-inv#001'.format(
             booking.event.date.strftime("%d%m%y%H%M")
         )
     )
     self.assertEqual(PaypalBookingTransaction.objects.count(), 1)
     booking_txn.transaction_id = "123"
     booking_txn.save()
     new_booking_txn = helpers.create_booking_paypal_transaction(user, booking)
     self.assertEqual(PaypalBookingTransaction.objects.count(), 2)
     self.assertEqual(
         new_booking_txn.invoice_id,
         'testuser-te-{}-inv#002'.format(
             booking.event.date.strftime("%d%m%y%H%M")
         )
     )

    def test_create_booking_with_duplicate_invoice_number(self):
        user = mommy.make_recipe('flex_bookings.user', username="testuser")
        booking = mommy.make_recipe(
            'flex_bookings.booking', user=user, event__name='test event',
            event__date=datetime(2015, 2, 1, 10, 0, tzinfo=timezone.utc)
        )
        booking1 = mommy.make_recipe(
            'flex_bookings.booking', user=user, event__name='test event1',
            event__date=datetime(2015, 2, 1, 10, 0, tzinfo=timezone.utc)
        )
        booking_txn = helpers.create_booking_paypal_transaction(user, booking)
        self.assertEqual(booking_txn.booking, booking)
        self.assertEqual(
            booking_txn.invoice_id,
            'testuser-te-{}-inv#001'.format(
                booking.event.date.strftime("%d%m%y%H%M")
            )
        )

        booking1_txn = helpers.create_booking_paypal_transaction(user, booking1)
        self.assertEqual(booking1_txn.booking, booking1)
        self.assertNotEqual(
            booking1_txn.invoice_id,
            'testuser-te-{}-inv#001'.format(
                booking1.event.date.strftime("%d%m%y%H%M")
            )
        )
        # to avoid duplication, the counter is set to 6 digits, the first 3
        # random between 100 and 999
        self.assertEqual(len(booking1_txn.invoice_id.split('#')[-1]), 6)

    def test_paypal_booking_admin_display(self):
        user = mommy.make_recipe(
            'flex_bookings.user', first_name='Test', last_name='User')
        booking = mommy.make_recipe('flex_bookings.booking', user=user)
        pptrans = helpers.create_booking_paypal_transaction(
            booking.user, booking
        )

        ppbooking_admin = PaypalBookingTransactionAdmin(
            PaypalBookingTransaction, AdminSite()
        )
        ppbooking_query = ppbooking_admin.get_queryset(None)[0]

        self.assertEqual(
            ppbooking_admin.get_booking_id(ppbooking_query), booking.id
        )
        self.assertEqual(
            ppbooking_admin.get_user(ppbooking_query), 'Test User'
        )
        self.assertEqual(
            ppbooking_admin.get_event(ppbooking_query), booking.event
        )
        self.assertEqual(
            ppbooking_admin.cost(ppbooking_query),
            u"\u00A3{}.00".format(booking.event.cost)
        )

    def test_paypal_block_admin_display(self):
        user = mommy.make_recipe(
            'flex_bookings.user', first_name='Test', last_name='User'
        )
        events = mommy.make_recipe('flex_bookings.future_EV', _quantity=5)
        block = mommy.make_recipe('flex_bookings.block')
        for event in events:
            block.events.add(event)
            booking = mommy.make_recipe(
                'flex_bookings.booking', user=user, event=event
            )
            block.bookings.add(booking)
        pptrans = helpers.create_block_paypal_transaction(
            user, block
        )

        ppblock_admin = PaypalBlockTransactionAdmin(
            PaypalBlockTransaction, AdminSite()
        )
        ppblock_query = ppblock_admin.get_queryset(None)[0]

        self.assertEqual(
            ppblock_admin.get_block_id(ppblock_query), block.id
        )
        self.assertEqual(
            ppblock_admin.get_user(ppblock_query), 'Test User'
        )
        self.assertEqual(
            ppblock_admin.cost(ppblock_query),
            u"\u00A3{:.2f}".format(
                block.item_cost * block.bookings.count()
            )
        )

    def test_categories_with_published_articles(self):
        c1 = mommy.make_recipe('kb.tests.category_with_articles')
        c2 = mommy.make_recipe('kb.tests.category_without_articles')
        mommy.make_recipe('kb.tests.draft_article', category=c2)

        self.assertSeqEqual(Category.objects.all(), [c1, c2])
        self.assertSeqEqual(Category.objects.categories(), [c1])

 def test_upcoming_assignments_header(self):
     testuser = mommy.make(settings.AUTH_USER_MODEL)
     mommy.make_recipe('devilry.apps.core.assignment_activeperiod_start')
     mockresponse = self.mock_http200_getrequest_htmls(requestuser=testuser)
     self.assertEqual(
         mockresponse.selector.one('.devilry-student-dashboard-upcoming-assignments-header').alltext_normalized,
         'Upcoming assignments')

    def test_cannot_post_delete_block_page_if_block_has_bookings(self):
        mommy.make_recipe('booking.booking', block=self.block)
        resp = self._post_response(self.user, self.block.id)
        self.assertEqual(resp.status_code, 302)
        self.assertIn(resp.url, reverse('booking:permission_denied'))

        self.assertEqual(Block.objects.first(), self.block)

    def test_block_format_block_available_not_used(self):
        # paid (user has available block not used)
        event = mommy.make_recipe(
            'booking.future_EV', max_participants=1, event_type__subtype='Event'
        )
        # block used
        block_type = mommy.make(
            BlockType, event_type=event.event_type, size=3
        )
        block = mommy.make_recipe(
            'booking.block', block_type=block_type, user=self.user, paid=True
        )
        mommy.make_recipe(
            'booking.booking',  event=event, user=self.user, paid=True,
            payment_confirmed=True
        )
        resp = self._get_response(self.staff_user, event.slug)

        # block is hidden as booking is paid
        self.assertIn(
            '<span class="hide"><select class="form-control input-xs '
            'studioadmin-list" id="id_bookings-0-block" '
            'name="bookings-0-block">',
            resp.rendered_content
        )
        # block info is displayed
        self.assertIn(
            'Active block not used', resp.rendered_content
        )