Esempio n. 1
0
  def __init__(self, container_image=None):
    from apache_beam.runners.portability.portable_runner import PortableRunner

    if container_image:
      self.container_image = container_image
    else:
      self.container_image = PortableRunner.default_docker_image()
Esempio n. 2
0
def hasDockerImage():
    image = PortableRunner.default_docker_image()
    try:
        check_image = subprocess.check_output("docker images -q %s" % image,
                                              shell=True)
        return check_image != ''
    except Exception:
        return False
Esempio n. 3
0
 def test__create_default_environment(self):
     docker_image = PortableRunner.default_docker_image()
     self.assertEqual(
         PortableRunner._create_environment(
             PipelineOptions.from_dictionary({})),
         beam_runner_api_pb2.Environment(
             urn=common_urns.environments.DOCKER.urn,
             payload=beam_runner_api_pb2.DockerPayload(
                 container_image=docker_image).SerializeToString()))
 def test__create_default_environment(self):
   docker_image = PortableRunner.default_docker_image()
   self.assertEqual(
       PortableRunner._create_environment(PipelineOptions.from_dictionary({})),
       beam_runner_api_pb2.Environment(
           urn=common_urns.environments.DOCKER.urn,
           payload=beam_runner_api_pb2.DockerPayload(
               container_image=docker_image
           ).SerializeToString()))
Esempio n. 5
0
 def test__create_default_environment(self):
     docker_image = PortableRunner.default_docker_image()
     self.assertEqual(
         PortableRunner._create_environment(
             PipelineOptions.from_dictionary({})),
         environments.DockerEnvironment(container_image=docker_image))
def hasDockerImage():
    image = PortableRunner.default_docker_image()
    check_image = subprocess.check_output(["docker", "images", "-q", image])
    return check_image != ''