pip install test-helper-gae3
import unittest
from google.appengine.ext import ndb
from test_helper_gae3 import TestGae3
class ModelTest(ndb.Model):
name = ndb.StringProperty(default='')
class TestCase(unittest.TestCase, TestGae3):
def setUp(self):
super().setUp()
# For using queues names other then 'default', root_path dir must contain file
# 'queue.yaml' (or 'queue.yml') with correct queues definition.
# If root_path set to None, only 'default' queue is available.
TestGae3.set_up(self, 'path/to/folder/with/queue.yaml') # activate GAE testbed
def tearDown(self):
TestGae3.tear_down(self) # deactivate GAE testbed
super().tearDown()
def test_record_count(self):
self.check_db_tables([
(ModelTest, 0), # no records in ModelTest
])
ModelTest().put()
self.tester.check_db_tables([
(ModelTest, 1), # one record in ModelTest
])