def test_current_tenant_value_single(self):
from .models import Project, Account
projects = self.projects
account = projects[0].account
set_current_tenant(account)
self.assertEqual(get_current_tenant_value(), account.id)
unset_current_tenant()
def test_current_tenant_value_queryset(self):
from .models import Project, Account
projects = self.projects
accounts = Account.objects.all().order_by('id')
set_current_tenant(accounts)
value = get_current_tenant_value()
self.assertEqual(get_tenant_filters(Project),
{'account_id__in': list(accounts.values_list('id', flat=True))})
unset_current_tenant()
def test_current_tenant_value_queryset(self):
from .models import Project, Account
projects = self.projects
accounts = Account.objects.all().order_by("id")
set_current_tenant(accounts)
value = get_current_tenant_value()
self.assertTrue(isinstance(value, list))
self.assertEqual(value, accounts.values_list("id", flat=True))
unset_current_tenant()
def test_current_tenant_value_list(self):
from .models import Project, Account
projects = self.projects
accounts = [projects[0].account, projects[1].account]
set_current_tenant(accounts)
value = get_current_tenant_value()
self.assertTrue(isinstance(value, list))
self.assertEqual(value, [accounts[0].id, accounts[1].id])
unset_current_tenant()