test_upload_data.py

#!/usr/bin/env python3
import subprocess
import gspread
import os
import sys


def check_data():
    """
    Checks if Google Sheets is updated with the correct data.
    """
    global test_filename
    worksheet_value = worksheet.get_all_values()
    test_content = open(test_filename, "r").read()
    test_content_list = test_content.split("\n")
    test_content_list = list(filter(lambda elem: elem != "", test_content_list))
    expected_result = list(map(lambda x: x.split(","), test_content_list))
    expected_result = list(filter(lambda elem: elem != "", expected_result))
    if expected_result == worksheet_value:
        print("\u001b[32mTest successful\u001b[0m")
    else:
        print("\u001b[31mTest failed\u001b[0m")


# Initializes Google Sheets document.
gc = gspread.service_account()
sh = gc.open_by_key("11V4KfT00lrys2zHgLtRlF13q3SP-6n1CS_vbCyLmtqA")

#If program crashes, the google spreadsheet created earlier is not deleted. Delete old spreadsheet manually.
worksheet_title = "TEST"

worksheet = sh.add_worksheet(title=worksheet_title, rows="10", cols="10")

test_filename = "test_data.csv"

with open(test_filename, "w") as fd:
    data = ["DATUM,NTI,PROCIVITAS,NTI_TEACHER,PROCIVITAS_TEACHER\n", "2020-11-13,1,1,1,1\n", "2020-11-14,50,60,20,30\n", "2020-11-15,100,60,10,15\n"]
    fd.writelines(data)

args = [sys.executable, "upload_data.py", "--data", test_filename, "--worksheet", worksheet_title]
p = subprocess.Popen(args, stdout=subprocess.PIPE, stdin=subprocess.PIPE)
p.communicate()

# If the exit code from the subprocess does not equal 0
if p.returncode:
    print("\u001b[31mTest failed\u001b[0m")
else:
    check_data()

sh.del_worksheet(worksheet)
os.remove(test_filename)