Kristaps Fabians Geikins
57 lines
2.2 KiB
Python
Executable File
57 lines
2.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)
|
|
|
|
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
|
|
frontend_response = requests.get(urllib.parse.urljoin(SPECKLE_SERVER, 'logo.svg'))
|
|
# 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'
|
|
print("Frontend accessible")
|
|
|
|
# Test basic unauthenticated operation using specklepy
|
|
client = SpeckleClient(SPECKLE_SERVER, use_ssl=SPECKLE_SERVER.startswith('https://'))
|
|
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 == SERVER_VERSION, f"The deployed version {server_info.version} doesn't match 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')
|