def setUp(self): # pylint: disable=C0103 self.data_handler = PageEntryDataHandler() # First, create an instance of the Testbed class. self.testbed = testbed.Testbed() # Then activate the testbed, which prepares the service stubs for use. self.testbed.activate() # Next, declare which service stubs you want to use. self.testbed.init_datastore_v3_stub() self.testbed.init_memcache_stub()
class TestPageEntryDataHandler(unittest.TestCase): # pylint: disable=R0904 def setUp(self): # pylint: disable=C0103 self.data_handler = PageEntryDataHandler() # First, create an instance of the Testbed class. self.testbed = testbed.Testbed() # Then activate the testbed, which prepares the service stubs for use. self.testbed.activate() # Next, declare which service stubs you want to use. self.testbed.init_datastore_v3_stub() self.testbed.init_memcache_stub() def tearDown(self): # pylint: disable=C0103 self.testbed.deactivate def test_page_entry_get_and_get(self): actual = {} actual['subject'] = 'The subject' actual['content'] = 'The content' page_entry_id = self.data_handler.put(actual['subject'], actual['content']) expected = self.data_handler.get_by_id(page_entry_id) self.assertEqual(actual, expected) def test_get_all_page_entries(self): actual_input = {} actual_input['subject'] = 'The subject' actual_input['content'] = 'The content' self.data_handler.put(actual_input['subject'], actual_input['content']) self.data_handler.put(actual_input['subject'], actual_input['content']) actual = self.data_handler.get_all() expected = [actual_input, actual_input] self.assertEqual(actual, expected)