def run(self):
import os.path
from test import unittest
test_dir = os.path.join(os.path.dirname(__file__), 'test')
package_suite = unittest.TestLoader().discover(test_dir)
unittest.TextTestRunner(verbosity=2).run(package_suite)
def run_test(module, options):
suite = unittest.TestSuite()
test_utils.fail_incomplete_tests = options.incomplete
m = __import__(module)
if m.unittest is not unittest:
raise ImportError(
"%s is not using correct unittest\n\n" % module +
"should be: %s\n is using: %s" % (unittest.__file__,
m.unittest.__file__)
)
print 'loading', module
test = unittest.defaultTestLoader.loadTestsFromName(module)
suite.addTest(test)
output = StringIO.StringIO()
runner = unittest.TextTestRunner(stream = output)
results = runner.run(suite)
output = StringIOContents(output)
num_tests = results.testsRun
failures = results.failures
errors = results.errors
tests = results.tests
results = {module:from_namespace(locals(), RESULTS_TEMPLATE)}
if options.subprocess:
print TEST_RESULTS_START
print pformat(results)
else:
return results
results = self.Post.query.msearch('title:search').all()
self.assertEqual(len(results), 2)
results = self.Post.query.msearch('content:search').all()
self.assertEqual(len(results), 1)
results = self.Post.query.msearch(
'title:search OR content:title').all()
self.assertEqual(len(results), 3)
results = self.Post.query.msearch('*search').all()
self.assertEqual(len(results), 3)
results = self.Post.query.msearch('search*').all()
self.assertEqual(len(results), 2)
results = self.Post.query.msearch('search*', geohash='f37p0hxhg382', geofield='geo').all()
self.assertEqual(len(results), 1)
results = self.Post.query.msearch('', geohash='m8umpccxt26m', geofield='geo').all()
self.assertEqual(len(results), 2)
if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromNames(
[
'test_elasticsearch.TestSearch',
])
unittest.TextTestRunner(verbosity=1).run(suite)
