def test_program(self):
#channel = implementations.insecure_channel('localhost', 9713)
channel = implementations.insecure_channel(self.agro_server, 9713)
print "Connected"
sched = agro_pb2.beta_create_Scheduler_stub(channel)
filestore = agro_pb2.beta_create_FileStore_stub(channel)
task = agro_pb2.Task()
task_id = str(uuid.uuid4())
task.id = task_id
task.command = "/usr/bin/md5sum"
task.container = "ubuntu"
file_id = str(uuid.uuid4())
pyagro.upload_file(filestore, file_id, os.path.join(BASE_DIR, "..", "README"))
task.args.add( file_arg=agro_pb2.FileArgument(
id=file_id,
input=True,
silent=False,
type=agro_pb2.FileArgument.PATH),
)
output_uuid = str(uuid.uuid4())
task.args.add( file_arg=agro_pb2.FileArgument(
id=output_uuid,
input=False,
silent=False,
type=agro_pb2.FileArgument.STDOUT),
)
task.tags.extend( ['testing'] )
print "Adding task"
sched.AddTask(task, 10)
e = pyagro.wait(sched, task_id)
assert(e == 0)
print "Result File", output_uuid
sched = None
filestore = None
def filestore(self):
stub = agro_pb2.beta_create_FileStore_stub(self.channel)
return AgroFileStore(stub)
