Python mommy.make函数代码示例

 def test_siguiente_movimiento(self):
     tipo_movimiento = mommy.make(TipoMovimiento, codigo= '', incrementa = True)
     mov1 = mommy.make(Movimiento, id_movimiento = '', fecha_operacion = datetime.now(), tipo_movimiento = tipo_movimiento)
     mov2 = mommy.make(Movimiento, id_movimiento = '', fecha_operacion = datetime.now(), tipo_movimiento = tipo_movimiento)
     mov3 = mommy.make(Movimiento, id_movimiento = '', fecha_operacion = datetime.now(), tipo_movimiento = tipo_movimiento)
     self.assertEqual(mov2.pk, mov1.siguiente())
     self.assertEqual(mov3.pk, mov2.siguiente())

 def test_h1(self):
     testuser = mommy.make(settings.AUTH_USER_MODEL)
     mommy.make('core.Candidate', relatedstudent__user=testuser)
     mockresponse = self.mock_http200_getrequest_htmls(requestuser=testuser)
     self.assertEqual(
             'Student dashboard',
             mockresponse.selector.one('h1').alltext_normalized)

 def setUp(self):
     self.admin_user = User.objects.create_superuser(
         '[email protected]', 'example')
     self.future_date = timezone.now().date() + timedelta(days=10)
     self.job_open = mommy.make(
         Job,
         review_status=Job.OPEN,
     )
     self.job_under_review = mommy.make(
         Job,
         review_status=Job.UNDER_REVIEW,
         reviewer=self.admin_user,
     )
     self.job_ready_to_publish = mommy.make(
         Job,
         review_status=Job.READY_TO_PUBLISH,
         reviewer=self.admin_user,
     )
     self.job_published = mommy.make(
         Job,
         review_status=Job.PUBLISHED,
         reviewer=self.admin_user,
         published_date=timezone.now(),
         expiration_date=self.future_date,
     )
     self.job_rejected = mommy.make(
         Job,
         review_status=Job.REJECTED,
         reviewer=self.admin_user,
     )
     self.client.login(username=self.admin_user.email, password='example')

    def setUpTestData(cls):
        cls.semester = mommy.make(Semester)
        cls.course = mommy.make(Course, pk=7, state="published", semester=cls.semester)

        users = mommy.make(UserProfile, _quantity=3)
        cls.course.participants = users
        cls.course.voters = users[:2]

    def test_results_are_filtered_based_on_queryset(self):

        class Car(models.Model):
            name = models.CharField(max_length=255)

        class CarIndex(Index):
            class Meta:
                fields = ['name']
                model = Car

        with connection.schema_editor() as editor:
            editor.create_model(Car)

        # the signal handler will automatically add these to ES for us
        car = make(Car, name="hi", pk=1)
        car2 = make(Car, name="hi 2", pk=2)
        car3 = make(Car, name="hi 3", pk=3)

        class Form(SearchForm):
            def get_queryset(self):
                # we purposely exclude one of the options, so we can test that
                # it isn't in the search results
                return super().get_queryset().exclude(pk=1)

        form = Form({"q": "hi"}, index=CarIndex)
        # the count should be 2 (not 3), since the queryset excluded Car.pk=1
        self.assertEqual(form.search().count(), 2)
        results = form.results(page=1)
        self.assertEqual(set(results), set([car2, car3]))

        class Form(BaseSearchForm):
            pass

        form = Form({"q": "hi"}, index=CarIndex)
        self.assertEqual(set(form.results()), set([car, car2, car3]))

    def setUp(self):

        self.old_dade_fips = '12025'
        self.new_dade_fips = '12086'

        mommy.make(
            CountyMortgageData,
            current=100,
            date=datetime.date(2008, 1, 1),
            fips=self.old_dade_fips,
            ninety=100,
            other=100,
            sixty=100,
            thirty=100,
            total=500)

        mommy.make(
            CountyMortgageData,
            current=100,
            date=datetime.date(2008, 1, 1),
            fips=self.new_dade_fips,
            ninety=100,
            other=100,
            sixty=100,
            thirty=100,
            total=500)

