add load_mongo_fixture function

1 files changed, 13 insertions, 0 deletions

diff --git a/webapp/webapp/features/terrain.py b/webapp/webapp/features/terrain.py
index 69d392a..0ae8f58 100644
--- a/webapp/webapp/features/terrain.py
+++ b/webapp/webapp/features/terrain.py
@@ -38,6 +38,19 @@ def load_fixture(fixture_name):
                             verbosity=0)
 
 
+def load_mongo_fixture(fixture_name, collection):
+    POLLSTER_FIXTURES = os.path.join(settings.PROJECT_ROOT,
+                                     '../pollster/fixtures')
+    fixture_path = os.path.join(POLLSTER_FIXTURES, fixture_name + ".json")
+    kwargs = {
+        'db': settings.MONGO_SETTINGS['NAME'],
+        'collection': collection,
+        'fixture': fixture_path,
+    }
+    cmd = "mongoimport -d {db} -c {collection} --file {fixture} --jsonArray"
+    local(cmd.format(**kwargs))
+
+
 @before.each_feature
 def before_each_feature(feature):
     """This is the main lettuce hook "@before.each_feature"."""