Add build mailer capability to support gatekeeper_ng.

app_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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 ast
 import datetime
+import hashlib
+import hmac
+import json
 import os
+import random
+import time
 import unittest
 
 import app
+from third_party.BeautifulSoup.BeautifulSoup import BeautifulSoup
 
 
 TEST_DIR = os.path.join(os.path.dirname(__file__), 'tests')
 
 
 class GaeTestCase(unittest.TestCase):
   def setUp(self, *args, **kwargs):
     self.clear_datastore()
     super(GaeTestCase, self).setUp(*args, **kwargs)
 
@@ skipping 444 matching lines @@
     page = app.get_and_cache_pagedata('chromium/console')
     # Uncomment if deeper inspection is needed of the returned console.
     # with open(os.path.join(test_dir, 'expected.html'), 'w') as fh:
     #   fh.write(page['content'])
     self.assertEquals('interface', page['body_class'])
     self.assertEquals(expected_content, page['content'])
     self.assertEquals(
         'http://build.chromium.org/p/chromium/console/../',
         page['offsite_base'])
     self.assertEquals('BuildBot: Chromium', page['title'])
+
+
+class MailTestCase(GaeTestCase):
+  def setUp(self):
+    self.test_dir = os.path.join(TEST_DIR, 'test_mailer')
+    with open(os.path.join(self.test_dir, 'input.json')) as f:
+      self.input_json = json.load(f)
+    self.build_data = json.loads(self.input_json['message'])
+
+  @staticmethod
+  def _hash_message(mytime, message, url, secret):
+    salt = random.getrandbits(32)
+    hasher = hmac.new(secret, message, hashlib.sha256)
+    hasher.update(str(mytime))
+    hasher.update(str(salt))
+    client_hash = hasher.hexdigest()
+
+    return {'message': message,
+            'time': mytime,
+            'salt': salt,
+            'url': url,
+            'hmac-sha256': client_hash,
+           }
+
+  def test_html_format(self):
+    import gatekeeper_mailer
+    template = gatekeeper_mailer.MailTemplate(self.build_data['waterfall_url'],
+                                              self.build_data['build_url'],
+                                              self.build_data['project_name'],
+                                              'test@chromium.org')
+
+    _, html_content, _ = template.genMessageContent(self.build_data)
+
+    with open(os.path.join(self.test_dir, 'expected.html')) as f:
+      expected_html = ' '.join(f.read().splitlines())
+
+    saw = str(BeautifulSoup(html_content)).split()
+    expected = str(BeautifulSoup(expected_html)).split()
+
+    self.assertEqual(saw, expected)
+
+  def test_hmac_validation(self):
+    from mailer import Email
+    message = self.input_json['message']
+    url = 'http://invalid.chromium.org'
+    secret = 'pajamas'
+
+    test_json = self._hash_message(time.time(), message, url, secret)
+    # pylint: disable=W0212
+    self.assertTrue(Email._validate_message(test_json, url, secret))
+
+    # Test that a trailing slash doesn't affect URL parsing.
+    test_json = self._hash_message(time.time(), message, url + '/', secret)
+    # pylint: disable=W0212
+    self.assertTrue(Email._validate_message(test_json, url, secret))
+
+    tests = [
+        self._hash_message(time.time() + 61, message, url, secret),
+        self._hash_message(time.time() - 61, message, url, secret),
+        self._hash_message(time.time(), message, url + 'hey', secret),
+        self._hash_message(time.time(), message, url, secret + 'hey'),
+    ]
+
+    for test_json in tests:
+      # pylint: disable=W0212
+      self.assertFalse(Email._validate_message(test_json, url, secret))
+
+    test_json = self._hash_message(time.time(), message, url, secret)
+    test_json['message'] = test_json['message'] + 'hey'
+    # pylint: disable=W0212
+    self.assertFalse(Email._validate_message(test_json, url, secret))