NullDefaultFieldTests.java

package com.googlecode.objectify.test;

import javax.persistence.Embedded;
import javax.persistence.Id;

import org.testng.annotations.Test;

import com.google.appengine.api.datastore.Entity;
import com.googlecode.objectify.Key;
import com.googlecode.objectify.Objectify;
import com.googlecode.objectify.annotation.Cached;
import com.googlecode.objectify.annotation.Unindexed;

/**
 * Test behavior of null fields, and default values
 */
public class NullDefaultFieldTests extends TestBase
{
	public static class Struct {
		String s = "default1";

		public Struct()
		{
		}

		public Struct(String s)
		{
			this.s = s;
		}
	}
	
	@Cached
	public static class EntityWithDefault {
		@Id
		Long id;
		/** existing property */
		String a;
		/** new property */
		String b = "foo";
		/** new property */
		@Unindexed
		String c = "bar";
		/** new embedded */
		@Embedded
		Struct s = new Struct("default2");

		public EntityWithDefault()
		{
		}

		public EntityWithDefault(String a)
		{
			this.a = a;
		}

		public EntityWithDefault(String a, String b, String c)
		{
			this.a = a;
			this.b = b;
			this.c = c;
		}
	}

	/**
	 * Test that an entity in the datastore with absent fields loads correctly,
	 * when the fields in the entity class have default values
	 */
	@Test
	public void testNewVersionOfEntity() throws Exception {
		fact.register(EntityWithDefault.class);

		Objectify ofy = fact.begin();
		// e1 has absent properties
		Entity e1 = new Entity("EntityWithDefault");
		e1.setProperty("a", "1");
		com.google.appengine.api.datastore.Key k1 = ofy.getDatastore().put(e1);

		EntityWithDefault o = ofy.get(new Key<EntityWithDefault>(EntityWithDefault.class, k1.getId()));

		assert o.a != null;
		assert "1".equals(o.a);
		assert o.b != null;
		assert "foo".equals(o.b);
		assert o.c != null;
		assert "bar".equals(o.c);
		assert o.s != null;
		assert "default2".equals(o.s.s);
	}

	/**
	 * Test that writing null via Objectify is preserved, even when the fields have default values
	 */
	@Test
	public void testDefaultValuesAndNull() throws Exception {
		fact.register(EntityWithDefault.class);
		Objectify ofy = fact.begin();

		Key<EntityWithDefault> k1 = ofy.put(new EntityWithDefault("A"));
		Key<EntityWithDefault> k2 = ofy.put(new EntityWithDefault("A", "B", "C"));
		Key<EntityWithDefault> k3 = ofy.put(new EntityWithDefault("A", null, null));

		EntityWithDefault o1 = ofy.get(k1);
		EntityWithDefault o2 = ofy.get(k2);
		EntityWithDefault o3 = ofy.get(k3);

		assert "A".equals(o1.a);
		assert "foo".equals(o1.b);
		assert "bar".equals(o1.c);

		assert "A".equals(o2.a);
		assert "B".equals(o2.b);
		assert "C".equals(o2.c);

		assert "A".equals(o3.a);
		assert null == o3.b;
		assert null == o3.c;
	}
}