def test_run_additional_queries(self):
     """
     should run n additional queries, where n is the number of associations
     in the query 'includes' value
     """
     Site.includes("articles", "comments", "maintainer").all()
     self.assertEqual(len(self.adapter.calls), 4)
Exemple #2
0
 def test_run_additional_queries(self):
     """
     should run n additional queries, where n is the number of associations
     in the query 'includes' value
     """
     Site.includes('articles', 'comments', 'maintainer').all()
     self.assertEqual(len(self.adapter.calls), 4)
Exemple #3
0
 def test_preload_not_supported(self):
     """
     should raise AssociationPreloadNotSupported if the association requires
     an instance of a model (e.g. polymorphic belongs_to)
     """
     self.assertRaises(AssociationPreloadNotSupported,
                       Site.includes('fun_articles').all)
    def test_include_results(self):
        """should include the preload results on the relation"""
        rel = Site.includes("articles", "comments", "maintainer", "headline")
        results = rel.all()
        site = results[0]
        call_count = len(self.adapter.calls)

        # has_many
        self.assertEqual(type(site.articles), pyperry.Relation)
        for article in site.articles:
            self.assertEqual(type(article), Article)

        # has_many polymorphic
        self.assertEqual(type(site.comments), pyperry.Relation)
        for comment in site.comments:
            self.assertEqual(type(comment), Comment)

        # belongs_to
        self.assertEqual(type(site.maintainer), Person)

        # has_one
        self.assertEqual(type(site.headline), Article)

        # Ensure no more calls were made during our assertions
        self.assertEqual(len(self.adapter.calls), call_count)
 def test_preload_not_supported(self):
     """
     should raise AssociationPreloadNotSupported if the association requires
     an instance of a model (e.g. polymorphic belongs_to)
     """
     self.assertRaises(AssociationPreloadNotSupported,
                       Site.includes('fun_articles').all)
Exemple #6
0
    def test_include_results(self):
        """should include the preload results on the relation"""
        rel = Site.includes('articles', 'comments', 'maintainer', 'headline')
        results = rel.all()
        site = results[0]
        call_count = len(self.adapter.calls)

        # has_many
        self.assertEqual(type(site.articles), pyperry.Relation)
        for article in site.articles:
            self.assertEqual(type(article), Article)

        # has_many polymorphic
        self.assertEqual(type(site.comments), pyperry.Relation)
        for comment in site.comments:
            self.assertEqual(type(comment), Comment)

        # belongs_to
        self.assertEqual(type(site.maintainer), Person)

        # has_one
        self.assertEqual(type(site.headline), Article)

        # Ensure no more calls were made during our assertions
        self.assertEqual(len(self.adapter.calls), call_count)
    def test_nested_includes(self):
        """should nest includes queries using a tree like syntax"""
        data = self.data(5)
        for d in data:
            d.update({"Comment": {"parent_id": 1, "parent_type": "Article", "person_id": 1}, "Person": {"id": 1}})
        PreloadTestAdapter.data = data

        rel = Site.includes({"articles": "comments", "comments": "author"})
        sites = rel.all()
        self.assertEqual(len(self.adapter.calls), 5)
    def test_include_modifiers(self):
        """
        should include the modifiers value from the original query in
        subsequent queries
        """
        test = {"agentp": "platypus", "agentm": "monkey"}
        sites = Site.modifiers(test).includes("articles", "comments", "maintainer", "headline").all()

        for call in self.adapter.calls:
            self.assertEqual(call.modifiers_value(), test)
