示例#1
0
  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'])