#!/usr/bin/env python """Test the vfs gui interface.""" from unittest import mock from absl import app from grr_response_server.gui import api_call_router_with_approval_checks from grr_response_server.gui import gui_test_lib from grr_response_server.gui.api_plugins import vfs as api_vfs from grr.test_lib import fixture_test_lib from grr.test_lib import test_lib class VFSViewTest(gui_test_lib.GRRSeleniumTest): def setUp(self): super().setUp() # Prepare our fixture. self.client_id = "C.0000000000000001" with test_lib.FakeTime(test_lib.FIXED_TIME): fixture_test_lib.ClientFixture(self.client_id) gui_test_lib.CreateFileVersions(self.client_id) self.RequestAndGrantClientApproval("C.0000000000000001") def testUnicodeContentIsShownInTree(self): # Open VFS view for client 1 on a specific location. self.Open("/legacy#clients/C.0000000000000001/vfs/fs/os/c/Downloads/") # Click on the file containing unicode characters. self.Click('css=tr:contains("中.txt")') # Then click on the "Download" tab. self.Click("css=li[heading=Download]:not(.disabled)") self.WaitUntil(self.IsTextPresent, "中国新闻网新闻中.txt") def testFolderPathCanContainUnicodeCharacters(self): # Open VFS view for client 1 on a location containing unicode characters. self.Open( "/legacy#/clients/C.0000000000000001/vfs/fs/os/c/中国新闻网新闻中/" ) # Check that the correct file is listed. self.WaitUntil(self.IsElementPresent, 'css=tr:contains("bzcmp")') def testFolderPathWithUnicodeCharactersCanBeClicked(self): dirpath = "fs/os/home/user/foo/a看a" filepath = f"{dirpath}/bar/baz.txt" gui_test_lib.CreateFileVersion(self.client_id, filepath, b"foobar") self.Open(f"/#/clients/{self.client_id}/vfs/{dirpath}") self.Click('css=li > a:contains("a看a")') self.Click('css=li > a:contains("bar")') self.WaitUntil(self.IsElementPresent, 'css=tr:contains("baz.txt")') def testUrlSensitiveCharactersAreShownInTree(self): gui_test_lib.CreateFileVersion( self.client_id, "fs/os/c/foo?bar&oh/a&=?b.txt", "Hello World".encode("utf-8"), timestamp=gui_test_lib.TIME_1, ) # Open VFS view for client 1 on a specific location. self.Open("/legacy#/clients/C.0000000000000001/vfs/fs/os/c/") # Wait until the folder gets selected and its information displayed in # the details pane. self.WaitUntil( self.IsElementPresent, "css=grr-file-details:contains('VFSDirectory')" ) # Click on the "foo?bar&oh" subfolder. self.Click("css=#_fs-os-c-foo_3Fbar_26oh a:visible") # Some more unicode testing. self.Click('css=tr:contains("a&=?b.txt")') self.Click("css=li[heading=Download]") self.WaitUntil( self.IsElementPresent, "css=grr-file-details:contains('a&=?b.txt')" ) # Test the text viewer. self.Click("css=li[heading=TextView]") self.WaitUntilContains("Hello World", self.GetText, "css=div.monospace pre") def testFolderPathCanContainUrlSensitiveCharacters(self): gui_test_lib.CreateFileVersion( self.client_id, "fs/os/c/foo?bar&oh/a&=?b.txt", "Hello World".encode("utf-8"), timestamp=gui_test_lib.TIME_1, ) # Open VFS view for client 1 on a location containing unicode characters. self.Open( "/legacy#c=C.0000000000000001&main=VirtualFileSystemView&t=_fs-os-c" "-foo_3Fbar_26oh" ) # Check that the correct file is listed. self.WaitUntil(self.IsElementPresent, 'css=tr:contains("a&=?b.txt")') def testDoubleClickGoesInsideDirectory(self): """Tests that double click in FileTable goes inside the directory.""" self.Open("/legacy") self.Type("client_query", "C.0000000000000001") self.Click("client_query_submit") self.WaitUntilEqual( "C.0000000000000001", self.GetText, "css=span[type=subject]" ) # Choose client 1 and go to 'Browse Virtual Filesystem' self.Click("css=td:contains('0001')") self.Click("css=a[grrtarget='client.vfs']") self.Click("link=fs") # Now click on "os" inside the table. Tree shouldn't get updated, self.Click("css=td:contains('os')") # Now double click on "os". self.DoubleClick("css=td:contains('os')") # Now we should be inside the folder, and the tree should open. self.WaitUntil(self.IsElementPresent, "css=#_fs-os-c i.jstree-icon") # Check that breadcrumbs got updated. self.WaitUntil( self.IsElementPresent, "css=#content_rightPane .breadcrumb li:contains('os')", ) @mock.patch.object( api_call_router_with_approval_checks.ApiCallRouterWithApprovalChecks, "GetVfsFilesArchive", return_value=api_vfs.ApiGetVfsFilesArchiveHandler(), ) def testClickingOnDownloadCurrentFolderButtonStartsDownload( self, mock_method ): # Open VFS view for client 1 on a specific location. self.Open( "/legacy#c=C.0000000000000001&main=VirtualFileSystemView" "&t=_fs-os-c-proc" ) self.Click("css=grr-vfs-files-archive-button") self.Click("css=a[name=downloadCurrentFolder]") # Mock method will be called twice: once for HEAD request (to check # permissions) and once for GET request. self.WaitUntil(lambda: mock_method.call_count) mock_method.assert_called_with( api_vfs.ApiGetVfsFilesArchiveArgs( client_id="C.0000000000000001", file_path="fs/os/c/proc" ), context=mock.ANY, ) @mock.patch.object( api_call_router_with_approval_checks.ApiCallRouterWithApprovalChecks, "GetVfsFilesArchive", return_value=api_vfs.ApiGetVfsFilesArchiveHandler(), ) def testClickingOnDownloadEverythingButtonStartsDownload(self, mock_method): # Open VFS view for client 1 on a specific location. self.Open( "/legacy#c=C.0000000000000001&main=VirtualFileSystemView" "&t=_fs-os-c-proc" ) self.Click("css=grr-vfs-files-archive-button") self.Click("css=a[name=downloadEverything]:not([disabled])") # Mock method will be called twice: once for HEAD request (to check # permissions) and once for GET request. self.WaitUntil(lambda: mock_method.call_count) mock_method.assert_called_with( api_vfs.ApiGetVfsFilesArchiveArgs(client_id="C.0000000000000001"), context=mock.ANY, ) if __name__ == "__main__": app.run(test_lib.main)