adding test cases based on context

Change-Id: I14a9037f023a00ed97d3343970eb52c964abf522
intuit · Oct 13, 2016 · 822b59c · 822b59c
1 parent 77d7e79
commit 822b59c
Showing 1 changed file with 38 additions and 1 deletion.
diff --git a/.../java/com/intuit/wasabi/tests/service/assignment/EventSubmissionBeforeAssignmentTest.java b/.../java/com/intuit/wasabi/tests/service/assignment/EventSubmissionBeforeAssignmentTest.java
@@ -68,15 +68,52 @@ public void testSetUp() {
 
 
     @Test
-    public void createImpression() {
+    public void createDefaultImpression() {
         //create user
         User user = UserFactory.createUser();
 
         //create an event of type IMPRESSION and post it to the event endpoint
         Event impression = EventFactory.createImpression();
         postEvent(impression, experiment, user, HttpStatus.SC_NOT_FOUND);
     }
+
+
+    @Test
+    public void createImpressionWithContextSetForProd() {
+        //create user
+        User user = UserFactory.createUser();
+
+        //create an event of type IMPRESSION and post it to the event endpoint
+        Event impression = EventFactory.createImpression();
+        impression.context="prod";
+        postEvent(impression, experiment, user, HttpStatus.SC_NOT_FOUND);
+    }
+
+    @Test
+    public void createImpressionWithContextSetForQA() {
+        //create user
+        User user = UserFactory.createUser();
 
+
+        //create an event of type IMPRESSION and post it to the event endpoint
+        Event impression = EventFactory.createImpression();
+        impression.context="QA";
+        postEvent(impression, experiment, user, HttpStatus.SC_NOT_FOUND);
+    }
+
+
+    @Test
+    public void createImpressionWithContextSetForBothQAAndProd() {
+        //create user
+        User user = UserFactory.createUser();
+
+        //create an event of type IMPRESSION and post it to the event endpoint
+        Event impression = EventFactory.createImpression();
+        impression.context="QA";
+        postEvent(impression, experiment, user, HttpStatus.SC_NOT_FOUND);
+        impression.context ="PROD";
+        postEvent(impression, experiment, user, HttpStatus.SC_CREATED);
+    }
 
     @AfterClass
     public void testCleanUp() {