Upload file to app engine and generate public url for dmprof visualizer.

tools/deep_memory_profiler/visualizer/app_unittest.py

+# Copyright 2013 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import services
+import unittest
+
+from google.appengine.api import files
+from google.appengine.ext import testbed
+from google.appengine.ext.blobstore import BlobInfo
+
+
+class ServicesTest(unittest.TestCase):
+  def setUp(self):
+    self.testbed = testbed.Testbed()
+    self.testbed.activate()
+    self.testbed.init_all_stubs()
+
+    # Read sample file.
+    self.json_str = open('testdata/sample.json', 'r').read()
+
+    # Create file in blobstore according to sample file.
+    file_name = files.blobstore.create(mime_type='text/plain')
+    with files.open(file_name, 'a') as f:
+      f.write(self.json_str)
+    files.finalize(file_name)
+
+    # Get BlobInfo of sample file.
+    self.blob_info = BlobInfo.get(files.blobstore.get_blob_key(file_name))
+
+  def tearDown(self):
+    self.testbed.deactivate()
+
+  def testProfiler(self):
+    # Call services function to create Profiler entity.
+    run_id = services.CreateProfiler(self.blob_info)
+
+    # Test GetProfiler
+    self.assertEqual(services.GetProfiler(run_id), self.json_str)
+
+    # Create Profiler entity with the same file again and check uniqueness.
+    services.CreateProfiler(self.blob_info)
+    self.assertEqual(services.Profiler.query().count(), 1)
+
+  def testTemplate(self):
+    # Call services function to create template entities.
+    services.CreateTemplates(self.blob_info)
+    self.assertEqual(services.Template.query().count(), 2)

[sullivan, 2013/09/20 13:59:51]

Might want to do some testing of the content of the Templates to make sure
they're correct.

Also might want to add a test to check what happens if there is no
default_template.

[junjianx, 2013/09/24 05:53:53]

Done.