/MongoTest/test/mongo/test/TestActivity.java

https://github.com/zhijun/fusion · Java · 118 lines · 77 code · 23 blank · 18 comment · 0 complexity · a67a1d587072343922ea0c5f0565f1da MD5 · raw file

  1. /**
  2. *
  3. */
  4. package mongo.test;
  5. import static org.junit.Assert.*;
  6. import java.util.ArrayList;
  7. import java.util.Date;
  8. import org.junit.After;
  9. import org.junit.Before;
  10. import org.junit.Test;
  11. import com.fusion.model.core.Activity;
  12. import com.fusion.model.core.Criteria;
  13. import com.fusion.model.core.Criterias;
  14. import com.fusion.model.core.Parties;
  15. import com.fusion.model.core.Party;
  16. import com.fusion.model.core.Status;
  17. import com.fusion.model.core.User;
  18. import com.fusion.model.instance.AllConfirmCriteria;
  19. import com.fusion.model.instance.AllConfirmFact;
  20. import com.google.code.morphia.Datastore;
  21. import com.google.code.morphia.Morphia;
  22. import com.mongodb.DB;
  23. import com.mongodb.Mongo;
  24. /**
  25. * @author zhijun
  26. *
  27. */
  28. public class TestActivity {
  29. Mongo mongo;
  30. DB db;
  31. /**
  32. * @throws java.lang.Exception
  33. */
  34. @Before
  35. public void setUp() throws Exception {
  36. mongo=new Mongo();
  37. db=mongo.getDB("fusion");
  38. db.getCollection("Activity").drop();
  39. db.getCollection("User").drop();
  40. }
  41. /**
  42. * @throws java.lang.Exception
  43. */
  44. @After
  45. public void tearDown() throws Exception {
  46. mongo.close();
  47. }
  48. @Test
  49. public void test() {
  50. Datastore mdb=new Morphia().map(Activity.class).createDatastore(mongo, "fusion");
  51. mdb.ensureIndexes();
  52. Activity act=new Activity();
  53. act.setTitle("My first activity");
  54. act.setMemo("搭顺风车万科魅力");
  55. act.setStatus(Status.pending);
  56. // mdb.save(act);
  57. // act=mdb.find(Activity.class).get();
  58. Parties parties=new Parties();
  59. User user=new User();
  60. user.setUid("zhangkevin");
  61. user.setCredit(null);
  62. user.setNickName("kevin zhang");
  63. user.setPoints(102);
  64. user.setRegisterDate(new Date());
  65. User user2=new User();
  66. user2.setUid("jiangjason");
  67. user2.setCredit(null);
  68. user2.setNickName("jiang shuai");
  69. user2.setPoints(107);
  70. ArrayList<Party> partyList=new ArrayList<Party>();
  71. Party party1=new Party(user);
  72. Party party2=new Party(user2);
  73. partyList.add(party1);
  74. partyList.add(party2);
  75. parties.setParties(partyList);
  76. //act.setCreator(user);
  77. act.setParties(parties);
  78. Criterias criterias=new Criterias();
  79. Criteria criteria=new AllConfirmCriteria();
  80. criteria.setActivity(act);
  81. AllConfirmFact fact= (AllConfirmFact)criteria.getFact();
  82. fact.confirm(party1);
  83. fact.confirm(party2);
  84. //fact.takeBack(party2);
  85. criterias.addCriteria(criteria);
  86. act.setCriterias(criterias);
  87. assertTrue(act.getCriterias().isMet());
  88. mdb.save(user);
  89. mdb.save(user2);
  90. mdb.save(act);
  91. Activity dbact=mdb.find(Activity.class).get();
  92. assertEquals(2, dbact.getParties().getParties().size());
  93. //assertTrue(dbact.getCriterias().isMet());
  94. }
  95. }