def test_only_admin_or_coordinator_can_edit_course(client, admin_client):
    course = Course.objects.create(slug='dbsql', name='A course', abstract='asdf')

    professor = create_user('professor')
    client.login(username=professor.username, password='password')

    response = client.post('/api/course/{}'.format(str(course.id)),
                           {'id': course.id,
                            'slug': course.slug,
                            'abstract': 'A abstract'},
                           content_type='application/json;charset=UTF-8')

    assert response.status_code == 403
    assert course.abstract == 'asdf'

    mommy.make('CourseProfessor', user=professor, course=course, role='coordinator')
    response = client.post('/api/course/{}'.format(str(course.id)), {'id': str(course.id),
                                                                     'slug': course.slug,
                                                                     'abstract': 'A abstract'})

    changed_course = Course.objects.get(id=course.id)
    assert response.status_code == 200
    assert changed_course.abstract == 'A abstract'

    response = admin_client.post('/api/course/{}'.format(str(course.id)), {'id': str(course.id),
                                                                           'slug': course.slug,
                                                                           'abstract': 'Another abstract'})
    changed_course = Course.objects.get(id=course.id)
    assert response.status_code == 200
    assert changed_course.abstract == 'Another abstract'

 def test_post_comment_private_history_visibility(self):
     testgroup = mommy.make('core.AssignmentGroup', parentnode__parentnode=self.__make_active_period())
     testuser = mommy.make(settings.AUTH_USER_MODEL)
     self.__make_examiner_for_user(user=testuser, group=testgroup)
     testfeedbackset = group_mommy.feedbackset_first_attempt_unpublished(group=testgroup)
     groupcomment = mommy.make('devilry_group.GroupComment',
                               user=testuser,
                               user_role='examiner',
                               text='unedited',
                               part_of_grading=True,
                               visibility=group_models.GroupComment.VISIBILITY_PRIVATE,
                               feedback_set=testfeedbackset)
     messagesmock = mock.MagicMock()
     self.mock_http302_postrequest(
         cradmin_role=testgroup,
         requestuser=testuser,
         viewkwargs={'pk': groupcomment.id},
         requestkwargs={
             'data': {
                 'text': 'edited'
             }
         },
         messagesmock=messagesmock)
     edit_history_entry = group_models.GroupCommentEditHistory.objects.get()
     self.assertEqual(edit_history_entry.visibility, group_models.GroupComment.VISIBILITY_PRIVATE)

    def test_publish_latest_offer_publishes_oldest_offer(self):
        """
        Test that publish_latest_offer publishes the oldest offer
        """
        self.offer.delete()

        offers = mommy.make(
            Offer,
            _quantity=3,
            is_ready=True,
            is_request=True,
            status=Offer.UNPUBLISHED,
            provider=self.provider,
            creator=self.user
        )
        # Create some offers who should not be affected since they are invalid
        mommy.make(Offer, is_ready=False, is_request=True)
        mommy.make(Offer, status=Offer.PUBLISHED)

        # Set the oldest to newest
        offers[0].readied_at -= timedelta(days=3)
        offers[1].readied_at -= timedelta(days=2)
        offers[2].readied_at -= timedelta(days=1)

        self.assertTrue(publish_latest_offer.delay().successful())

        new_offer = Offer.objects.get(pk=offers[0].pk)

        self.assertEqual(new_offer.status, Offer.PUBLISHED)
        self.assertFalse(new_offer.is_request)

        self.assertEqual(Offer.objects.filter(status=Offer.PUBLISHED, is_request=False).count(), 2)

