def test_deploy_schedule(self, monkeypatch): monkeypatch.setenv("GOOGLE_PROJECT", "goblet") monkeypatch.setenv("GOOGLE_LOCATION", "us-central1") monkeypatch.setenv("GOBLET_TEST_NAME", "schedule-deploy") monkeypatch.setenv("GOBLET_HTTP_TEST", "REPLAY") goblet_name = "goblet_example" scheduler = Scheduler(goblet_name) scheduler.register_job( "test-job", None, kwargs={ "schedule": "* * * * *", "timezone": "UTC", "kwargs": {} }, ) scheduler.deploy() responses = get_responses("schedule-deploy") assert goblet_name in responses[0]["body"]["name"] assert (responses[1]["body"]["httpTarget"]["headers"]["X-Goblet-Name"] == "test-job") assert (responses[1]["body"]["httpTarget"]["headers"]["X-Goblet-Type"] == "schedule") assert responses[1]["body"]["schedule"] == "* * * * *"
def test_deploy_multiple_schedule(self, monkeypatch): monkeypatch.setenv("GOOGLE_PROJECT", "goblet") monkeypatch.setenv("GOOGLE_LOCATION", "us-central1") monkeypatch.setenv("GOBLET_TEST_NAME", "schedule-deploy-multiple") monkeypatch.setenv("GOBLET_HTTP_TEST", "REPLAY") goblet_name = "goblet-test-schedule" scheduler = Scheduler(goblet_name) scheduler.register_job("test-job", None, kwargs={ "schedule": "* * 1 * *", "kwargs": {} }) scheduler.register_job( "test-job", None, kwargs={ "schedule": "* * 2 * *", "kwargs": { "httpMethod": "POST" } }, ) scheduler.register_job( "test-job", None, kwargs={ "schedule": "* * 3 * *", "kwargs": { "headers": { "X-HEADER": "header" } }, }, ) scheduler.deploy() post_job_1 = get_response( "schedule-deploy-multiple", "post-v1-projects-goblet-locations-us-central1-jobs_1.json", ) post_job_2 = get_response( "schedule-deploy-multiple", "post-v1-projects-goblet-locations-us-central1-jobs_2.json", ) post_job_3 = get_response( "schedule-deploy-multiple", "post-v1-projects-goblet-locations-us-central1-jobs_3.json", ) assert (post_job_1["body"]["httpTarget"]["headers"]["X-Goblet-Name"] == "test-job") assert (post_job_2["body"]["httpTarget"]["headers"]["X-Goblet-Name"] == "test-job-2") assert post_job_2["body"]["httpTarget"]["httpMethod"] == "POST" assert (post_job_3["body"]["httpTarget"]["headers"]["X-Goblet-Name"] == "test-job-3") assert post_job_3["body"]["httpTarget"]["headers"][ "X-HEADER"] == "header"