Skip to content
Merged
Show file tree
Hide file tree
Changes from all 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
72 changes: 72 additions & 0 deletions integration/tests/posit/connect/test_system.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
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

@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()

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

# Barret 2024/12:
# Caches only showing up in Connect versions >= 2024.05.0
# I do not know why.
# Since we are not logic testing Connect, we can confirm our code works given more recent versions of Connect.
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 @@ -15,6 +15,7 @@
from .metrics import Metrics
from .oauth import OAuth
from .resources import ResourceParameters, _PaginatedResourceSequence, _ResourceSequence
from .system import System
from .tags import Tags
from .tasks import Tasks
from .users import User, Users
Expand Down Expand Up @@ -305,6 +306,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")

@property
@requires(version="2023.05.0")
def environments(self) -> Environments:
Expand Down
3 changes: 1 addition & 2 deletions src/posit/connect/groups.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,7 @@
if TYPE_CHECKING:
import requests

from posit.connect.context import Context

from .context import Context
from .users import User


Expand Down
12 changes: 12 additions & 0 deletions src/posit/connect/jobs.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
from collections.abc import Mapping, Sized
from typing import (
Any,
Iterable,
List,
Literal,
Protocol,
Expand Down Expand Up @@ -65,6 +66,17 @@ def __getitem__(self, index: SupportsIndex) -> Job: ...
@overload
def __getitem__(self, index: slice) -> List[Job]: ...

def fetch(self) -> Iterable[Job]:
"""Fetch all jobs.

Fetches all jobs from Connect.

Returns
-------
List[Job]
"""
...

def find(self, key: str, /) -> Job:
"""
Find a Job by its key.
Expand Down
Loading
Loading