Browse Source

Create organization in tests

David Cramer 10 years ago
parent
commit
fc4e619540
2 changed files with 4 additions and 4 deletions
  1. 2 2
      tests/integration/tests.py
  2. 2 2
      tests/sentry/coreapi/tests.py

+ 2 - 2
tests/integration/tests.py

@@ -15,7 +15,7 @@ from gzip import GzipFile
 from exam import fixture
 from raven import Client
 
-from sentry.models import Group, Event, Project, User
+from sentry.models import Group, Event, User
 from sentry.testutils import TestCase
 from sentry.testutils.helpers import get_auth_header
 from sentry.utils.compat import StringIO
@@ -76,7 +76,7 @@ class RavenIntegrationTest(TestCase):
     """
     def setUp(self):
         self.user = User.objects.create(username='coreapi')
-        self.project = Project.objects.create(owner=self.user, name='Foo', slug='bar')
+        self.project = self.create_project(owner=self.user, slug='bar')
         self.pm = self.project.team.member_set.get_or_create(user=self.user)[0]
         self.pk = self.project.key_set.get_or_create(user=self.user)[0]
 

+ 2 - 2
tests/sentry/coreapi/tests.py

@@ -7,7 +7,7 @@ import mock
 from datetime import datetime
 from uuid import UUID
 
-from sentry.models import Project, User
+from sentry.models import User
 from sentry.exceptions import InvalidTimestamp
 from sentry.coreapi import (
     extract_auth_vars, project_from_auth_vars, APIForbidden, ensure_has_ip,
@@ -19,7 +19,7 @@ from sentry.testutils import TestCase
 class BaseAPITest(TestCase):
     def setUp(self):
         self.user = User.objects.create(username='coreapi')
-        self.project = Project.objects.create(owner=self.user, name='Foo', slug='bar')
+        self.project = self.create_project(owner=self.user, slug='bar')
         self.pm = self.project.team.member_set.get_or_create(user=self.user)[0]
         self.pk = self.project.key_set.get_or_create(user=self.user)[0]