Extending unit test for remove parameter issue

Author: PicoCreator

src/test/java/picoded/dstack/struct/simple/StructSimple_DataObjectMap_test.java

@@ -614,13 +614,41 @@ public void removeViaDataObject() {
 		// Lets just rescycle old test for some dummy data
 		basicTest();
 
+		// Lets get DataObject list
+		DataObject[] oRes = null;
+		assertNotNull(oRes = mtObj.query(null, null));
+		assertTrue(oRes.length > 0);
+
+		// Lets get the first object
+		DataObject testObject = oRes[0];
+		String oid = testObject._oid();
+		testObject.remove("num");
+		testObject.remove("str_val");
+		testObject.saveDelta();
+		
+		// Get the object again
+		DataObject changedObject = mtObj.get(oid);
+		assertNotNull( changedObject );
+
+		// Check the respective value is null
+		assertNull( changedObject.get("num") );
+		assertNull( changedObject.get("str_val") );
+		
+	}
+	
+	@Test
+	public void removeViaMetaOID() {
+		
+		// Lets just rescycle old test for some dummy data
+		basicTest();
+		
 		// Lets get DataObject list
 		DataObject[] oRes = null;
 		assertNotNull(oRes = mtObj.query(null, null));
 		assertTrue(oRes.length > 0);
 
 		// Lets remove one object
-		mtObj.remove(oRes[0]);
+		mtObj.remove(oRes[0]._oid());
 
 		// Lets query to make sure its removed
 		DataObject[] qRes = null;
@@ -629,7 +657,7 @@ public void removeViaDataObject() {
 	}
 
 	@Test
-	public void removeViaMetaOID() {
+	public void removePropertyForMetaObject() {
 
 		// Lets just rescycle old test for some dummy data
 		basicTest();
@@ -640,7 +668,7 @@ public void removeViaMetaOID() {
 		assertTrue(oRes.length > 0);
 
 		// Lets remove one object
-		mtObj.remove(oRes[0]._oid());
+		mtObj.remove(oRes[0]);
 
 		// Lets query to make sure its removed
 		DataObject[] qRes = null;