228 lines
9.6 KiB
Python
Executable File
228 lines
9.6 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
import unittest
|
|
from unittest.mock import patch, call
|
|
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.keys import Keys
|
|
|
|
import jardin
|
|
|
|
class TestJardin(unittest.TestCase):
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_initialize_webdriver_calls_Firefox(self, mockFirefox):
|
|
jardin.initialize_webdriver()
|
|
|
|
mockFirefox.assert_called_once()
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_navigate_to_filegarden(self, mockFirefox):
|
|
jardin.navigate_to_filegarden(mockFirefox)
|
|
|
|
mockFirefox.get.assert_called_once_with("https://filegarden.com/login")
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_input_email_enters_email(self, mockFirefox):
|
|
mockEmail = "email@mail.com"
|
|
|
|
jardin.input_email(mockFirefox, mockEmail)
|
|
|
|
mockFirefox.find_element.assert_called_with(By.ID, "email")
|
|
mockFirefox.find_element.return_value.send_keys.assert_has_calls([
|
|
call(mockEmail),
|
|
call(Keys.RETURN)
|
|
])
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_click_password_button_clicks_button(self, mockFirefox):
|
|
jardin.click_password_button(mockFirefox)
|
|
|
|
mockFirefox.find_element.assert_called_with(By.XPATH, "/html/body/div[2]/div[3]/div[1]/form/div[1]/span/button[3]")
|
|
mockFirefox.find_element.return_value.click.assert_called_once()
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_input_password_enters_password(self, mockFirefox):
|
|
mockPassword = "p4$$w0rd"
|
|
|
|
jardin.input_password(mockFirefox, mockPassword)
|
|
|
|
mockFirefox.find_element.assert_called_with(By.ID, "password")
|
|
mockFirefox.find_element.return_value.send_keys.assert_has_calls([
|
|
call(mockPassword),
|
|
call(Keys.RETURN)
|
|
])
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
@patch("jardin.input_email")
|
|
@patch("jardin.click_password_button")
|
|
@patch("jardin.input_password")
|
|
def test_login_with_password_performs_login_steps(self, mockInputPassword, mockClickPasswordButton, mockInputEmail, mockFirefox):
|
|
mockEmail = "email@mail.com"
|
|
mockPassword = "p4$$w0rd"
|
|
|
|
jardin.login_with_password(mockFirefox, mockEmail, mockPassword)
|
|
|
|
mockInputEmail.assert_called_once_with(mockFirefox, mockEmail)
|
|
mockClickPasswordButton.assert_called_once_with(mockFirefox)
|
|
mockInputPassword.assert_called_once_with(mockFirefox, mockPassword)
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_click_go_to_your_garden_clicks_button(self, mockFirefox):
|
|
jardin.click_go_to_your_garden(mockFirefox)
|
|
|
|
mockFirefox.find_element.assert_called_with(By.LINK_TEXT, "GO TO YOUR GARDEN")
|
|
mockFirefox.find_element.return_value.click.assert_called_once()
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
@patch("jardin.bs4.BeautifulSoup")
|
|
def test_get_garden_filenames_creates_beautifulsoup_and_finds_names(self, mockBeautifulSoup, mockFirefox):
|
|
jardin.get_garden_filenames(mockFirefox)
|
|
|
|
mockBeautifulSoup.assert_called_once_with(mockFirefox.page_source, "lxml")
|
|
mockBeautifulSoup.return_value.find_all.assert_called_once_with(attrs={"class": "name"})
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_get_garden_filenames_returns_filenames_correctly(self, mockFirefox):
|
|
mockFirefox.page_source = """
|
|
<html lang="en">
|
|
<body class="mdc-typography">
|
|
<div id="items" class="items">
|
|
<a class="item typeFile selected" draggable="false" ondragstart="return false;" href="https://file.garden/fakeid/file1.jpg">
|
|
<div class="cell icon">
|
|
<button class="mdc-icon-button material-icons" type="button">image</button>
|
|
</div>
|
|
<div class="cell name" title="file1.jpg">file1.jpg</div>
|
|
<div class="cell size" title="10 B">10 B</div>
|
|
<div class="cell type" title="image/jpeg">image/jpeg</div>
|
|
<div class="cell date" title="Wed Aug 30 2023 13:04:10 GMT+0100 (British Summer Time)">Aug 30 2023 13:04:10</div>
|
|
</a>
|
|
<a class="item typeFile" draggable="false" ondragstart="return false;" href="https://file.garden/fakeid/file2.jpg">
|
|
<div class="cell icon">
|
|
<button class="mdc-icon-button material-icons" type="button">image</button>
|
|
</div>
|
|
<div class="cell name" title="file2.jpg">file2.jpg</div>
|
|
<div class="cell size" title="20 B">20 B</div>
|
|
<div class="cell type" title="image/jpeg">image/jpeg</div>
|
|
<div class="cell date" title="Wed Jun 29 2022 17:37:14 GMT+0100 (British Summer Time)">Jun 29 2022 17:37:14</div>
|
|
</a>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
filenames = jardin.get_garden_filenames(mockFirefox)
|
|
|
|
self.assertEqual(filenames, ["file1.jpg", "file2.jpg"])
|
|
|
|
@patch("jardin.os.listdir")
|
|
def test_get_local_filenames_gets_filenames_in_local_directory(self, mockListdir):
|
|
mockDirectory = "/home/luser/gardenfiles/"
|
|
mockFiles = ["file1", "file2", "file3"]
|
|
mockListdir.return_value = mockFiles
|
|
|
|
localFiles = jardin.get_local_filenames(mockDirectory)
|
|
|
|
mockListdir.assert_called_once_with(mockDirectory)
|
|
self.assertEqual(localFiles, mockFiles)
|
|
|
|
def test_get_missing_filenames_returns_local_filenames_without_garden_filenames(self):
|
|
mockLocalFilenames = ["file1.jpg", "file2.jpg", "file3.jpg", "file4.jpg", "file5.jpg"]
|
|
mockGardenFilenames = ["file2.jpg", "file4.jpg", "file6.jpg", "file8.jpg", "file10.jpg"]
|
|
|
|
missingFilenames = jardin.get_missing_filenames(mockLocalFilenames, mockGardenFilenames)
|
|
|
|
self.assertEqual(missingFilenames, ["file1.jpg", "file3.jpg", "file5.jpg"])
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_create_file_input_runs_correct_script(self, mockFirefox):
|
|
expectedScript = """const fileInput = document.createElement("input");
|
|
fileInput.id = "upload";
|
|
fileInput.type = "file";
|
|
fileInput.multiple = true;
|
|
|
|
fileInput.addEventListener("change", () => {
|
|
for (const file of fileInput.files) {
|
|
console.log(file); // Debug
|
|
// Miro.request(file); // What I'll probably actually use (or wrapper method)
|
|
//addFile(file); // Original (token not accessible from minified code)
|
|
}
|
|
fileInput.value = null;
|
|
});
|
|
|
|
document.body.appendChild(fileInput);
|
|
"""
|
|
|
|
jardin.create_file_input(mockFirefox)
|
|
|
|
mockFirefox.execute_script.assert_called_once_with(expectedScript)
|
|
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_get_user_id_returns_user_id_from_url(self, mockFirefox):
|
|
mockUserId = "73927475b259be67f8cea98f"
|
|
mockUrl = f"https://filegarden.com/users/{mockUserId}/garden/#"
|
|
mockFirefox.current_url = mockUrl
|
|
|
|
userId = jardin.get_user_id(mockFirefox)
|
|
|
|
self.assertEqual(userId, mockUserId)
|
|
|
|
@patch("jardin.get_user_id")
|
|
@patch("jardin.create_file_input")
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_upload_files_calls_correct_setup_methods(self, mockFirefox, mockCreateFileInput, mockGetUserId):
|
|
jardin.upload_files(mockFirefox, "", [])
|
|
|
|
mockCreateFileInput.assert_called_once_with(mockFirefox)
|
|
mockGetUserId.assert_called_once_with(mockFirefox)
|
|
|
|
@patch("jardin.get_user_id")
|
|
@patch("jardin.webdriver.Firefox")
|
|
def test_upload_files_calls_upload_file_once_per_file(self, mockFirefox, mockGetUserId):
|
|
mockGetUserId.return_value = "userid"
|
|
mockFiles = ["file1.jpg", "file2.jpg", "file3.jpg"]
|
|
mockDirectory = "/home/luser/gardenfiles/"
|
|
|
|
jardin.upload_files(mockFirefox, mockDirectory, mockFiles)
|
|
|
|
mockFirefox.find_element.return_value.send_keys.assert_has_calls([
|
|
call(mockDirectory + "file1.jpg"),
|
|
call(mockDirectory + "file2.jpg"),
|
|
call(mockDirectory + "file3.jpg")
|
|
])
|
|
|
|
@patch("jardin.initialize_webdriver")
|
|
@patch("jardin.navigate_to_filegarden")
|
|
@patch("jardin.login_with_password")
|
|
@patch("jardin.click_go_to_your_garden")
|
|
@patch("jardin.get_garden_filenames")
|
|
@patch("jardin.get_local_filenames")
|
|
@patch("jardin.get_missing_filenames")
|
|
@patch("jardin.upload_files")
|
|
def test_main_calls_methods_in_correct_order(self,
|
|
mockUploadFiles,
|
|
mockGetMissingFilenames,
|
|
mockGetLocalFilenames,
|
|
mockGetGardenFilenames,
|
|
mockClickGoToYourGarden,
|
|
mockLoginWithPassword,
|
|
mockNavigateToFilegarden,
|
|
mockInitializeWebdriver):
|
|
mockEmail = "email@mail.com"
|
|
mockPassword = "p4$$w0rd"
|
|
mockDirectory = "/home/luser/gardenfiles/"
|
|
|
|
jardin.main(mockEmail, mockPassword, mockDirectory)
|
|
|
|
mockInitializeWebdriver.assert_called_once()
|
|
mockNavigateToFilegarden.assert_called_once_with(mockInitializeWebdriver.return_value)
|
|
mockLoginWithPassword.assert_called_once_with(mockInitializeWebdriver.return_value, mockEmail, mockPassword)
|
|
mockClickGoToYourGarden.assert_called_once_with(mockInitializeWebdriver.return_value)
|
|
mockGetGardenFilenames.assert_called_once_with(mockInitializeWebdriver.return_value)
|
|
mockGetLocalFilenames.assert_called_once_with(mockDirectory)
|
|
mockGetMissingFilenames.assert_called_once_with(mockGetLocalFilenames.return_value, mockGetGardenFilenames.return_value)
|
|
mockUploadFiles.assert_called_once_with(mockInitializeWebdriver.return_value, mockDirectory, mockGetMissingFilenames.return_value)
|
|
#mockInitializeWebdriver.return_value.close.assert_called_once()
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|