test.py

import sys

import grpc
from src.grpc_connector import client_pb2_grpc
from src.grpc_connector import client_pb2
import time

def full_test():
    ip = "192.168.161.164"
    user = "ubuntu"
    passphrase = ""

    start_time = time.time()
    #START VM
    channel = grpc.insecure_channel("localhost:50052")
    stub = client_pb2_grpc.OperationHandlerStub(channel)

    f = open("tests/ansible-package.tar", "r")
    dc = client_pb2.FileMessage(file=f.read())
    f.close()
    rg = stub.Create(dc)

    print(rg)

    # Wait a bit so that the vm has time to init
    time.sleep(5)
    identifier = client_pb2.ResourceIdentifier(resource_id="vm1")

    stub.StopContainer(identifier)

    time.sleep(2)

    stub.StartContainer(identifier)
    time.sleep(5)
    
    # UPLOAD FILES
    f = open("README.md", "rb")
    upld_message_with_path = client_pb2.RuntimeMessage(resource_id=ip,
                                                       property=["README.md", user, passphrase], file=f.read())
    stub.UploadFile(upld_message_with_path)
    f.close()

    upld_message_with_path = client_pb2.RuntimeMessage(resource_id=ip,
                                                       property=["withPath", user, passphrase, ".gitignore", ".gitignore"])
    stub.UploadFile(upld_message_with_path)

    # EXECUTE LS
    command = "ls"
    commandMsg = client_pb2.RuntimeMessage(resource_id=ip, property=[command, user, passphrase], file=None)
    response = stub.ExecuteCommand(commandMsg)
    print(response)

    # DOWNLOAD FILE
    dwnld_message = client_pb2.RuntimeMessage(resource_id=ip, property=[".gitignore", user, passphrase], file=None)
    output = stub.DownloadFile(dwnld_message)
    print(output)

    # TERMINATE
    stub.Remove(identifier)

    print("Test finished in: " + str(time.time() - start_time))


if __name__ == "__main__":

    full_test()