Skip to content

Commit

Permalink
inherit correct class, and add minimum one test method to the class.
Browse files Browse the repository at this point in the history
  • Loading branch information
thomasht86 committed Oct 29, 2024
1 parent e07e4d6 commit 4b22d71
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion tests/integration/test_integration_vespa_cloud_vector_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import os
import shutil
import unittest
import pytest
import time
from vespa.application import Vespa, ApplicationPackage
from vespa.package import (
Expand Down Expand Up @@ -238,7 +239,7 @@ def tearDown(self) -> None:
self.vespa_cloud.delete()


class TestProdDeploymentFromDisk(TestVectorSearch):
class TestProdDeploymentFromDisk(unittest.TestCase):
def setUp(self) -> None:
self.app_package = create_vector_ada_application_package()
prod_region = "aws-us-east-1c"
Expand Down Expand Up @@ -286,6 +287,14 @@ def setUp(self) -> None:
application_root=self.application_root,
)

def test_deployment_status(self):
with pytest.raises(TimeoutError):
success = self.vespa_cloud.wait_for_prod_deployment(
build_no=self.build_no,
max_wait=1, # Can take very long. Set to 10 seconds for testing. Will raise TimeoutError if not done.
)
print(success)

@unittest.skip(
"This test is too slow for normal testing. Can be used for manual testing if related code is changed."
)
Expand Down

0 comments on commit 4b22d71

Please sign in to comment.