switch space method signature change and test

cookbook/tests/other/test_permission_helper.py

@@ -6,8 +6,8 @@ from django_scopes import scopes_disabled
 
 from cookbook.forms import ImportExportBase
 from cookbook.helper.ingredient_parser import IngredientParser
-from cookbook.helper.permission_helper import has_group_permission
-from cookbook.models import ExportLog, UserSpace
+from cookbook.helper.permission_helper import has_group_permission, is_space_owner, switch_user_active_space
+from cookbook.models import ExportLog, UserSpace, Food
 
 
 def test_has_group_permission(u1_s1, a_u, space_2):
@@ -36,3 +36,39 @@ def test_has_group_permission(u1_s1, a_u, space_2):
         assert not has_group_permission(auth.get_user(a_u), ('guest',))
         assert not has_group_permission(auth.get_user(a_u), ('user',))
         assert not has_group_permission(auth.get_user(a_u), ('admin',))
+
+
+def test_is_space_owner(u1_s1, u2_s1, space_1, space_2):
+    with scopes_disabled():
+        f = Food.objects.create(name='Test', space=space_1)
+        space_1.created_by = auth.get_user(u1_s1)
+        space_1.save()
+
+        assert is_space_owner(auth.get_user(u1_s1), f)
+        assert is_space_owner(space_1.created_by, f)
+        assert not is_space_owner(auth.get_user(u2_s1), f)
+
+        f.space = space_2
+        f.save()
+
+        assert not is_space_owner(auth.get_user(u1_s1), f)
+        assert not is_space_owner(space_1.created_by, f)
+        assert not is_space_owner(auth.get_user(u2_s1), f)
+
+
+def test_switch_user_active_space(u1_s1, u1_s2, space_1, space_2):
+    with scopes_disabled():
+        # can switch to already active space
+        assert switch_user_active_space(auth.get_user(u1_s1), space_1) == auth.get_user(u1_s1).userspace_set.filter(active=True).get()
+
+        # cannot switch to a space the user does not belong to
+        assert switch_user_active_space(auth.get_user(u1_s1), space_2) is None
+
+        # can join another space and be member of two spaces
+        us = UserSpace.objects.create(user=auth.get_user(u1_s1), space=space_2, active=False)
+        assert len(auth.get_user(u1_s1).userspace_set.all()) == 2
+
+        # can switch into newly created space
+        assert switch_user_active_space(auth.get_user(u1_s1), space_2) == us
+
+