Skip to content

Commit 3864bcf

Browse files
authored
update Token Authentication for TestMnistJobSubmit before step get_cluster (#561)
1 parent 6798b74 commit 3864bcf

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

Diff for: tests/upgrade/raycluster_sdk_upgrade_test.py

+5-6
Original file line numberDiff line numberDiff line change
@@ -82,19 +82,18 @@ def run_mnist_raycluster_sdk_oauth(self):
8282
class TestMnistJobSubmit:
8383
def setup_method(self):
8484
initialize_kubernetes_client(self)
85-
self.namespace = namespace
86-
self.cluster = get_cluster("mnist", self.namespace)
87-
if not self.cluster:
88-
raise RuntimeError("TestRayClusterUp needs to be run before this test")
89-
90-
def test_mnist_job_submission(self):
9185
auth = TokenAuthentication(
9286
token=run_oc_command(["whoami", "--show-token=true"]),
9387
server=run_oc_command(["whoami", "--show-server=true"]),
9488
skip_tls=True,
9589
)
9690
auth.login()
91+
self.namespace = namespace
92+
self.cluster = get_cluster("mnist", self.namespace)
93+
if not self.cluster:
94+
raise RuntimeError("TestRayClusterUp needs to be run before this test")
9795

96+
def test_mnist_job_submission(self):
9897
self.assert_jobsubmit_withoutLogin(self.cluster)
9998
self.assert_jobsubmit_withlogin(self.cluster)
10099

0 commit comments

Comments
 (0)