Update creat tenant factory (#409)

* create_tenant accepts a list of staff or a number

Author: yunnie

tests/factory_fixtures/tenant.py

@@ -18,10 +18,15 @@ def _tenant_attributes(staff=None):
 
 
 @pytest.fixture
-def create_tenant(tenant_attributes):
-    def _create_tenant():
+def create_tenant(tenant_attributes, create_join_staff):
+    def _create_tenant(staff=None):
+        if isinstance(staff, int):
+            staff = [create_join_staff().id for _ in range(staff)]
+        elif staff is None:
+            staff = []
+
         return TenantModel.create(
-            schema=TenantSchema, payload=tenant_attributes(staff=[])
+            schema=TenantSchema, payload=tenant_attributes(staff=staff)
         )
 
     yield _create_tenant

tests/unit/test_tenant.py

@@ -17,14 +17,13 @@ class TestTenantModel:
     def test_json(self, create_tenant, create_lease):
         tenant = create_tenant()
         lease = create_lease(tenant=tenant)
-
         assert tenant.json() == {
             "id": tenant.id,
             "firstName": tenant.firstName,
             "lastName": tenant.lastName,
             "fullName": "{} {}".format(tenant.firstName, tenant.lastName),
             "phone": tenant.phone,
-            "staff": tenant.staff,
+            "staff": tenant.staff.json(),
             "lease": lease.json(),
             "created_at": Time.format_date(tenant.created_at),
             "updated_at": Time.format_date(tenant.updated_at),
@@ -36,3 +35,13 @@ def test_json(self, create_tenant, create_lease):
 class TestTenantFactory:
     def test_create_tenant(self, create_tenant):
         assert create_tenant()
+
+    def test_create_tenant_with_staff_as_list(self, create_tenant, create_join_staff):
+        staff = [create_join_staff().id for _ in range(3)]
+        tenant = create_tenant(staff)
+        assert [join_staff.id for join_staff in tenant.staff] == staff
+
+    def test_create_tenant_with_staff_as_int(self, create_tenant, create_join_staff):
+        number_of_staff = 4
+        tenant = create_tenant(number_of_staff)
+        assert len(tenant.staff) == number_of_staff