edan_f0/tests/integration/test_wishlist.py

from django.test import TestCase

from oscar.apps.wishlists.models import WishList, WishListSharedEmail
from oscar.core.compat import get_user_model

User = get_user_model()


class TestAWishlist(TestCase):
    def test_can_generate_a_random_key(self):
        key = WishList.random_key(6)
        self.assertTrue(len(key) == 6)


class TestAPublicWishList(TestCase):
    def setUp(self):
        self.wishlist = WishList(visibility=WishList.PUBLIC, owner=User(id=1))

    def test_is_visible_to_anyone(self):
        user = User()
        self.assertTrue(self.wishlist.is_allowed_to_see(user))


class TestASharedWishList(TestCase):
    def setUp(self):
        user = User.objects.create(email="test1@example.com")
        self.wishlist = WishList.objects.create(visibility=WishList.SHARED, owner=user)

    def test_is_visible_to_anyone(self):
        user = User()
        self.assertFalse(self.wishlist.is_allowed_to_see(user))

    def test_is_visible_for_shared_email(self):
        WishListSharedEmail.objects.create(
            wishlist=self.wishlist, email="test2@example.com"
        )
        self.assertTrue(
            self.wishlist.is_allowed_to_see(User(email="test2@example.com"))
        )


class TestAPrivateWishList(TestCase):
    def setUp(self):
        self.owner = User(id=1)
        self.another_user = User(id=2)
        self.wishlist = WishList(owner=self.owner)

    def test_is_visible_only_to_its_owner(self):
        self.assertTrue(self.wishlist.is_allowed_to_see(self.owner))
        self.assertFalse(self.wishlist.is_allowed_to_see(self.another_user))

    def test_can_only_be_edited_by_its_owner(self):
        self.assertTrue(self.wishlist.is_allowed_to_edit(self.owner))
        self.assertFalse(self.wishlist.is_allowed_to_edit(self.another_user))