Python BuildJobFactory.refresh_from_db Examples

Example #1

File: test_set_dockerfile.py Project: ttsvetanov/polyaxon

    def test_set_dockerfile(self):
        build_job = BuildJobFactory()

        assert build_job.dockerfile is None

        build_jobs_set_dockerfile(build_job_uuid=build_job.uuid.hex, dockerfile='dockerfile')

        build_job.refresh_from_db()
        assert build_job.dockerfile == 'dockerfile'

Example #2

File: test_set_dockerfile.py Project: xuduofeng/polyaxon

    def test_set_dockerfile(self):
        build_job = BuildJobFactory()

        assert build_job.dockerfile is None

        build_jobs_set_dockerfile(build_job_uuid=build_job.uuid.hex,
                                  dockerfile='dockerfile')

        build_job.refresh_from_db()
        assert build_job.dockerfile == 'dockerfile'

Example #3

File: test_models.py Project: thandongtb/polyaxon

 def test_status_update_results_in_new_updated_at_datetime(self):
     job = BuildJobFactory()
     updated_at = job.updated_at
     # Create new status
     BuildJobStatus.objects.create(job=job, status=JobLifeCycle.BUILDING)
     job.refresh_from_db()
     assert updated_at < job.updated_at
     updated_at = job.updated_at
     # Create status Using set_status
     job.set_status(JobLifeCycle.RUNNING)
     job.refresh_from_db()
     assert updated_at < job.updated_at

Example #4

File: test_heart_beat_checks.py Project: yu-iskw/polyaxon

    def test_build_jobs_check_heartbeat(self):
        build1 = BuildJobFactory()
        BuildJobStatusFactory(job=build1, status=JobLifeCycle.RUNNING)
        RedisHeartBeat.build_ping(build_id=build1.id)
        build2 = BuildJobFactory()
        BuildJobStatusFactory(job=build2, status=JobLifeCycle.RUNNING)

        build_jobs_check_heartbeat(build1.id)
        build1.refresh_from_db()
        self.assertEqual(build1.last_status, JobLifeCycle.RUNNING)

        build_jobs_check_heartbeat(build2.id)
        build2.refresh_from_db()
        self.assertEqual(build2.last_status, JobLifeCycle.FAILED)