43 lines
819 B
Python
43 lines
819 B
Python
import pytest
|
|
import unittest
|
|
|
|
from django.test import TestCase
|
|
|
|
from access.models.organization import Organization
|
|
|
|
from app.tests.abstract.models import TenancyModel
|
|
|
|
from itim.models.clusters import Cluster
|
|
|
|
|
|
|
|
@pytest.mark.django_db
|
|
class ClusterModel(
|
|
TestCase,
|
|
TenancyModel
|
|
):
|
|
|
|
model = Cluster
|
|
|
|
@classmethod
|
|
def setUpTestData(self):
|
|
"""Setup Test
|
|
|
|
1. Create an organization for user and item
|
|
2. Create an item
|
|
|
|
"""
|
|
|
|
self.organization = Organization.objects.create(name='test_org')
|
|
|
|
|
|
self.item = self.model.objects.create(
|
|
organization = self.organization,
|
|
name = 'one',
|
|
)
|
|
|
|
self.second_item = self.model.objects.create(
|
|
organization = self.organization,
|
|
name = 'one_two',
|
|
)
|