Skip to content

Commit bba11e9

Browse files
committed
add test
1 parent 870eda0 commit bba11e9

File tree

1 file changed

+42
-0
lines changed

1 file changed

+42
-0
lines changed

Diff for: projects/fal/tests/test_apps.py

+42
Original file line numberDiff line numberDiff line change
@@ -516,6 +516,48 @@ def test_404_response(test_app: str, request: pytest.FixtureRequest):
516516
apps.run(test_app, path="/other", arguments={"lhs": 1, "rhs": 2})
517517

518518

519+
def test_app_deploy_scale(aliased_app: tuple[str, str]):
520+
import uuid
521+
from dataclasses import replace
522+
523+
app_alias = str(uuid.uuid4()) + "-alias"
524+
app_revision = addition_app.host.register(
525+
func=addition_app.func,
526+
options=addition_app.options,
527+
application_name=app_alias,
528+
application_auth_mode="private",
529+
)
530+
531+
host: api.FalServerlessHost = addition_app.host # type: ignore
532+
options = replace(
533+
addition_app.options, host={**addition_app.options.host, "max_concurrency": 30}
534+
)
535+
kwargs = dict(
536+
func=addition_app.func,
537+
options=options,
538+
application_name=app_alias,
539+
application_auth_mode="private",
540+
)
541+
542+
app_revision = addition_app.host.register(**kwargs, scale=False)
543+
544+
with host._connection as client:
545+
res = client.list_aliases()
546+
found = next(filter(lambda alias: alias.alias == app_alias, res), None)
547+
assert found, f"Could not find app {app_alias} in {res}"
548+
assert found.revision == app_revision
549+
assert found.max_multiplexing == 1
550+
551+
app_revision = addition_app.host.register(**kwargs, scale=True)
552+
553+
with host._connection as client:
554+
res = client.list_aliases()
555+
found = next(filter(lambda alias: alias.alias == app_alias, res), None)
556+
assert found, f"Could not find app {app_alias} in {res}"
557+
assert found.revision == app_revision
558+
assert found.max_multiplexing == 30
559+
560+
519561
def test_app_update_app(aliased_app: tuple[str, str]):
520562
app_revision, app_alias = aliased_app
521563

0 commit comments

Comments
 (0)