def test_getDirectorySize(self): ''' tests the output of getDirectorySize ''' resource = XROOTStorage('storageName', self.parameterDict) statusMock = xrootStatusMock() statusMock.makeOk() statInfoMock = xrootStatInfoMock() statInfoMock.makeDir() updateStatMockReferences(statusMock, statInfoMock) statDir1 = xrootStatInfoMock() statDir1.makeDir() statDir1.size = 1 dir1 = xrootListEntryMock("dir1", "host", statDir1) statFile1 = xrootStatInfoMock() statFile1.makeFile() statFile1.size = 4 file1 = xrootListEntryMock("file1", "host", statFile1) directoryListMock = xrootDirectoryListMock("parent", [dir1, file1]) mocked_xrootclient.dirlist.return_value = (statusMock, directoryListMock) # We have 1 file (size4) and 1 subdir in the directory res = resource.getDirectorySize('A') self.assertEqual(True, res['OK']) self.assertEqual(1, res['Value']['Successful']["A"]["Files"]) self.assertEqual(1, res['Value']['Successful']["A"]["SubDirs"]) self.assertEqual(4, res['Value']['Successful']["A"]["Size"]) self.assertEqual({}, res['Value']['Failed'])