def test_add_package_handler(self):
PackagesStream(id='stream_id_1', name='stream1',
packages_id_list=[]).put()
request = webapp2.Request.blank(
'/admin/streams/stream_id_1/packages/add')
response = request.get_response(main.app)
self.assertEqual(response.status_int, 200)
def test_get_packages_list_handler(self):
PackagesStream(id='1', name='stream1', packages_id_list=['1',
'2']).put()
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/1')
response = request.get_response(main.app)
self.assertEqual(response.status_int, 200)
right_json = u'''{
"packages": [
{
"id": "1",
"name": "package1",
"release_time": 1
},
{
"id": "2",
"name": "package2",
"release_time": 2
}
]
}
'''
right = json.loads(right_json)
response_struct = json.loads(response.body)
self.assertEqual(response_struct, right)
def post(self, *args, **kwargs):
if PackagesStream.query(PackagesStream.id == self.request.get(
'stream_id')).get() is None:
PackagesStream(id=self.request.get('stream_id'),
name=self.request.get('stream_name'),
packages_id_list=[]).put()
self.redirect('/admin/streams')
def test_get_streams_list_handler(self):
PackagesStream(id='1', name='stream1', packages_id_list=['1',
'2']).put()
PackagesStream(id='2', name='stream2', packages_id_list=['3']).put()
request = webapp2.Request.blank('/device_id_1/streams')
response = request.get_response(main.app)
self.assertEqual(response.status_int, 200)
right_json = u'''{
"streams": [
{
"id": "1",
"name": "stream1"
},
{
"id": "2",
"name": "stream2"
}
]
}
'''
right = json.loads(right_json)
response_struct = json.loads(response.body)
self.assertEqual(response_struct, right)
