Add package data for GPU rent and update CLI documentation
- Added package data configuration for the 'gpu_rent' package in pyproject.toml. - Updated README.md to include usage instructions for Windows and Unix launchers. - Enhanced CLI documentation in cli.md to reflect new commands and their functionalities. - Revised setup.md to clarify installation steps and environment setup. - Improved error handling and command descriptions in the CLI implementation. - Added new functions for model version handling and flavor resolution in the codebase. - Updated state management to include additional properties for better tracking.
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
from gpu_rent.cloud import ensure_floating_ip, server_floating_ip, server_status
|
||||
from gpu_rent.lock import SessionLock
|
||||
from gpu_rent.paths import lock_path
|
||||
|
||||
|
||||
class Obj:
|
||||
def __init__(self, **kwargs):
|
||||
self.__dict__.update(kwargs)
|
||||
|
||||
|
||||
def test_server_floating_ip():
|
||||
server = Obj(
|
||||
status="ACTIVE",
|
||||
addresses={"gpu-rent": [{"addr": "192.168.77.10", "OS-EXT-IPS:type": "fixed"}, {"addr": "203.0.113.9", "OS-EXT-IPS:type": "floating"}]},
|
||||
)
|
||||
assert server_status(server) == "ACTIVE"
|
||||
assert server_floating_ip(server) == "203.0.113.9"
|
||||
|
||||
|
||||
def test_lock_roundtrip():
|
||||
with SessionLock():
|
||||
assert lock_path().is_file()
|
||||
assert not lock_path().is_file()
|
||||
|
||||
|
||||
def test_ensure_floating_ip_reuses_existing():
|
||||
class Net:
|
||||
def ports(self, device_id=None):
|
||||
return [Obj(id="p1")]
|
||||
|
||||
def update_ip(self, existing_id, port_id=None):
|
||||
assert existing_id == "old-fip"
|
||||
assert port_id == "p1"
|
||||
return Obj(floating_ip_address="9.9.9.9")
|
||||
|
||||
def create_ip(self, floating_network_id=None):
|
||||
raise AssertionError("must reuse FIP")
|
||||
|
||||
conn = Obj(network=Net())
|
||||
server = Obj(id="s1", addresses={})
|
||||
ip, fip_id = ensure_floating_ip(conn, server, "old-fip", "9.9.9.9", lambda m: None)
|
||||
assert ip == "9.9.9.9"
|
||||
assert fip_id == "old-fip"
|
||||
Reference in New Issue
Block a user