Add ability to dismiss notifications

test/test_api_usage.py

@@ -23,7 +23,8 @@ from endpoints.api.trigger import (BuildTriggerActivate, BuildTriggerSources, Bu
 from endpoints.api.repoemail import RepositoryAuthorizedEmail
 from endpoints.api.repositorynotification import RepositoryNotification, RepositoryNotificationList
 from endpoints.api.user import (PrivateRepositories, ConvertToOrganization, Signout, Signin, User,
-                                UserAuthorizationList, UserAuthorization)
+                                UserAuthorizationList, UserAuthorization, UserNotification,
+                                UserNotificationList)
 
 from endpoints.api.repotoken import RepositoryToken, RepositoryTokenList
 from endpoints.api.prototype import PermissionPrototype, PermissionPrototypeList
@@ -208,6 +209,26 @@ class TestLoggedInUser(ApiTestCase):
     assert json['username'] == READ_ACCESS_USER
 
 
+class TestUserNotification(ApiTestCase):
+  def test_get(self):
+    self.login(ADMIN_ACCESS_USER)
+    json = self.getJsonResponse(UserNotificationList)
+
+    # Make sure each notification can be retrieved.
+    for notification in json['notifications']:
+      njson = self.getJsonResponse(UserNotification, params=dict(uuid=notification['id']))
+      self.assertEquals(notification['id'], njson['id'])
+
+    # Update a notification.
+    assert json['notifications']
+    assert not json['notifications'][0]['dismissed']
+    
+    pjson = self.putJsonResponse(UserNotification, params=dict(uuid=notification['id']),
+                                 data=dict(dismissed=True))
+      
+    self.assertEquals(True, pjson['dismissed'])
+
+
 class TestGetUserPrivateAllowed(ApiTestCase):
   def test_nonallowed(self):
     self.login(READ_ACCESS_USER)