def test_get_courses_user_has_role(client):
    course = mommy.make('Course', slug='dbsql', name='A course')

    another_course = mommy.make('Course', slug='mysql', name='Another course')

    course_whose_professor_coordinate = mommy.make('Course', slug='coordinatedcourse',
                                                   name='Course whose professor coordinate')

    another_course_whose_professor_coordinate = mommy.make('Course', slug='anothercoordinatedcourse',
                                                           name='Another course whose professor coordinate')

    professor1 = assign_professor_to_course(course, new_professor_username='professor1', role='assistant')

    assign_professor_to_course(another_course, existing_professor=professor1, role='assistant')

    assign_professor_to_course(course_whose_professor_coordinate, existing_professor=professor1, role='coordinator')

    assign_professor_to_course(another_course_whose_professor_coordinate, existing_professor=professor1,
                               role='coordinator')

    client.login(username=professor1.username, password='password')

    response = client.get('/my-courses/')

    assert response.status_code == 200

    courses_user_assist = response.context[-1]['courses_user_assist']

    assert courses_user_assist

    courses_user_coordinate = response.context[-1]['courses_user_coordinate']

    assert courses_user_coordinate

    def setUp(self):
        self.provider = mommy.make(Provider)
        self.offer = mommy.make(Offer, provider=self.provider, status=Offer.PUBLISHED)

        self.user = User.objects.create_user('user', '[email protected]', 'pass')
        self.user.user_profile.provider = self.provider
        self.user.user_profile.save()

 def setUp(self):
     self.fecha_actual = date.today()
     self.fecha_proxima = date(2017,1,1)
     self.pe1 = mommy.make(Pedido,codigo = '',fecha = self.fecha_actual)
     self.pe2 = mommy.make(Pedido,codigo = '',fecha = self.fecha_actual)
     self.pe3 = mommy.make(Pedido,codigo = '',fecha = self.fecha_actual)
     self.pe4 = mommy.make(Pedido,codigo = '',fecha = self.fecha_proxima)

 def test_anterior_movimiento(self):
     tipo_movimiento = mommy.make(TipoMovimiento, codigo= '', incrementa = False)
     mov1 = mommy.make(Movimiento, id_movimiento = '', fecha_operacion = datetime.now(), tipo_movimiento = tipo_movimiento)
     mov2 = mommy.make(Movimiento, id_movimiento = '', fecha_operacion = datetime.now(), tipo_movimiento = tipo_movimiento)
     mov3 = mommy.make(Movimiento, id_movimiento = '', fecha_operacion = datetime.now(), tipo_movimiento = tipo_movimiento)
     self.assertEqual(mov1.pk, mov2.anterior())
     self.assertEqual(mov2.pk, mov3.anterior())

 def test_get_gender_with_registration(self):
     """ Pull from registration, if available """
     reg = mommy.make(Registration, trips_year=self.trips_year, gender='FEMALE')
     incoming = mommy.make(
         IncomingStudent, trips_year=self.trips_year, gender='MALE', registration=reg
     )
     self.assertEqual(incoming.get_gender(), 'female')

 def test_post_initial_empty_comment_can_be_edited(self):
     testgroup = mommy.make('core.AssignmentGroup', parentnode__parentnode=self.__make_active_period())
     testuser = mommy.make(settings.AUTH_USER_MODEL)
     self.__make_examiner_for_user(user=testuser, group=testgroup)
     testfeedbackset = group_mommy.feedbackset_first_attempt_unpublished(group=testgroup)
     groupcomment = mommy.make('devilry_group.GroupComment',
                               user=testuser,
                               user_role='examiner',
                               feedback_set=testfeedbackset)
     messagesmock = mock.MagicMock()
     self.mock_http302_postrequest(
         cradmin_role=testgroup,
         requestuser=testuser,
         viewkwargs={'pk': groupcomment.id},
         requestkwargs={
             'data': {
                 'text': 'edited'
             }
         },
         messagesmock=messagesmock)
     db_comment = group_models.GroupComment.objects.get(id=groupcomment.id)
     edit_history = group_models.GroupCommentEditHistory.objects.get()
     self.assertEqual(group_models.GroupCommentEditHistory.objects.count(), 1)
     self.assertEqual('edited', db_comment.text)
     self.assertEqual('', edit_history.pre_edit_text)
     self.assertEqual('edited', edit_history.post_edit_text)
     messagesmock.add.assert_called_once_with(messages.SUCCESS, 'Comment updated!', '')