Iain Sproat
94 lines
3.2 KiB
Python
Executable File
94 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
|
||
# Script-style deployment testing: any error should fail the test and have non-zero exit code.
|
||
|
||
import os
|
||
import sys
|
||
import requests
|
||
import urllib.parse
|
||
from specklepy.api.client import SpeckleClient
|
||
from specklepy.api.models import ServerInfo
|
||
|
||
|
||
# Setting the SPECKLE_SERVER to test on
|
||
SPECKLE_SERVER = ""
|
||
|
||
if len(sys.argv) > 1:
|
||
SPECKLE_SERVER = sys.argv[1]
|
||
if not SPECKLE_SERVER:
|
||
SPECKLE_SERVER = os.getenv("SPECKLE_SERVER", "")
|
||
if not SPECKLE_SERVER:
|
||
print(
|
||
"❌ Error: No Speckle server specified. Use SPECKLE_SERVER environment variable or pass it as the first command-line argument"
|
||
)
|
||
exit(1)
|
||
|
||
VERIFY_CERTIFICATE = (
|
||
True if os.getenv("VERIFY_CERTIFICATE", "1") != "0" else False
|
||
) # default to True except in very narrow case where value is explicitly "0"
|
||
|
||
FRONTEND_VERSION = os.getenv("FRONTEND_VERSION", "")
|
||
if not FRONTEND_VERSION:
|
||
print(
|
||
"❌ Error: No frontend version specified. Use FRONTEND_VERSION environment variable."
|
||
)
|
||
exit(1)
|
||
|
||
if not SPECKLE_SERVER.startswith("http://") and not SPECKLE_SERVER.startswith(
|
||
"https://"
|
||
):
|
||
SPECKLE_SERVER = "http://" + SPECKLE_SERVER
|
||
|
||
print(f"ℹ️ Using Speckle server '{SPECKLE_SERVER}'")
|
||
|
||
# Test if frontend is accessible
|
||
if FRONTEND_VERSION == "1":
|
||
frontend_response = requests.get(
|
||
urllib.parse.urljoin(SPECKLE_SERVER, "logo.svg"), verify=VERIFY_CERTIFICATE
|
||
)
|
||
# don't check for status code, the frontend app will server the 404 page with a status code 200
|
||
# even if the rote doesn't exist
|
||
assert frontend_response.headers.get("Content-Type", "").startswith(
|
||
"image/"
|
||
), "❌ Frontend logo Content-Type is not an image"
|
||
elif FRONTEND_VERSION == "2":
|
||
frontend_response = requests.get(SPECKLE_SERVER, verify=VERIFY_CERTIFICATE)
|
||
assert (
|
||
frontend_response.status_code == 200
|
||
), f"❌ Frontend did not return a 200 status code, instead returned status code is {frontend_response.status_code}."
|
||
else:
|
||
print(f"❌ Unknown frontend version '{FRONTEND_VERSION}'")
|
||
exit(1)
|
||
print("✅ Frontend accessible")
|
||
|
||
# Test basic unauthenticated operation using specklepy
|
||
client = SpeckleClient(
|
||
SPECKLE_SERVER,
|
||
use_ssl=SPECKLE_SERVER.startswith("https://"),
|
||
verify_certificate=VERIFY_CERTIFICATE,
|
||
)
|
||
server_info = client.server.get()
|
||
assert isinstance(server_info, ServerInfo), "❌ GraphQL ServerInfo query error"
|
||
print(f"✅ GraphQL operation succeeded. Server name: {server_info.name}")
|
||
|
||
# Test that the deployed server version matches the expected version
|
||
SERVER_VERSION = ""
|
||
if len(sys.argv) > 2:
|
||
SERVER_VERSION = sys.argv[2]
|
||
if not SERVER_VERSION:
|
||
SERVER_VERSION = os.getenv("SERVER_VERSION")
|
||
if SERVER_VERSION:
|
||
if not SERVER_VERSION == "latest":
|
||
assert server_info.version.startswith(
|
||
SERVER_VERSION
|
||
), f"❌ The deployed version {server_info.version} should match, or be prefixed by, the expected {SERVER_VERSION}"
|
||
print(f"✅ Server version {SERVER_VERSION} is deployed and available")
|
||
else:
|
||
print("🟡 Not testing server version, as it was set to 'latest'")
|
||
else:
|
||
print(
|
||
"🟡 Not testing server version, as an expected value was not provided via environment variables or command-line argument"
|
||
)
|
||
|
||
print("✅ Deployment tests PASS ✅")
|