From 62f1c36f891cb4242b1c79d3a7d403400544255f Mon Sep 17 00:00:00 2001
From: Denis Bilenko <denis.bilenko@databricks.com>
Date: Tue, 4 Mar 2025 16:11:50 +0100
Subject: [PATCH] lint fix

---
 libs/patchwheel/patch_test.go                 |  49 ++++++++++++++++++
 libs/patchwheel/testdata/empty.zip            | Bin 0 -> 22 bytes
 .../my_test_code-0.0.1-py3-none-any.whl       | Bin 0 -> 1909 bytes
 3 files changed, 49 insertions(+)
 create mode 100644 libs/patchwheel/testdata/empty.zip
 create mode 100644 libs/patchwheel/testdata/my_test_code-0.0.1-py3-none-any.whl

diff --git a/libs/patchwheel/patch_test.go b/libs/patchwheel/patch_test.go
index ad05acbd8..5aeae890d 100644
--- a/libs/patchwheel/patch_test.go
+++ b/libs/patchwheel/patch_test.go
@@ -164,3 +164,52 @@ func TestPatchWheel(t *testing.T) {
 		})
 	}
 }
+
+var (
+	prebuiltWheel = "testdata/my_test_code-0.0.1-py3-none-any.whl"
+	emptyZip      = "testdata/empty.zip"
+)
+
+func TestPrebuilt(t *testing.T) {
+	tempDir := t.TempDir()
+	ctx := context.Background()
+	outname, err := PatchWheel(ctx, prebuiltWheel, tempDir)
+	require.NoError(t, err)
+
+	_, err = os.Stat(outname)
+	require.NoError(t, err)
+}
+
+func errPatchWheel(t *testing.T, name, out string) {
+	ctx := context.Background()
+	outname, err := PatchWheel(ctx, name, out)
+	assert.Error(t, err, "PatchWheel(%s, %s) expected to error", name, out)
+	assert.Empty(t, outname)
+}
+
+func TestError(t *testing.T) {
+	// empty name and dir
+	errPatchWheel(t, "", "")
+
+	// empty name
+	errPatchWheel(t, "", ".")
+
+	// file not found
+	errPatchWheel(t, "not-found.txt", ".")
+
+	// output directory not found
+	errPatchWheel(t, prebuiltWheel, "not-found/a/b/c")
+}
+
+func TestEmptyZip(t *testing.T) {
+	tempDir := t.TempDir()
+	errPatchWheel(t, emptyZip, tempDir)
+}
+
+func TestNonZip(t *testing.T) {
+	tempDir := t.TempDir()
+
+	_, err := os.Stat("patch.go")
+	require.NoError(t, err, "file must exist for this test")
+	errPatchWheel(t, "patch.go", tempDir)
+}
diff --git a/libs/patchwheel/testdata/empty.zip b/libs/patchwheel/testdata/empty.zip
new file mode 100644
index 0000000000000000000000000000000000000000..15cb0ecb3e219d1701294bfdf0fe3f5cb5d208e7
GIT binary patch
literal 22
NcmWIWW@Tf*000g10H*)|

literal 0
HcmV?d00001

diff --git a/libs/patchwheel/testdata/my_test_code-0.0.1-py3-none-any.whl b/libs/patchwheel/testdata/my_test_code-0.0.1-py3-none-any.whl
new file mode 100644
index 0000000000000000000000000000000000000000..14702281d87f6180c03799b19ec1ab6a10f95509
GIT binary patch
literal 1909
zcmWIWW@Zs#U|`^2*jp?fe!}|sR6QWi0Ei`kIJYvsB(=CCJ~=-nRX;vHGcU6wK3=b&
zvb$g3TUYDcne*NK;Tu8&j4l{oFb;Xt)$gyPd-8;S6UQ!<X)0=>scPxs+J%e^0VsBe
z?D2WA7-;q<>~`cPX6AwI2<_X=cSu2`t$f}ie!eZ;{1=vN?&DMITQaeELIL+u(>;5)
z$X+r2xBYYBxtn(dG^WH%T*S*7xY*=nSa&BQzw;A`e9N@juAK?%jFWvPw$0CZ@vCv+
zUC#34^olIq*6pQyn)$`I!%h`$?Gjn`C#t5}?EPHZ3!U1Q;tPwvd=RbT^uPH3*Yt>e
zM<1Tg)_I<`wpMwQR8ZT=$x|fUIW*1fG><Y?9N5H<@EA|1Ot|E^TRO{tp85#Hsz8h!
zc)A9926~2iDVfD3x|w-t`TD-DA&xGNA&#*<j(kl90<Pa}UG*<4(%XHhrM0umZSgwo
zo^7{`Zb&BW*gHW(E$@E)`}~I7)SIh>RX#;XTV4(hyusmi%+F9@wUD;JW4nu!?^$dy
zymf_ll8nmFA_hkt!=|h^YuC#bJFA^;`|kbeX>&<upiWfTjGi;@e;m0MeLZ->dRJ69
z$LIY8Z@pKnw6_ymom8!NL@#1rQkVUv_Vm}n6HeJ~oXB#=tVhD5W4+QnRF8HC7_Eu{
zdJu$_2zfNz!`0P?`@Dzm*^9i8G_yI#;EM5sBLB1Ax?Vab^*051c%9VI>(pmo8F<0q
zg0b0k;|ou?PMz1k_*Lf`uZFJI*^@q-f;2R>e4lc8`fLx<SYdMQ?CJB~SAEZ{e)5P3
zDRqG&!JReblP1tjT0pEz$Ze^4B}J9-1^JnICB=Fr6(!w%`ls}DJIs5YYH6L_e0FV*
zzNfBUr~f&B&#T@2=Y7wd>F(do6c`hOW@qmTJp+EAUA$0x@kV?}enEUrYFTOy$imY|
zRy;-bO4j6ipW_%A7#x@w7?cTlCCJs;Kggx^!s((W1BTXzwewju6GS3P6a`)N*1GB}
zo6R$|BhW*GGw|1!x5sAg%8Y;T`}~}=1Cuy9zwc4$mk4%KJyvC%@hBo<^2^E7%>*5v
zoVe}L_+0g#4p+3F{uw5}TWgyqthzT(J<ZBXU#dWER`ohfjopvu_;87JD}9;#k)`p*
z!{2_rcKp}lLRR=DYe(>Rh{-dqHsOvazjxvS@7*ijT2DGQG1vS2$uXOsx><UkME+vV
z6WPjZcK^MZ8{TRqdu+DKR-1B3^UL>kEV}tyyM^QLgJ<_E<bKrEOg{Pa(f5`j;V5gN
z6fNW2XH(wa-cS|Dd}g7^r|r)ru4PZperLa$bMoDWXup{vyY`%lW`1keayS3rvyl0`
z+YI&`y#4je#yHmSo0np5pEc!-?Kov~@n5yuC$Z};6MvOIc>QLtFYDR9(i4JT8T(!p
zvsMQSZum7N^~wA&w+~NeY*}qm{HI*u>qBp=>s#9M1H2iTM3`}xmcTdygC&h13ON-Z
zm!0UQpqGabGZ+|_G^Svh0xunrO~F?9AWX3ZCLv_E;4F&JjYQAp2qV>iDH5-d;7pHh
z9(vY9n0JVfdFUAx-Dvb2h%nlWnSfJK@*}#r=vfeE?iR*gLgqp;WPmp-8%Ps75E=u0
Jsly850RXT?oKXM(

literal 0
HcmV?d00001