package net.sue445.service;

import static org.hamcrest.Matchers.*;
import static org.junit.Assert.*;

import java.util.Map;

import net.sue445.model.Slim3Model;

import org.junit.Before;
import org.junit.Test;
import org.slim3.datastore.Datastore;
import org.slim3.tester.AppEngineTestCase;

import com.google.appengine.api.datastore.Key;

public class SampleServiceTest extends AppEngineTestCase {

	private Key key1;
	private Key key2;
	private Key key3;

	@Before
	public void setUpModels(){
		key1 = Datastore.createKey(Slim3Model.class, "key1");
		key2 = Datastore.createKey(Slim3Model.class, "key2");
		key3 = Datastore.createKey(Slim3Model.class, "key3");

		Slim3Model model1 = new Slim3Model();
		model1.setKey(key1);
		Datastore.put(model1);

		Slim3Model model2 = new Slim3Model();
		model2.setKey(key2);
		Datastore.put(model2);
	}

	@Test
	public void getAsMap() throws Exception {
		// exercise
		Map<Key, Slim3Model> actual = Datastore.getAsMap(Slim3Model.class, key1, key2, key3);

		// assertion
		assertThat(actual.size(), is(2));

		assertThat(actual, hasKey(key1));
		assertThat(actual.get(key1).getKey(), is(key1));

		assertThat(actual, hasKey(key2));
		assertThat(actual.get(key2).getKey(), is(key2));

		assertThat(actual, not(hasKey(key3)));
		assertThat(actual.get(key3), is(nullValue()));
	}
}