Exemple #9
0
    def test_include_modifiers(self):
        """
        should include the modifiers value from the original query in
        subsequent queries
        """
        test = {'agentp': 'platypus', 'agentm': 'monkey'}
        sites = Site.modifiers(test).includes('articles', 'comments',
                            'maintainer', 'headline').all()

        for call in self.adapter.calls:
            self.assertEqual(call.modifiers_value(), test)
    def test_include_modifiers(self):
        """
        should include the modifiers value from the original query in
        subsequent queries
        """
        test = {'agentp': 'platypus', 'agentm': 'monkey'}
        sites = Site.modifiers(test).includes('articles', 'comments',
                            'maintainer', 'headline').all()

        for call in self.adapter.calls:
            self.assertEqual(call.modifiers_value(), test)
    def test_cache_results(self):
        """
        should cache results for collection associations in a relation with the
        correct scope
        """
        site = Site.includes("articles", "comments")[0]

        article_relation = site.defined_associations["articles"].scope(site)
        comment_relation = site.defined_associations["comments"].scope(site)

        self.assertEqual(site.articles.query(), article_relation.query())
        self.assertEqual(site.comments.query(), comment_relation.query())
Exemple #12
0
    def test_cache_results(self):
        """
        should cache results for collection associations in a relation with the
        correct scope
        """
        site = Site.includes('articles', 'comments')[0]

        article_relation = site.defined_associations['articles'].scope(site)
        comment_relation = site.defined_associations['comments'].scope(site)

        self.assertEqual(site.articles.query(), article_relation.query())
        self.assertEqual(site.comments.query(), comment_relation.query())
Exemple #13
0
    def test_nested_includes(self):
        """should nest includes queries using a tree like syntax"""
        data = self.data(5)
        for d in data:
            d.update({
                'Comment': {'parent_id': 1, 'parent_type': 'Article',
                            'person_id': 1},
                'Person': {'id': 1}
            })
        PreloadTestAdapter.data = data

        rel = Site.includes({'articles': 'comments', 'comments': 'author'})
        sites = rel.all()
        self.assertEqual(len(self.adapter.calls), 5)
    def test_nested_includes(self):
        """should nest includes queries using a tree like syntax"""
        data = self.data(5)
        for d in data:
            d.update({
                'Comment': {'parent_id': 1, 'parent_type': 'Article',
                            'person_id': 1},
                'Person': {'id': 1}
            })
        PreloadTestAdapter.data = data

        rel = Site.includes({'articles': 'comments', 'comments': 'author'})
        sites = rel.all()
        self.assertEqual(len(self.adapter.calls), 5)
    def test_only_include_matching_records(self):
        """should only include records that match for a given model"""
        rel = Site.includes("articles", "comments", "maintainer", "headline")
        sites = rel.all()
        call_count = len(self.adapter.calls)

        for site in sites:
            # has_many
            for article in site.articles:
                self.assertEqual(article.site_id, site.id)
            # has_many polymorphic
            for comment in site.comments:
                self.assertEqual(comment.parent_id, site.id)
                self.assertEqual(comment.parent_type, "Site")
            # belongs_to
            self.assertEqual(site.maintainer.id, site.maintainer_id)
            # has_one
            if site.id == 1:
                self.assertEqual(site.headline.site_id, site.id)

        call_count = len(self.adapter.calls)
Exemple #16
0
    def test_only_include_matching_records(self):
        """should only include records that match for a given model"""
        rel = Site.includes('articles', 'comments', 'maintainer', 'headline')
        sites = rel.all()
        call_count = len(self.adapter.calls)

        for site in sites:
            # has_many
            for article in site.articles:
                self.assertEqual(article.site_id, site.id)
            # has_many polymorphic
            for comment in site.comments:
                self.assertEqual(comment.parent_id, site.id)
                self.assertEqual(comment.parent_type, 'Site')
            # belongs_to
            self.assertEqual(site.maintainer.id, site.maintainer_id)
            # has_one
            if site.id == 1:
                self.assertEqual(site.headline.site_id, site.id)

        call_count = len(self.adapter.calls)
 def test_association_not_found(self):
     """
     should raise AssociationNotFound for associations given in 'includes'
     that do not exist on the model
     """
     self.assertRaises(AssociationNotFound, Site.includes("batteries").all)
Exemple #18
0
 def test_association_not_found(self):
     """
     should raise AssociationNotFound for associations given in 'includes'
     that do not exist on the model
     """
     self.assertRaises(AssociationNotFound, Site.includes('batteries').all)