Ejemplo n.º 1
0
 def get(self):
     """
     Chart about specified counters
     
     @param names: comma separated list of counter names
     @param captions: (optional) comma separated list of user-readable names (should match counter names)
     @param title: (optional)
     
     Ex: "?names=email_bounce,email_unsubscribe&captions=Bounces,Unsubscribes&title=Emails Stats"
     """
     names = self.request.get('names')
     if not names:
         self.response.content_type = 'text/plain'
         self.response.write("Invalid Input.\n\n" + self.get.__doc__)
         return
     names = [n.strip() for n in names.split(',')]
     
     captions = self.request.get('captions')
     prefix = None
     if captions:
         captions = [n.strip() for n in captions.split(',')]
         if captions and len(captions) != len(names):
             self.response.content_type = 'text/plain'
             self.response.write("Invalid Input.\n\n" + self.get.__doc__)
             return
     else:
         for name in names:
             if prefix:
                 common = ''
                 for cn, cp in zip(name, prefix):
                     if cn == cp:
                         common += cn
                 prefix = common
             else:
                 prefix = name
         
         if prefix:
             captions = [name[len(prefix):] for name in names]
         else:
             captions = names
     
     title = self.request.get('title') or prefix or 'Counters Chart'
     
     counterFuts = [Counter.getCurrent(name) for name in names]
     counters = [fut.get_result() for fut in counterFuts]
     
     slices = set()
     for counter in counters:
         slices.update(counter.bySlice)
     slices = sorted(slices)
     
     chartData = [["On"] + captions]
     for item in slices:
         row = [item]
         for counter in counters:
             row.append(counter.bySlice.get(item, 0))
         chartData.append(row)
         
     self.writeTemplate("countersChart.html", {"chartData": json.dumps(chartData), "title": title})
Ejemplo n.º 2
0
 def testMaxNameLength(self):
     longCounterName = 't' * Counter.NAME_MAX_LENGTH
     
     incrementSuccess = Counter.increment(longCounterName, 1).get_result()
     self.assertTrue(incrementSuccess)
     
     counter = Counter.getCurrent(longCounterName).get_result()
     self.assertEqual(counter.name, longCounterName, "Name integrity was not maintained")
Ejemplo n.º 3
0
 def testSlices(self):
     """ Test using default and custom slices while counting """
     # Default slice
     today = datetime.datetime.utcnow().strftime("%Y-%m-%d")
     Counter.increment('testDefault', 1).get_result()
     counter = Counter.getCurrent('testDefault').get_result()
     self.assertEqual(counter.value, 1)
     self.assertDictEqual(counter.bySlice, {today: 1})
     
     # Custom slice
     futs = [Counter.increment('testCustom', 1, sliceId="13")]
     futs.append(Counter.increment('testCustom', 2, sliceId="12"))
     [fut.get_result() for fut in futs]
     
     counter = Counter.getCurrent('testCustom').get_result()
     self.assertEqual(counter.value, 3)
     self.assertDictEqual(counter.bySlice, {"12": 2, "13": 1})
Ejemplo n.º 4
0
 def testShards(self):
     """ Test that 'getCurrent' correctly retrieves values from different shards """
     futs = []
     for _ in xrange(10):
         # Because of NDB issue I can't perform all at the same time
         # https://code.google.com/p/googleappengine/issues/detail?id=10617
         futs.append(Counter.increment('testShards', nbShards=3).get_result())
     successes = futs  # getAllResults(futs)
     self.assertEqual(10, sum(successes))
     
     counter = Counter.getCurrent('testShards', nbShards=3).get_result()
     self.assertEqual(counter.value, 10)