From 3f22ac12f5d64c6d23017f903313f7d70232190d Mon Sep 17 00:00:00 2001
From: freearhey <7253922+freearhey@users.noreply.github.com>
Date: Fri, 24 Jan 2025 20:00:25 +0300
Subject: [PATCH] Update grab.test.ts

---
 tests/commands/epg/grab.test.ts | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/tests/commands/epg/grab.test.ts b/tests/commands/epg/grab.test.ts
index 8d40f16f..648d5057 100644
--- a/tests/commands/epg/grab.test.ts
+++ b/tests/commands/epg/grab.test.ts
@@ -97,6 +97,30 @@ describe('epg:grab', () => {
 
     expect(stdout).toContain('ERR: Connection timeout')
   })
+
+  it('can grab epg via https proxy', () => {
+    const cmd = `${ENV_VAR} npm run grab --- --site=example.com --proxy=https://bob:123456@proxy.com:1234 --output="${path.resolve(
+      'tests/__data__/output/guide.xml'
+    )}"`
+    const stdout = execSync(cmd, { encoding: 'utf8' })
+    if (process.env.DEBUG === 'true') console.log(cmd, stdout)
+
+    expect(content('tests/__data__/output/guide.xml')).toEqual(
+      content('tests/__data__/expected/guide2.xml')
+    )
+  })
+
+  it('can grab epg via socks5 proxy', () => {
+    const cmd = `${ENV_VAR} npm run grab --- --site=example.com --proxy=socks5://bob:123456@proxy.com:1234 --output="${path.resolve(
+      'tests/__data__/output/guide.xml'
+    )}"`
+    const stdout = execSync(cmd, { encoding: 'utf8' })
+    if (process.env.DEBUG === 'true') console.log(cmd, stdout)
+
+    expect(content('tests/__data__/output/guide.xml')).toEqual(
+      content('tests/__data__/expected/guide2.xml')
+    )
+  })
 })
 
 function content(filepath: string) {