def test_k8s_agent_deploy_flows(monkeypatch, runner_token):
k8s_config = MagicMock()
monkeypatch.setattr("kubernetes.config", k8s_config)
batch_client = MagicMock()
batch_client.create_namespaced_job.return_value = {}
monkeypatch.setattr("kubernetes.client.BatchV1Api",
MagicMock(retrurn_value=batch_client))
agent = KubernetesAgent()
agent.deploy_flows(flow_runs=[
GraphQLResult({
"flow":
GraphQLResult({
"storage":
Docker(registry_url="test", image_name="name",
image_tag="tag").serialize(),
"id":
"id",
}),
"id":
"id",
})
])
assert agent.batch_client.create_namespaced_job.called
assert (agent.batch_client.create_namespaced_job.call_args[1]["namespace"]
== "default")
assert (agent.batch_client.create_namespaced_job.call_args[1]["body"]
["apiVersion"] == "batch/v1")
def test_k8s_agent_deploy_flows_continues(monkeypatch, runner_token):
k8s_config = MagicMock()
monkeypatch.setattr("kubernetes.config", k8s_config)
batch_client = MagicMock()
batch_client.create_namespaced_job.return_value = {}
monkeypatch.setattr("kubernetes.client.BatchV1Api",
MagicMock(retrurn_value=batch_client))
agent = KubernetesAgent()
agent.deploy_flows(flow_runs=[
GraphQLResult(
{
"flow": GraphQLResult({
"storage": Local().serialize(),
"id": "id"
}),
"id": "id",
})
])
assert not agent.batch_client.create_namespaced_job.called
