| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License |
| */ |
| |
| package org.apache.unomi.itests; |
| |
| import org.apache.unomi.api.Event; |
| import org.apache.unomi.api.Profile; |
| import org.apache.unomi.api.rules.Rule; |
| import org.apache.unomi.api.services.EventService; |
| import org.apache.unomi.api.services.ProfileService; |
| import org.apache.unomi.api.services.RulesService; |
| import org.apache.unomi.persistence.spi.CustomObjectMapper; |
| import org.apache.unomi.plugins.baseplugin.actions.UpdatePropertiesAction; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.ops4j.pax.exam.junit.PaxExam; |
| import org.ops4j.pax.exam.spi.reactors.ExamReactorStrategy; |
| import org.ops4j.pax.exam.spi.reactors.PerSuite; |
| import org.ops4j.pax.exam.util.Filter; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import javax.inject.Inject; |
| import java.io.IOException; |
| import java.text.SimpleDateFormat; |
| import java.time.LocalDateTime; |
| import java.time.ZoneId; |
| import java.util.ArrayList; |
| import java.util.Date; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Objects; |
| import java.util.TimeZone; |
| |
| /** |
| * Created by amidani on 12/10/2017. |
| */ |
| |
| @RunWith(PaxExam.class) |
| @ExamReactorStrategy(PerSuite.class) |
| public class PropertiesUpdateActionIT extends BaseIT { |
| private final static Logger LOGGER = LoggerFactory.getLogger(PropertiesUpdateActionIT.class); |
| |
| private final static String PROFILE_TARGET_TEST_ID = "profile-target-event"; |
| private final static String PROFILE_TEST_ID = "profile-to-update-by-event"; |
| |
| @Inject |
| @Filter(timeout = 600000) |
| protected ProfileService profileService; |
| @Inject |
| @Filter(timeout = 600000) |
| protected EventService eventService; |
| @Inject |
| @Filter(timeout = 600000) |
| protected RulesService rulesService; |
| |
| @Before |
| public void setUp() throws InterruptedException { |
| Profile profile = new Profile(); |
| profile.setItemId(PROFILE_TEST_ID); |
| profile.setProperties(new HashMap<>()); |
| profile.setProperty("lastName", "Jose"); // property that have a propertyType registered in the system |
| profile.setProperty("prop4", "New property 4"); // property that do not have a propertyType registered in the system |
| profileService.save(profile); |
| LOGGER.info("Profile saved with ID [{}].", profile.getItemId()); |
| |
| Profile profileTarget = new Profile(); |
| profileTarget.setItemId(PROFILE_TARGET_TEST_ID); |
| profileService.save(profileTarget); |
| LOGGER.info("Profile saved with ID [{}].", profileTarget.getItemId()); |
| |
| keepTrying("Profile " + PROFILE_TEST_ID + " not found in the required time", () -> profileService.load(PROFILE_TEST_ID), |
| Objects::nonNull, DEFAULT_TRYING_TIMEOUT, DEFAULT_TRYING_TRIES); |
| keepTrying("Profile " + PROFILE_TARGET_TEST_ID + " not found in the required time", |
| () -> profileService.load(PROFILE_TARGET_TEST_ID), Objects::nonNull, DEFAULT_TRYING_TIMEOUT, DEFAULT_TRYING_TRIES); |
| } |
| |
| @Test |
| public void testUpdateProperties_CurrentProfile() { |
| Profile profile = profileService.load(PROFILE_TARGET_TEST_ID); |
| Assert.assertNull(profile.getProperty("firstName")); |
| |
| Event updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| |
| Map<String, Object> propertyToUpdate = new HashMap<>(); |
| propertyToUpdate.put("properties.firstName", "UPDATED FIRST NAME CURRENT PROFILE"); |
| |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_UPDATE, propertyToUpdate); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TARGET_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| |
| int changes = eventService.send(updateProperties); |
| |
| LOGGER.info("Changes of the event : {}", changes); |
| |
| Assert.assertTrue(changes > 0); |
| Assert.assertEquals("UPDATED FIRST NAME CURRENT PROFILE", profile.getProperty("firstName")); |
| } |
| |
| @Test |
| public void testUpdateProperties_NotCurrentProfile() { |
| Profile profile = profileService.load(PROFILE_TARGET_TEST_ID); |
| Profile profileToUpdate = profileService.load(PROFILE_TEST_ID); |
| Assert.assertNull(profileToUpdate.getProperty("firstName")); |
| |
| Event updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| |
| Map<String, Object> propertyToUpdate = new HashMap<>(); |
| propertyToUpdate.put("properties.firstName", "UPDATED FIRST NAME"); |
| |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_UPDATE, propertyToUpdate); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| eventService.send(updateProperties); |
| |
| profileToUpdate = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("UPDATED FIRST NAME", profileToUpdate.getProperty("firstName")); |
| } |
| |
| @Test |
| public void testUpdateProperties_CurrentProfile_PROPS_TO_ADD() throws InterruptedException { |
| Profile profile = profileService.load(PROFILE_TEST_ID); |
| |
| Event updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| |
| Map<String, Object> propertyToAdd = new HashMap<>(); |
| propertyToAdd.put("properties.prop1", "New property 1"); |
| propertyToAdd.put("properties.prop2", "New property 2"); |
| propertyToAdd.put("properties.prop3", "New property 3"); |
| propertyToAdd.put("properties.prop4", "Updated property 4"); |
| propertyToAdd.put("properties.lastName", "Michel"); |
| |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_ADD, propertyToAdd); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| eventService.send(updateProperties); |
| profileService.save(profile); |
| refreshPersistence(); |
| |
| profile = keepTrying("Props_to_add should set the prop if it's missing", () -> profileService.load(PROFILE_TEST_ID), |
| loadedProfile -> "New property 1".equals(loadedProfile.getProperty("prop1")), DEFAULT_TRYING_TIMEOUT, DEFAULT_TRYING_TRIES); |
| |
| Assert.assertEquals("Props_to_add should set the prop if it's missing", "New property 2", profile.getProperty("prop2")); |
| Assert.assertEquals("Props_to_add should set the prop if it's missing", "New property 3", profile.getProperty("prop3")); |
| Assert.assertEquals("Props_to_add should not override existing prop", "New property 4", profile.getProperty("prop4")); |
| Assert.assertEquals("Props_to_add should not override existing prop", "Jose", profile.getProperty("lastName")); |
| } |
| |
| @Test |
| public void testUpdateProperties_CurrentProfile_PROPS_TO_ADD_TO_SET() throws InterruptedException { |
| Profile profile = profileService.load(PROFILE_TEST_ID); |
| Event updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| |
| Map<String, Object> propertyToAddToSet = new HashMap<>(); |
| propertyToAddToSet.put("properties.prop1", "New property 1"); |
| propertyToAddToSet.put("properties.prop2", "New property 2"); |
| propertyToAddToSet.put("properties.prop3", "New property 3"); |
| |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_ADD, propertyToAddToSet); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| eventService.send(updateProperties); |
| profileService.save(profile); |
| refreshPersistence(); |
| |
| profile = keepTrying("prop1 not updated", () -> profileService.load(PROFILE_TEST_ID), |
| loadedProfile -> "New property 1".equals(loadedProfile.getProperty("prop1")), DEFAULT_TRYING_TIMEOUT, DEFAULT_TRYING_TRIES); |
| |
| Assert.assertEquals("New property 1", profile.getProperty("prop1")); |
| Assert.assertEquals("New property 2", profile.getProperty("prop2")); |
| Assert.assertEquals("New property 3", profile.getProperty("prop3")); |
| |
| // Add set and check |
| propertyToAddToSet = new HashMap<>(); |
| propertyToAddToSet.put("properties.prop1", "New property 1 bis"); |
| propertyToAddToSet.put("properties.prop3", "New property 3 bis"); |
| |
| updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_ADD_TO_SET, propertyToAddToSet); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| eventService.send(updateProperties); |
| profileService.save(profile); |
| refreshPersistence(); |
| |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals(2, ((List<String>) profile.getProperty("prop1")).size()); |
| Assert.assertEquals(2, ((List<String>) profile.getProperty("prop3")).size()); |
| Assert.assertEquals("New property 1", ((List<String>) profile.getProperty("prop1")).get(1)); |
| Assert.assertEquals("New property 2", profile.getProperty("prop2")); |
| Assert.assertEquals("New property 3 bis", ((List<String>) profile.getProperty("prop3")).get(0)); |
| } |
| |
| @Test |
| public void testUpdateProperties_CurrentProfile_PROPS_TO_DELETE() throws InterruptedException { |
| Profile profile = profileService.load(PROFILE_TEST_ID); |
| Event updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| |
| Map<String, Object> propertyToAdd = new HashMap<>(); |
| propertyToAdd.put("properties.prop1", "New property 1"); |
| propertyToAdd.put("properties.prop1bis", "New property 1 bis"); |
| propertyToAdd.put("properties.prop2", "New property 2"); |
| propertyToAdd.put("properties.prop3", "New property 3"); |
| |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_ADD, propertyToAdd); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| eventService.send(updateProperties); |
| profileService.save(profile); |
| refreshPersistence(); |
| |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("New property 1", profile.getProperty("prop1")); |
| Assert.assertEquals("New property 1 bis", profile.getProperty("prop1bis")); |
| Assert.assertEquals("New property 2", profile.getProperty("prop2")); |
| Assert.assertEquals("New property 3", profile.getProperty("prop3")); |
| |
| // Delete property and check |
| List<String> propertyToDelete = new ArrayList<>(); |
| propertyToDelete.add("properties.prop1bis"); |
| |
| updateProperties = new Event("updateProperties", null, profile, null, null, null, new Date()); |
| updateProperties.setPersistent(false); |
| updateProperties.setProperty(UpdatePropertiesAction.PROPS_TO_DELETE, propertyToDelete); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_ID_KEY, PROFILE_TEST_ID); |
| updateProperties.setProperty(UpdatePropertiesAction.TARGET_TYPE_KEY, "profile"); |
| |
| eventService.send(updateProperties); |
| profileService.save(profile); |
| refreshPersistence(); |
| |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertNull(profile.getProperty("prop1bis")); |
| Assert.assertEquals("New property 1", profile.getProperty("prop1")); |
| Assert.assertEquals("New property 2", profile.getProperty("prop2")); |
| Assert.assertEquals("New property 3", profile.getProperty("prop3")); |
| } |
| |
| @Test |
| public void testVisitsDatePropertiesUpdate() throws InterruptedException { |
| SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'"); |
| dateFormat.setTimeZone(TimeZone.getTimeZone("UTC")); |
| |
| Profile profile = profileService.load(PROFILE_TEST_ID); |
| |
| // Test init of lastVisit and firstVisit |
| Date eventTimeStamp = new Date(); |
| String eventTimeStamp1 = dateFormat.format(eventTimeStamp); |
| Event sessionReassigned = new Event("sessionReassigned", null, profile, null, null, profile, eventTimeStamp); |
| sessionReassigned.setPersistent(false); |
| eventService.send(sessionReassigned); |
| profileService.save(profile); |
| refreshPersistence(); |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("lastVisit should be updated", eventTimeStamp1, profile.getProperty("lastVisit")); |
| Assert.assertEquals("firstVisit should be updated", eventTimeStamp1, profile.getProperty("firstVisit")); |
| Assert.assertNull("previousVisit should be null", profile.getProperty("previousVisit")); |
| |
| // test event dated -1 day: should update only firstVisit |
| eventTimeStamp = new Date(); |
| LocalDateTime ldt = LocalDateTime.ofInstant(eventTimeStamp.toInstant(), ZoneId.systemDefault()); |
| ldt = ldt.minusDays(1); |
| eventTimeStamp = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant()); |
| String eventTimeStamp2 = dateFormat.format(eventTimeStamp); |
| sessionReassigned = new Event("sessionReassigned", null, profile, null, null, profile, eventTimeStamp); |
| sessionReassigned.setPersistent(false); |
| eventService.send(sessionReassigned); |
| profileService.save(profile); |
| refreshPersistence(); |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("lastVisit should not be updated", eventTimeStamp1, profile.getProperty("lastVisit")); |
| Assert.assertEquals("firstVisit should be updated", eventTimeStamp2, profile.getProperty("firstVisit")); |
| Assert.assertNull("previousVisit should be null", profile.getProperty("previousVisit")); |
| |
| // test event dated +1 day: should update only lastVisit and previousVisit |
| eventTimeStamp = new Date(); |
| ldt = LocalDateTime.ofInstant(eventTimeStamp.toInstant(), ZoneId.systemDefault()); |
| ldt = ldt.plusDays(1); |
| eventTimeStamp = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant()); |
| String eventTimeStamp3 = dateFormat.format(eventTimeStamp); |
| sessionReassigned = new Event("sessionReassigned", null, profile, null, null, profile, eventTimeStamp); |
| sessionReassigned.setPersistent(false); |
| eventService.send(sessionReassigned); |
| profileService.save(profile); |
| refreshPersistence(); |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("lastVisit should be updated", eventTimeStamp3, profile.getProperty("lastVisit")); |
| Assert.assertEquals("firstVisit should not be updated", eventTimeStamp2, profile.getProperty("firstVisit")); |
| Assert.assertEquals("previousVisit should be updated", eventTimeStamp1, profile.getProperty("previousVisit")); |
| |
| // test event dated +5 hours: should update only previousVisit |
| eventTimeStamp = new Date(); |
| ldt = LocalDateTime.ofInstant(eventTimeStamp.toInstant(), ZoneId.systemDefault()); |
| ldt = ldt.plusHours(5); |
| eventTimeStamp = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant()); |
| String eventTimeStamp4 = dateFormat.format(eventTimeStamp); |
| sessionReassigned = new Event("sessionReassigned", null, profile, null, null, profile, eventTimeStamp); |
| sessionReassigned.setPersistent(false); |
| eventService.send(sessionReassigned); |
| profileService.save(profile); |
| refreshPersistence(); |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("lastVisit should not be updated", eventTimeStamp3, profile.getProperty("lastVisit")); |
| Assert.assertEquals("firstVisit should not be updated", eventTimeStamp2, profile.getProperty("firstVisit")); |
| Assert.assertEquals("previousVisit should be updated", eventTimeStamp4, profile.getProperty("previousVisit")); |
| } |
| |
| @Test |
| public void testSetPropertyActionDates() throws InterruptedException, IOException { |
| SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'"); |
| dateFormat.setTimeZone(TimeZone.getTimeZone("UTC")); |
| |
| // register test rule |
| Rule rule = CustomObjectMapper.getObjectMapper() |
| .readValue(getValidatedBundleJSON("testSetPropertyActionRule.json", new HashMap<>()), Rule.class); |
| createAndWaitForRule(rule); |
| |
| try { |
| Profile profile = profileService.load(PROFILE_TEST_ID); |
| Date eventTimeStamp = new Date(); |
| String eventTimeStamp1 = dateFormat.format(eventTimeStamp); |
| Event sessionReassigned = new Event("sessionReassigned", null, profile, null, null, profile, eventTimeStamp); |
| sessionReassigned.setPersistent(false); |
| Thread.sleep(4000); // small sleep to create time dif between eventTimeStamp and current system date |
| eventService.send(sessionReassigned); |
| profileService.save(profile); |
| refreshPersistence(); |
| profile = profileService.load(PROFILE_TEST_ID); |
| Assert.assertEquals("currentEventTimeStamp should be the exact date of the event timestamp", eventTimeStamp1, |
| profile.getProperty("currentEventTimeStamp")); |
| Assert.assertNotEquals("currentDate should be the current system date", eventTimeStamp1, profile.getProperty("currentDate")); |
| Assert.assertNotNull("currentDate should be set", profile.getProperty("currentDate")); |
| } finally { |
| rulesService.removeRule(rule.getItemId()); |
| } |
| } |
| } |