Пример #1
0
 def test_visibility(self):
     document = Document('title1', 'desc1', 1, ['1.pdf', '2.pdf'], 'pdf')
     self.assertFalse(document.is_public())
     document.make_public()
     self.assertTrue(document.is_public())
     document.make_private()
     self.assertFalse(document.is_public())
Пример #2
0
    def test_export_documents(self):
        try:
            shutil.rmtree('/tmp/test_repo3')
            shutil.rmtree('/tmp/exported_documents')
        except OSError as error:
            pass

        for test_file in ['part1.pdf', 'part2.pdf', 'data.doc']:
            file_to_path = reduce(os.path.join,
                                  [SAMPLE_DIR_PATH, 'importable', test_file])
            file_from_path = reduce(os.path.join, [
                SAMPLE_DIR_PATH, 'importable', 'export_files_for_test',
                test_file
            ])
            if not os.path.exists(file_to_path):
                copy2(file_from_path, file_to_path)

        repository = Repository('Empty', '/tmp/test_repo3')
        alice = User('Alice', 'Smith', date(1980, 10, 10), '*****@*****.**',
                     '****')
        bob = User('Bob', 'Marker', date(1970, 11, 11), '*****@*****.**', '****')
        alice_id = repository._user_manager.add_user(alice)
        bob_id = repository._user_manager.add_user(bob)
        first_document = Document(
            'Some important doc', 'Contains various documentations', alice_id,
            [
                SAMPLE_DIR_PATH + '/importable/part1.pdf',
                SAMPLE_DIR_PATH + '/importable/part2.pdf'
            ], 'pdf')
        first_document.make_public()
        first_document.change_state('pending')
        first_document.change_state('accepted')
        second_document = Document('Data report', 'Figures and graphs mainly',
                                   bob_id,
                                   [SAMPLE_DIR_PATH + '/importable/data.doc'],
                                   'doc')
        second_document.make_public()
        second_document.change_state('pending')
        second_document.change_state('accepted')
        first_id = repository._document_manager.add_document(first_document)
        second_id = repository._document_manager.add_document(second_document)
        repository.export_documents([first_id, second_id],
                                    '/tmp/exported_documents')

        self.assertTrue(os.path.exists('/tmp/exported_documents/part1.pdf'))
        self.assertTrue(os.path.exists('/tmp/exported_documents/part2.pdf'))
        self.assertTrue(os.path.exists('/tmp/exported_documents/data.doc'))
        self.assertTrue(os.path.exists('/tmp/exported_documents/1.edd'))
        self.assertTrue(os.path.exists('/tmp/exported_documents/2.edd'))

        with open('/tmp/exported_documents/1.edd') as edd_file:
            lines = edd_file.readlines()
            self.assertIn('[document]\n', lines)
            self.assertIn('title=Some important doc\n', lines)
            self.assertIn('description=Contains various documentations\n',
                          lines)
            self.assertIn('author=Alice Smith\n', lines)
            self.assertIn("files=['part1.pdf', 'part2.pdf']\n", lines)
            self.assertIn('doc_format=pdf\n', lines)

        with open('/tmp/exported_documents/2.edd') as edd_file:
            lines = edd_file.readlines()
            self.assertIn('[document]\n', lines)
            self.assertIn('title=Data report\n', lines)
            self.assertIn('description=Figures and graphs mainly\n', lines)
            self.assertIn('author=Bob Marker\n', lines)
            self.assertIn("files=['data.doc']\n", lines)
            self.assertIn('doc_format=doc\n', lines)

        shutil.rmtree('/tmp/exported_documents')
        shutil.rmtree('/tmp/test_repo3')