def test_retrieve_tenant_returns_none_when_no_tenant_found(self): self.ds_handler.find_one = MagicMock(return_value=None) with patch('meniscus.data.model.tenant_util._db_handler', self.ds_handler): tenant = tenant_util.retrieve_tenant(self.tenant_id) self.ds_handler.find_one.assert_called_once_with( 'tenant', {'tenant_id': self.tenant_id}) self.assertIsNone(tenant)
def test_retrieve_tenant_returns_tenant_obj(self): self.ds_handler.find_one = MagicMock(return_value=self.tenant_dict) with patch('meniscus.data.model.tenant_util._db_handler', self.ds_handler): tenant = tenant_util.retrieve_tenant(self.tenant_id) self.ds_handler.find_one.assert_called_once_with( 'tenant', {'tenant_id': self.tenant_id}) self.assertIsInstance(tenant, Tenant)