示例#1
0
def test_store_file(monkeypatch, tmp_path):
    def fake_store_file(_ignored, _ignored2):
        pass

    note = Note()
    note.conversion_settings.export_folder = tmp_path
    attachment_id = '1234'
    image_attachment = sn_attachment.FileNSAttachment(note, attachment_id)
    image_attachment._json = {
        'attachment':
            {
                '1234':
                    {'md5': 'abcd', 'name': 'not_my_file.pdf'}
            }
    }

    note.parent_notebook.attachment_md5_file_name_dict = {'0987': "my_file.pdf"}

    monkeypatch.setattr(file_writer, 'store_file', fake_store_file)
    image_attachment.create_file_name()
    image_attachment.generate_relative_path_to_notebook()
    image_attachment.generate_absolute_path()
    assert image_attachment.file_name == Path('my_name.png')
    assert image_attachment._full_path == Path(tmp_path, 'notebook_folder', 'attachments', 'my_name.png')

    image_attachment.store_file()

    assert len(note.parent_notebook.attachment_md5_file_name_dict) == 2
示例#2
0
def test_FileNSAttachment_get_content_to_save():
    note = Note()
    attachment_id = '1234'
    file_attachment = sn_attachment.FileNSAttachment(note, attachment_id)
    result = file_attachment.get_content_to_save()

    assert result == 'file name in nsx'
示例#3
0
def test_FileNSAttachment_create_file_name(raw_name, expected):
    note = Note()
    attachment_id = '1234'
    file_attachment = sn_attachment.FileNSAttachment(note, attachment_id)

    file_attachment._name = raw_name
    file_attachment.create_file_name()

    assert file_attachment.file_name == Path(expected)
示例#4
0
def test_is_duplicate_file_not_a_duplicate():
    note = Note()
    attachment_id = '1234'
    image_attachment = sn_attachment.FileNSAttachment(note, attachment_id)

    image_attachment._name = 'my_file.pdf'
    note.parent_notebook.attachment_md5_file_name_dict = {'abcd': "zxyz"}

    assert image_attachment.is_duplicate_file() is False
示例#5
0
def test_test_change_file_name_file_attachment_good_file_name():
    note = Note()
    attachment_id = '1234'
    image_attachment = sn_attachment.FileNSAttachment(note, attachment_id)

    image_attachment._name = 'my_file.pdf'
    image_attachment.create_file_name()

    assert image_attachment.file_name == Path('my_file.pdf')
示例#6
0
def test_FileNSAttachment_create_html_link():
    note = Note()
    attachment_id = '1234'
    file_attachment = sn_attachment.FileNSAttachment(note, attachment_id)

    file_attachment._file_name = Path('my_file.png')
    file_attachment._path_relative_to_notebook = Path('attachments/my_file.png')
    file_attachment.create_html_link()

    assert file_attachment.html_link == '<a href="attachments/my_file.png">my_file.png</a>'
示例#7
0
def test_test_change_file_name_file_attachment_no_extension_on_file_name_and_not_recognised(mocker):
    note = Note()
    attachment_id = '1234'
    image_attachment = sn_attachment.FileNSAttachment(note, attachment_id)

    image_attachment._name = 'my_file'
    mocker.patch('zip_file_reader.read_binary_file', return_value=b'1234')
    mocker.patch('helper_functions.file_extension_from_bytes', return_value=None)
    image_attachment.create_file_name()

    assert image_attachment.file_name == Path('my_file')
示例#8
0
def test_is_duplicate_file_is_a_duplicate():
    note = Note()
    attachment_id = '1234'
    image_attachment = sn_attachment.FileNSAttachment(note, attachment_id)
    image_attachment._json = {
        'attachment':
            {
                '1234':
                    {'md5': '0987', 'name': 'my_file.pdf'}
            }
    }

    note.parent_notebook.attachment_md5_file_name_dict = {'0987': "my_file.pdf"}

    assert image_attachment.is_duplicate_file() is True
示例#9
0
def test_store_file_will_not_store_as_duplicate_md5_and_name(monkeypatch):
    def fake_store_file(_ignored, _ignored2):
        pass

    note = Note()
    attachment_id = '1234'
    image_attachment = sn_attachment.FileNSAttachment(note, attachment_id)
    image_attachment._json = {
        'attachment':
            {
                '1234':
                    {'md5': '0987', 'name': 'my_file.pdf'}
            }
    }

    note.parent_notebook.attachment_md5_file_name_dict = {'0987': "my_file.pdf"}
    monkeypatch.setattr(file_writer, 'store_file', fake_store_file)
    image_attachment.store_file()

    assert len(note.parent_notebook.attachment_md5_file_name_dict) == 1
示例#10
0
    def create_attachments(self):
        if self._attachments_json is None:
            self.logger.warning(
                f'Note - {self._title} - Has Null set for attachments. '
                f'There may be a sync issues between desktop and web version of Note Station.'
            )
            return 0, 0

        for attachment_id in self._attachments_json:
            if self._attachments_json[attachment_id]['type'].startswith('image') \
                    and \
                    self._attachments_json[attachment_id].get('ref'):
                self._attachments[
                    attachment_id] = sn_attachment.ImageNSAttachment(
                        self, attachment_id)
                self._image_count += 1
            else:
                self._attachments[
                    attachment_id] = sn_attachment.FileNSAttachment(
                        self, attachment_id)
                self._attachment_count += 1

        return self._image_count, self._attachment_count