class DataStoreTestCase(unittest.TestCase):
def setUp(self):
self.datastore = DataStore()
def test_add_value(self):
"""The values are added under the series name key."""
self.datastore.add_value('name', 123456789, 42)
self.assertEqual(len(self.datastore.keys()), 1)
self.assert_('name' in self.datastore.keys())
self.assertEqual(len(self.datastore['name']), 1)
self.assertEqual(self.datastore['name'], [(123456789, 42)])
def test_download_render_POST(self):
"""Download JSON data file."""
datastore = DataStore()
datastore.add_value("var1", 1, 1)
datastore.add_value("var2", 2, 2)
resource = Index(datastore)
request = DummyRequest(['foo'])
request.args = {'download': 'download'}
d = request.notifyFinish()
result = resource.render_POST(request)
self.assertEqual(result, server.NOT_DONE_YET)
yield d
expected = '["var1", 1, 1]\n["var2", 2, 2]'
self.assertEqual(expected, ''.join(request.written))
def test_data_render_POST(self):
"""Main page rendering with data."""
datastore = DataStore()
datastore.add_value("var1", 1, 1)
datastore.add_value("var2", 2, 2)
resource = Index(datastore)
request = DummyRequest(['foo'])
d = request.notifyFinish()
result = resource.render_GET(request)
self.assertEqual(result, server.NOT_DONE_YET)
yield d
element = ChartDataElement(datastore)
expected = yield flattenString(request, element)
self.assertIn(expected, ''.join(request.written))
