from django.test import TestCase

from bookmarks.models import Bookmark


class BookmarkTestCase(TestCase):

    def test_bookmark_resolved_title(self):
        bookmark = Bookmark(title='Custom title', website_title='Website title', url='https://example.com')
        self.assertEqual(bookmark.resolved_title, 'Custom title')

        bookmark = Bookmark(title='', website_title='Website title', url='https://example.com')
        self.assertEqual(bookmark.resolved_title, 'Website title')

        bookmark = Bookmark(title='', website_title='', url='https://example.com')
        self.assertEqual(bookmark.resolved_title, 'https://example.com')