/src/test/java/com/google/ie/business/dao/impl/VoteDaoImplTest.java

http://thoughtsite.googlecode.com/ · Java · 65 lines · 45 code · 15 blank · 5 comment · 2 complexity · 4cc55da396482b0213e8d420176390c6 MD5 · raw file

  1. package com.google.ie.business.dao.impl;
  2. import static org.junit.Assert.assertEquals;
  3. import com.google.appengine.api.datastore.DatastoreServiceFactory;
  4. import com.google.appengine.api.datastore.Query;
  5. import com.google.ie.business.domain.IdeaVote;
  6. import com.google.ie.test.DatastoreTest;
  7. import org.junit.Before;
  8. import org.junit.Test;
  9. /**
  10. * Test case for VoteDaoImpl class.
  11. *
  12. * @author gmaurya
  13. */
  14. public class VoteDaoImplTest extends DatastoreTest {
  15. private VoteDaoImpl voteDao;
  16. @Before
  17. public void setUp() {
  18. super.setUp();
  19. if (voteDao == null) {
  20. voteDao = new VoteDaoImpl();
  21. voteDao.setPersistenceManagerFactory(pmf);
  22. }
  23. }
  24. @Test
  25. public void addVote() {
  26. IdeaVote vote = new IdeaVote();
  27. vote.setCreatorKey("creatorKey");
  28. vote.setIdeaKey("ideaKey");
  29. vote.setVotePoints(15);
  30. vote.setPositiveVote(true);
  31. voteDao.saveVote(vote);
  32. Query query = new Query(IdeaVote.class.getSimpleName());
  33. assertEquals(1, DatastoreServiceFactory.getDatastoreService().prepare(query)
  34. .countEntities());
  35. }
  36. @Test
  37. public void isIdeaAlreadyVotedByUser() {
  38. IdeaVote vote = new IdeaVote();
  39. vote.setCreatorKey("userKey");
  40. vote.setIdeaKey("ideaKey");
  41. vote.setVotePoints(15);
  42. vote.setPositiveVote(true);
  43. voteDao.saveVote(vote);
  44. assertEquals(true, voteDao.isIdeaAlreadyVotedByUser("userKey", "ideaKey"));
  45. }
  46. @Test
  47. public void ideaNotVotedByUser() {
  48. assertEquals(false, voteDao.isIdeaAlreadyVotedByUser("userKey", "ideaKey"));
  49. }
  50. }