rehabstudio / rehabstudio-gae-scaffold Goto Github PK

from google.appengine.ext import ndb
from tests.testcases import BaseTestCase

# This import (unused in this particular test case) is found in many
# of the test modules already present in app/tests/base. It appears
# that the call to GetApplicationConfiguration within app/config.py
# causes the bug. Remove this import to see that the problem is
# resolved.
from app import config


class MyModel(ndb.Model):
    value = ndb.StringProperty()


class ExampleTestCase(BaseTestCase):

    def test_1_model_is_created(self):
        """ Persist an object in the datastore with and ID of 1234.
        """
        m = MyModel(id=1234, value='foobar')
        m.put()
        self.assertEqual(len(MyModel.query().fetch()), 1)

    def test_2_model_no_longer_exists(self):
        """ Assert that the datastore is reset from the previous test
        run and that object 1234 is no longer present in the datastore.
        """
        self.assertEqual(len(MyModel.query().fetch()), 0)

        # This assertion will fail even though the datastore is shown
        # to be reset in the previous assertion. This hints that the
        # datastore's in-context cache is not being cleared.
        self.assertIsNone(MyModel.get_by_id(1234))