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")