added count/count_by to assignmapper, plus a test in activemapper to try it out

3 files changed, 5 insertions, 1 deletions

diff --git a/CHANGES b/CHANGES
index acf61f870..d080ad9ea 100644
--- a/CHANGES
+++ b/CHANGES
@@ -9,6 +9,8 @@ activated when activemapper is imported
 - fixes to Session expunge/update/etc.
 - select_table mappers *still* werent always compiling
 - fixed up Boolean datatype
+- added count()/count_by() to list of methods proxied by assignmapper;
+this also adds them to activemapper
 
 0.2.5
 - fixed endless loop bug in select_by(), if the traversal hit
diff --git a/lib/sqlalchemy/ext/assignmapper.py b/lib/sqlalchemy/ext/assignmapper.py
index 29cc70b3e..240a46e3b 100644
--- a/lib/sqlalchemy/ext/assignmapper.py
+++ b/lib/sqlalchemy/ext/assignmapper.py
@@ -31,7 +31,7 @@ def assign_mapper(ctx, class_, *args, **kwargs):
         extension = ctx.mapper_extension
     m = mapper(class_, extension=extension, *args, **kwargs)
     class_.mapper = m
-    for name in ['get', 'select', 'select_by', 'selectone', 'get_by', 'join_to', 'join_via']:
+    for name in ['get', 'select', 'select_by', 'selectone', 'get_by', 'join_to', 'join_via', 'count', 'count_by']:
         monkeypatch_query_method(ctx, class_, name)
     for name in ['flush', 'delete', 'expire', 'refresh', 'expunge', 'merge', 'save', 'update', 'save_or_update']:
         monkeypatch_objectstore_method(ctx, class_, name)
diff --git a/test/ext/activemapper.py b/test/ext/activemapper.py
index 85466e29b..1ff8b0d7b 100644
--- a/test/ext/activemapper.py
+++ b/test/ext/activemapper.py
@@ -218,6 +218,8 @@ class testcase(testbase.PersistTest):
         )
         self.assertEquals(len(results), 1)
 
+        self.assertEquals(Person.count(), 2)
+        
 class testselfreferential(testbase.PersistTest):
     def setUpAll(self):
         global TreeNode