adds session_file name customization

This commit is contained in:
msramalho
2022-11-03 16:00:58 +00:00
parent 3a8a3f54c0
commit d1b27bef1d
5 changed files with 37 additions and 3 deletions

View File

@@ -14,6 +14,18 @@ def test_login_fail():
VkScraper("invalid", "combination")
def test_login_custom_file():
session_filename = "test-session.json"
VkScraper(
os.environ["VK_USERNAME"],
os.environ["VK_PASSWORD"],
os.environ.get("VK_TOKEN"),
session_file=session_filename,
)
assert os.path.isfile(session_filename)
os.unlink(session_filename)
def test_login_success():
global vks
vks = VkScraper(