Skip to content
Merged
Show file tree
Hide file tree
Changes from 10 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
92 changes: 92 additions & 0 deletions integration/tests/posit/connect/test_system.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
import time
from pathlib import Path

import pytest
from packaging import version

from posit.connect import Client
from posit.connect.system import SystemRuntimeCache
from posit.connect.tasks import Task

from . import CONNECT_VERSION


@pytest.mark.skipif(
# Added to the v2023.05.0 milestone
# https://github.com/rstudio/connect/pull/23148
CONNECT_VERSION < version.parse("2023.05.0"),
reason="Cache runtimes not implemented",
)
class TestSystem:
@classmethod
def setup_class(cls):
cls.client = Client()
assert cls.client.content.count() == 0
cls.content_item = cls.client.content.create(name="Content_A")

# Copied from from integration/tests/posit/connect/test_packages.py
def _deploy_python_bundle(self):
path = Path("../../../resources/connect/bundles/example-flask-minimal/bundle.tar.gz")
path = (Path(__file__).parent / path).resolve()
bundle = self.content_item.bundles.create(str(path))
task = bundle.deploy()
task.wait_for()

def _remove_all_caches(self):
caches: list[SystemRuntimeCache] = self.client.system.caches.runtime.find()
for cache in caches:
assert isinstance(cache, SystemRuntimeCache)
none_val = cache.destroy(dry_run=True)
assert none_val is None
task: Task = cache.destroy()
assert isinstance(task, Task)
task.wait_for()
assert len(self.client.system.caches.runtime.find()) == 0

def _wait_for_cache_jobs(self):
start_time = time.time()
continue_while = True
while continue_while:
now_time = time.time()
if now_time - start_time > 30:
raise TimeoutError("Timeout while waiting for cache to deploy")

jobs = self.content_item.jobs.fetch()
for job in jobs:
if str(job["status"]) == "0":
# Stop for loop and restart while loop
break
else:
# Only executed if the for loop did NOT break

# Break the while loop
continue_while = False

@classmethod
def teardown_class(cls):
cls.content_item.delete()
assert cls.client.content.count() == 0

def test_runtime_caches(self):
# Get current caches
caches: list[SystemRuntimeCache] = self.client.system.caches.runtime.find()
assert isinstance(caches, list)

# Remove all caches
self._remove_all_caches()

# Deploy a new cache
self._deploy_python_bundle()

# Wait for the cache jobs to finish
self._wait_for_cache_jobs()

# Check if the cache is deployed
caches = self.client.system.caches.runtime.find()

# Caches only added in Connect versions >= 2024.05.0
if CONNECT_VERSION >= version.parse("2024.05.0"):
assert len(caches) > 0

# Remove all caches
self._remove_all_caches()
5 changes: 5 additions & 0 deletions src/posit/connect/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

from requests import Response, Session

from posit.connect.system import System
from posit.connect.tags import Tags

from . import hooks, me
Expand Down Expand Up @@ -303,6 +304,10 @@ def packages(self) -> Packages:
def vanities(self) -> Vanities:
return Vanities(self.resource_params)

@property
def system(self) -> System:
return System(self._ctx, "v1/system")

def __del__(self):
"""Close the session when the Client instance is deleted."""
if hasattr(self, "session") and self.session is not None:
Expand Down
Loading
Loading