Changed the working of the iterative_task decorator.
The decorator now needs the logic for the model which it fetches. This is much nicer then "fetching" it using import.
Also order has been removed since it has no use when iterating over all entities when key is involved. Fields to filter can now dynamically be set by the function that is wrapped.
Note that it will merge the filter with the task_default kwarg and it will only use the start_key if their is no key present in the context.
Also note that task_default gathers all kwargs. I tried to do it otherwise (with default kwargs) but I couldn't get it to work. Feel free to improve :).
#!/usr/bin/python2.5
#
# Copyright 2009 the Melange authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
__authors__ = [
'"Augie Fackler" <durin42@gmail.com>',
]
import sys
import os
HERE = os.path.normpath(os.path.join(os.path.dirname(os.path.abspath(__file__)),
'..'))
appengine_location = os.path.join(HERE, 'thirdparty', 'google_appengine')
extra_paths = [HERE,
os.path.join(appengine_location, 'lib', 'django'),
os.path.join(appengine_location, 'lib', 'webob'),
os.path.join(appengine_location, 'lib', 'yaml', 'lib'),
os.path.join(appengine_location, 'lib', 'antlr3'),
appengine_location,
os.path.join(HERE, 'app'),
os.path.join(HERE, 'thirdparty', 'coverage'),
]
import nose
from nose import plugins
class AppEngineDatastoreClearPlugin(plugins.Plugin):
"""Nose plugin to clear the AppEngine datastore between tests.
"""
name = 'AppEngineDatastoreClearPlugin'
enabled = True
def options(self, parser, env):
return plugins.Plugin.options(self, parser, env)
def configure(self, parser, env):
plugins.Plugin.configure(self, parser, env)
self.enabled = True
def afterTest(self, test):
from google.appengine.api import apiproxy_stub_map
datastore = apiproxy_stub_map.apiproxy.GetStub('datastore')
# clear datastore iff one is available
if datastore is not None:
datastore.Clear()
def main():
sys.path = extra_paths + sys.path
os.environ['SERVER_SOFTWARE'] = 'Development via nose'
os.environ['SERVER_NAME'] = 'Foo'
os.environ['SERVER_PORT'] = '8080'
os.environ['APPLICATION_ID'] = 'test-app-run'
os.environ['USER_EMAIL'] = 'test@example.com'
os.environ['CURRENT_VERSION_ID'] = 'testing-version'
os.environ['HTTP_HOST'] = 'some.testing.host.tld'
import main as app_main
from google.appengine.api import apiproxy_stub_map
from google.appengine.api import datastore_file_stub
from google.appengine.api import mail_stub
from google.appengine.api import user_service_stub
from google.appengine.api import urlfetch_stub
from google.appengine.api.memcache import memcache_stub
apiproxy_stub_map.apiproxy = apiproxy_stub_map.APIProxyStubMap()
apiproxy_stub_map.apiproxy.RegisterStub('urlfetch',
urlfetch_stub.URLFetchServiceStub())
apiproxy_stub_map.apiproxy.RegisterStub('user',
user_service_stub.UserServiceStub())
apiproxy_stub_map.apiproxy.RegisterStub('datastore',
datastore_file_stub.DatastoreFileStub('test-app-run', None, None))
apiproxy_stub_map.apiproxy.RegisterStub('memcache',
memcache_stub.MemcacheServiceStub())
apiproxy_stub_map.apiproxy.RegisterStub('mail', mail_stub.MailServiceStub())
import django.test.utils
django.test.utils.setup_test_environment()
# register a core for the test modules to use
from soc.modules import callback
from soc.modules import core
callback.registerCore(core.Core())
callback.getCore().registerModuleCallbacks()
plugins = [AppEngineDatastoreClearPlugin()]
if '--coverage' in sys.argv:
from nose.plugins import cover
plugin = cover.Coverage()
plugins.append(plugin)
args = ['--with-coverage',
'--cover-package=soc',
'--cover-erase',
'--cover-html',
'--cover-html-dir=coverageResults']
sys.argv.remove('--coverage')
sys.argv += args
nose.main(plugins=plugins)
if __name__ == '__main__':
main()