Skip to content

Commit

Permalink
Upgrading to factory-boy 2.5: updating tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
matagus committed Apr 19, 2015
1 parent a3dc6e1 commit a6c9f1e
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 12 deletions.
2 changes: 1 addition & 1 deletion .travis.yml
Expand Up @@ -18,7 +18,7 @@ install:
- pip install BeautifulSoup
- pip install coverage==3.6
- pip install python-coveralls==2.4.0
- pip install factory-boy
- pip install factory-boy==2.5.0
- pip install mock
- pip install django-discover-runner
- export PYTHONPATH=.
Expand Down
33 changes: 22 additions & 11 deletions planet/tests/factories.py
Expand Up @@ -11,32 +11,37 @@


class SiteFactory(factory.DjangoModelFactory):
FACTORY_FOR = Site
class Meta:
model = Site

domain = factory.Sequence(lambda n: u'example-site-{}.com'.format(n))


class BlogFactory(factory.DjangoModelFactory):
FACTORY_FOR = Blog
class Meta:
model = Blog

title = factory.Sequence(lambda n: u'blog-{}'.format(n))
url = factory.LazyAttribute(lambda obj: u'http://{}.blogspot.com/'.format(obj.title))


class GeneratorFactory(factory.DjangoModelFactory):
FACTORY_FOR = Generator
class Meta:
model = Generator

name = factory.Sequence(lambda n: u'generator-{}'.format(n))


class CategoryFactory(factory.DjangoModelFactory):
FACTORY_FOR = Category
class Meta:
model = Category

title = factory.Sequence(lambda n: u'Category #{}'.format(n))


class FeedFactory(factory.DjangoModelFactory):
FACTORY_FOR = Feed
class Meta:
model = Feed

title = factory.Sequence(lambda n: u'Feed-{}'.format(n))
blog = factory.SubFactory(BlogFactory)
Expand All @@ -47,14 +52,16 @@ class FeedFactory(factory.DjangoModelFactory):


class AuthorFactory(factory.DjangoModelFactory):
FACTORY_FOR = Author
class Meta:
model = Author

name = factory.Sequence(lambda n: u'Author #{}'.format(n))
email = factory.LazyAttribute(lambda obj: u'{}@gmail.com'.format(obj.name))


class PostFactory(factory.DjangoModelFactory):
FACTORY_FOR = Post
class Meta:
model = Post

title = factory.Sequence(lambda n: u'Post Title #{}'.format(n))
feed = factory.SubFactory(FeedFactory)
Expand All @@ -75,14 +82,16 @@ def authors(self, create, extracted, **kwargs):


class PostAuthorDataFactory(factory.DjangoModelFactory):
FACTORY_FOR = PostAuthorData
class Meta:
model = PostAuthorData

post = factory.SubFactory(PostFactory)
author = factory.SubFactory(AuthorFactory)


class PostLinkFactory(factory.DjangoModelFactory):
FACTORY_FOR = PostLink
class Meta:
model = PostLink

title = factory.Sequence(lambda n: u'post-link-{}'.format(n))
post = factory.SubFactory(PostFactory)
Expand All @@ -92,7 +101,8 @@ class PostLinkFactory(factory.DjangoModelFactory):


class FeedLinkFactory(factory.DjangoModelFactory):
FACTORY_FOR = FeedLink
class Meta:
model = FeedLink

feed = factory.SubFactory(FeedFactory)
rel = "alternative"
Expand All @@ -101,7 +111,8 @@ class FeedLinkFactory(factory.DjangoModelFactory):


class EnclosureFactory(factory.DjangoModelFactory):
FACTORY_FOR = Enclosure
class Meta:
model = Enclosure

title = factory.Sequence(lambda n: u'enclosure-#{}'.format(n))
post = factory.SubFactory(PostFactory)
Expand Down

0 comments on commit a6c9f1e

Please sign in to comment.