feast-dev
diff --git a/‎job-controller/src/main/java/feast/jobcontroller/runner/dataflow/DataflowJobManager.java‎
Lines changed: 4 additions & 0 deletions b/‎job-controller/src/main/java/feast/jobcontroller/runner/dataflow/DataflowJobManager.java‎
Lines changed: 4 additions & 0 deletions
diff --git a/‎job-controller/src/test/java/feast/jobcontroller/runner/dataflow/DataflowJobManagerTest.java‎
Lines changed: 2 additions & 1 deletion b/‎job-controller/src/test/java/feast/jobcontroller/runner/dataflow/DataflowJobManagerTest.java‎
Lines changed: 2 additions & 1 deletion
@@ -304,6 +304,10 @@ public List<Job> listRunningJobs() {
                       .setStores(
                           stores.stream()
                               .collect(Collectors.toMap(StoreProto.Store::getName, s -> s)))
+                      .setLabels(
+                          dfJob.getLabels() == null
+                              ? new HashMap<>()
+                              : new HashMap<>(dfJob.getLabels()))
                       .build();
 
               job.setExtId(dfJob.getId());
 
@@ -275,7 +275,8 @@ public void shouldRetrieveRunningJobsFromDataflow() {
                 hasProperty("stores", hasValue(store)),
                 hasProperty("extId", equalTo("job-2")),
                 hasProperty("created", equalTo(created.toDate())),
-                hasProperty("lastUpdated", equalTo(created.toDate())))));
+                hasProperty("lastUpdated", equalTo(created.toDate())),
+                hasProperty("labels", hasEntry("application", "feast")))));
   }
 
   @Test
Original file line number	Diff line number	Diff line change
`@@ -275,7 +275,8 @@ public void shouldRetrieveRunningJobsFromDataflow() {`
`275`	`275`	`hasProperty("stores", hasValue(store)),`
`276`	`276`	`hasProperty("extId", equalTo("job-2")),`
`277`	`277`	`hasProperty("created", equalTo(created.toDate())),`
`278`		`- hasProperty("lastUpdated", equalTo(created.toDate())))));`
	`278`	`+ hasProperty("lastUpdated", equalTo(created.toDate())),`
	`279`	`+ hasProperty("labels", hasEntry("application", "feast")))));`
`279`	`280`	`}`
`280`	`281`
`281`	`282`	`@Test`