Merge pull request #758 from 18F/include-grade-in-timecards-json

Include 'grade' in timecards.json response.

api-docs/projects.md

@@ -32,6 +32,7 @@ To fetch a list of projects  information.
         "billable": true,
         "start_date": null,
         "end_date": null,
+        "grade": 16,
         "active": true
     },...
 
@@ -48,5 +49,8 @@ To fetch a list of projects  information.
 $ curl https://tock.18f.gov/api/projects.json -H 'Authorization: Token randomalphanumericstringed854b18ba024327'
 ```
 
-* **Notes:** None.
-
+* **Notes:**
+
+Note that `grade` may be `null` if no grade information is
+available. If it is non-null, it will be a number corresponding to a grade;
+the mapping is defined by `GRADE_CHOICES` in `tock/employees/models.py`.

tock/api/tests.py

@@ -70,6 +70,20 @@ def test_timecards_json(self):
         clean_res = json.loads(res.decode())
         self.assertEqual(len(clean_res), 2)
 
+    def test_timecards_grade_is_null_when_absent(self):
+        res = client(self).get(
+            reverse('TimecardList'),
+            data={'date': '2016-06-01'}).content
+        clean_res = json.loads(res.decode())
+        self.assertEqual(clean_res[0]['grade'], None)
+
+    def test_timecards_grade_is_populated_when_present(self):
+        res = client(self).get(
+            reverse('TimecardList'),
+            data={'date': '2015-06-01'}).content
+        clean_res = json.loads(res.decode())
+        self.assertEqual(clean_res[0]['grade'], 4)
+
     # TODO: test with more diverse data
     def test_get_timecards(self):
         """ Check that get time cards returns the correct queryset """

tock/api/views.py

@@ -109,6 +109,8 @@ class TimecardSerializer(serializers.Serializer):
     project_organization = serializers.CharField(
         source='project.organization_name'
     )
+    grade = serializers.IntegerField(
+        source='grade.grade')
 
 # API Views
 
@@ -167,6 +169,7 @@ class TimecardList(generics.ListAPIView):
         'timecard__user',
         'project__accounting_code__agency',
         'timecard__reporting_period',
+        'grade',
     ).order_by(
         'timecard__reporting_period__start_date'
     )

tock/hours/fixtures/timecards.json

@@ -1,4 +1,13 @@
 [
+  {
+    "model": "employees.EmployeeGrade",
+    "fields": {
+      "employee": 1,
+      "grade": 4,
+      "g_start_date": "2015-05-28"
+    },
+    "pk": 1
+  },
   {
     "model": "hours.reportingperiod",
     "fields": {
@@ -24,6 +33,7 @@
       "timecard": 1,
       "project": 1,
       "hours_spent": 20,
+      "grade": 1,
       "created": "2015-06-08T12:00:00.000Z",
       "modified": "2015-06-08T12:00:00.000Z"
     },