def test_get_package_handler(self):
PackageDictionary(id='1',
name='package1',
release_time=1,
words=['tea', 'coffee']).put()
PackageDictionary(id='2',
name='package2',
release_time=2,
words=['apple', 'banana']).put()
request = webapp2.Request.blank(r'/device_id_1/streams/packages/1')
response = request.get_response(main.app)
self.assertEqual(response.status_int, 200)
right_json = u'''{
"id": "1",
"name": "package1",
"release_time": 1,
"words": [
"tea",
"coffee"
]
}
'''
right = json.loads(right_json)
response_struct = json.loads(response.body)
self.assertEqual(right, response_struct)
def test_change_words_handler(self):
PackageDictionary(id='package_id_1',
name='package1',
release_time=1,
words=['apple', 'banana']).put()
request = webapp2.Request.blank(
'/admin/streams/packages/package_id_1/words')
response = request.get_response(main.app)
self.assertEqual(response.status_int, 200)
def post(self, **kwargs):
if PackageDictionary.query(PackageDictionary.id == self.request.get(
'package_id')).get() is None:
PackageDictionary(id=self.request.get('package_id'),
name=self.request.get('package_name'),
release_time=int(
self.request.get('release_time')),
words=[]).put()
stream = PackagesStream.query(
PackagesStream.id == kwargs.get('stream_id')).get()
if self.request.get('package_id') not in stream.packages_id_list:
stream.packages_id_list.append(self.request.get('package_id'))
stream.put()
self.redirect('/admin/streams/' + kwargs.get('stream_id') +
'/packages/add')
