Skip to content

Commit

Permalink
update Token Authentication for TestMnistJobSubmit before step get_cl…
Browse files Browse the repository at this point in the history
…uster (#561)
  • Loading branch information
Srihari1192 authored Jun 14, 2024
1 parent 6798b74 commit 3864bcf
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions tests/upgrade/raycluster_sdk_upgrade_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,19 +82,18 @@ def run_mnist_raycluster_sdk_oauth(self):
class TestMnistJobSubmit:
def setup_method(self):
initialize_kubernetes_client(self)
self.namespace = namespace
self.cluster = get_cluster("mnist", self.namespace)
if not self.cluster:
raise RuntimeError("TestRayClusterUp needs to be run before this test")

def test_mnist_job_submission(self):
auth = TokenAuthentication(
token=run_oc_command(["whoami", "--show-token=true"]),
server=run_oc_command(["whoami", "--show-server=true"]),
skip_tls=True,
)
auth.login()
self.namespace = namespace
self.cluster = get_cluster("mnist", self.namespace)
if not self.cluster:
raise RuntimeError("TestRayClusterUp needs to be run before this test")

def test_mnist_job_submission(self):
self.assert_jobsubmit_withoutLogin(self.cluster)
self.assert_jobsubmit_withlogin(self.cluster)

Expand Down

0 comments on commit 3864bcf

Please sign in to comment.