Fix key and active data mapping when adding a data object

Change-Id: Ibfaf4b62673c008b94e039deb77f6f6d73d83753

client/src/com/vaadin/client/connectors/data/typed/DataSourceConnector.java

@@ -88,6 +88,7 @@ public class DataSourceConnector extends AbstractExtensionConnector {
             @Override
             public void add(JsonObject dataObject) {
                 ds.asList().add(dataObject);
+                keyToJson.put(getKey(dataObject), dataObject);
             }
 
             @Override

server/src/com/vaadin/server/communication/data/typed/SimpleDataProvider.java

@@ -15,6 +15,7 @@
  */
 package com.vaadin.server.communication.data.typed;
 
+import java.util.Collections;
 import java.util.HashSet;
 import java.util.Set;
 
@@ -138,6 +139,7 @@ public class SimpleDataProvider<T> extends DataProvider<T> {
      */
     protected void add(T data) {
         rpc.add(getDataObject(data));
+        handler.addActiveData(Collections.singleton(data));
     }
 
     /**