@@ -120,15 +120,18 @@ class SparkPipelinesSuite extends SparkSubmitTestUtils with BeforeAndAfterEach {
120
120
test(" spark.api.mode arg" ) {
121
121
var args = Array (" --conf" , " spark.api.mode=classic" )
122
122
intercept[SparkUserAppException ] {
123
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" )
123
+ SparkPipelines .constructSparkSubmitArgs(
124
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args)
124
125
}
125
126
args = Array (" -c" , " spark.api.mode=classic" )
126
127
intercept[SparkUserAppException ] {
127
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" )
128
+ SparkPipelines .constructSparkSubmitArgs(
129
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args)
128
130
}
129
131
args = Array (" --conf" , " spark.api.mode=CONNECT" )
130
132
assert(
131
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" ) ==
133
+ SparkPipelines .constructSparkSubmitArgs(
134
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args) ==
132
135
Seq (
133
136
" --conf" ,
134
137
" spark.api.mode=connect" ,
@@ -139,7 +142,8 @@ class SparkPipelinesSuite extends SparkSubmitTestUtils with BeforeAndAfterEach {
139
142
)
140
143
args = Array (" --conf" , " spark.api.mode=CoNNect" )
141
144
assert(
142
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" ) ==
145
+ SparkPipelines .constructSparkSubmitArgs(
146
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args) ==
143
147
Seq (
144
148
" --conf" ,
145
149
" spark.api.mode=connect" ,
@@ -150,7 +154,8 @@ class SparkPipelinesSuite extends SparkSubmitTestUtils with BeforeAndAfterEach {
150
154
)
151
155
args = Array (" --conf" , " spark.api.mode=connect" )
152
156
assert(
153
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" ) ==
157
+ SparkPipelines .constructSparkSubmitArgs(
158
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args) ==
154
159
Seq (
155
160
" --conf" ,
156
161
" spark.api.mode=connect" ,
@@ -161,7 +166,8 @@ class SparkPipelinesSuite extends SparkSubmitTestUtils with BeforeAndAfterEach {
161
166
)
162
167
args = Array (" --conf" , " spark.api.mode= connect" )
163
168
assert(
164
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" ) ==
169
+ SparkPipelines .constructSparkSubmitArgs(
170
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args) ==
165
171
Seq (
166
172
" --conf" ,
167
173
" spark.api.mode=connect" ,
@@ -172,7 +178,8 @@ class SparkPipelinesSuite extends SparkSubmitTestUtils with BeforeAndAfterEach {
172
178
)
173
179
args = Array (" -c" , " spark.api.mode=connect" )
174
180
assert(
175
- SparkPipelines .constructSparkSubmitArgs(args, sparkHome = " abc" ) ==
181
+ SparkPipelines .constructSparkSubmitArgs(
182
+ pipelinesCliFile = " abc/python/pyspark/pipelines/cli.py" , args = args) ==
176
183
Seq (
177
184
" --conf" ,
178
185
" spark.api.mode=connect" ,
0 commit comments