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

http://thoughtsite.googlecode.com/ · Java · 91 lines · 61 code · 18 blank · 12 comment · 2 complexity · 39bd964723142f65d672ad69418d7832 MD5 · raw file

  1. // Copyright 2009 Google Inc. All Rights Reserved.
  2. /**
  3. *
  4. */
  5. package com.google.ie.business.dao.impl;
  6. import static org.junit.Assert.assertEquals;
  7. import com.google.appengine.api.datastore.DatastoreServiceFactory;
  8. import com.google.appengine.api.datastore.Query;
  9. import com.google.ie.business.domain.AdminRequest;
  10. import com.google.ie.business.domain.Idea;
  11. import com.google.ie.dto.RetrievalInfo;
  12. import com.google.ie.test.DatastoreTest;
  13. import org.junit.Test;
  14. import java.util.Date;
  15. /**
  16. * @author asirohi
  17. *
  18. */
  19. public class AdminRequestDaoImplTest extends DatastoreTest {
  20. private AdminRequestDaoImpl adminRequestDao;
  21. @Override
  22. public void setUp() {
  23. super.setUp();
  24. if (adminRequestDao == null) {
  25. adminRequestDao = new AdminRequestDaoImpl();
  26. adminRequestDao.setPersistenceManagerFactory(pmf);
  27. }
  28. }
  29. @Test
  30. public void addRequest() {
  31. AdminRequest adminRequest = new AdminRequest();
  32. adminRequest.setEntityKey("ideakey");
  33. adminRequest.setEntityType(Idea.class.getSimpleName());
  34. adminRequest.setRequesterkey("userkey");
  35. adminRequest.setRequestType(AdminRequest.REQUEST_OBJECTIONABLE);
  36. adminRequest.setCreatedOn(new Date());
  37. /* Get title of the idea */
  38. Idea idea = new Idea();
  39. idea.setTitle("Great Idea");
  40. adminRequest.setEntityTitle(idea.getTitle());
  41. adminRequestDao.saveRequest(adminRequest);
  42. Query query = new Query(AdminRequest.class.getSimpleName());
  43. assertEquals(1, DatastoreServiceFactory.getDatastoreService().prepare(query)
  44. .countEntities());
  45. }
  46. /**
  47. *
  48. */
  49. @Test
  50. public void getAllAdminRequests() {
  51. AdminRequest adminRequest = new AdminRequest();
  52. adminRequest.setEntityKey("ideakey");
  53. adminRequest.setEntityType(Idea.class.getSimpleName());
  54. adminRequest.setRequesterkey("userkey");
  55. adminRequest.setRequestType(AdminRequest.REQUEST_OBJECTIONABLE);
  56. adminRequest.setCreatedOn(new Date());
  57. adminRequest.setStatus(AdminRequest.STATUS_PENDING);
  58. adminRequest.setEntityTitle("idea title");
  59. adminRequestDao.saveRequest(adminRequest);
  60. AdminRequest adminRequest2 = new AdminRequest();
  61. adminRequest2.setEntityKey("ideakey");
  62. adminRequest2.setEntityType(Idea.class.getSimpleName());
  63. adminRequest2.setRequesterkey("userkey");
  64. adminRequest2.setRequestType(AdminRequest.REQUEST_OBJECTIONABLE);
  65. adminRequest2.setCreatedOn(new Date());
  66. adminRequest.setStatus(AdminRequest.STATUS_REJECTED);
  67. adminRequest2.setEntityTitle("idea title2");
  68. adminRequestDao.saveRequest(adminRequest2);
  69. RetrievalInfo retrievalInfo = this.createDummyRetrievalParam(0, 10, "createdOn", "asc");
  70. adminRequestDao.getAllAdminRequests(retrievalInfo);
  71. assertEquals(1, adminRequestDao.getAllAdminRequests(retrievalInfo).size());
  72. }
  73. }