source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2016 SonarSource SA
   4  * mailto:contact AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.server.activity.index;
  21
  22 import java.util.Map;
  23 import org.assertj.core.data.MapEntry;
  24 import org.elasticsearch.action.update.UpdateRequest;
  25 import org.junit.Rule;
  26 import org.junit.Test;
  27 import org.sonar.api.utils.DateUtils;
  28 import org.sonar.api.utils.System2;
  29 import org.sonar.db.DbTester;
  30
  31 import static org.assertj.core.api.Assertions.assertThat;
  32
  33
  34 public class ActivityResultSetIteratorTest {
  35
  36   @Rule
  37   public DbTester dbTester = DbTester.create(System2.INSTANCE);
  38
  39   /**
  40    * Iterate over two rows in table.
  41    */
  42   @Test
  43   public void traverse() {
  44     dbTester.prepareDbUnit(getClass(), "traverse.xml");
  45     ActivityResultSetIterator it = ActivityResultSetIterator.create(dbTester.getDbClient(), dbTester.getSession(), 0L);
  46
  47     assertThat(it.hasNext()).isTrue();
  48     UpdateRequest request = it.next();
  49     Map<String, Object> doc = request.doc().sourceAsMap();
  50     assertThat(doc.get(ActivityIndexDefinition.FIELD_KEY)).isEqualTo("UUID1");
  51     assertThat(doc.get(ActivityIndexDefinition.FIELD_ACTION)).isEqualTo("THE_ACTION");
  52     assertThat(doc.get(ActivityIndexDefinition.FIELD_MESSAGE)).isEqualTo("THE_MSG");
  53     assertThat((Map) doc.get(ActivityIndexDefinition.FIELD_DETAILS)).containsOnly(MapEntry.entry("foo", "bar"));
  54     assertThat(doc.get(ActivityIndexDefinition.FIELD_LOGIN)).isEqualTo("THE_AUTHOR");
  55
  56     assertThat(it.hasNext()).isTrue();
  57     assertThat(it.next()).isNotNull();
  58     assertThat(it.hasNext()).isFalse();
  59     it.close();
  60
  61     assertThat(formatLongDate(it.getMaxRowDate())).startsWith("2015-01-01");
  62   }
  63
  64   @Test
  65   public void traverse_after_date() {
  66     dbTester.prepareDbUnit(getClass(), "traverse.xml");
  67     ActivityResultSetIterator it = ActivityResultSetIterator.create(dbTester.getDbClient(), dbTester.getSession(), DateUtils.parseDate("2014-12-01").getTime());
  68
  69     assertThat(it.hasNext()).isTrue();
  70     UpdateRequest request = it.next();
  71     assertThat(request).isNotNull();
  72     Map<String, Object> doc = request.doc().sourceAsMap();
  73     assertThat(doc.get(ActivityIndexDefinition.FIELD_KEY)).isEqualTo("UUID2");
  74
  75     assertThat(it.hasNext()).isFalse();
  76     it.close();
  77
  78     assertThat(formatLongDate(it.getMaxRowDate())).startsWith("2015-01-01");
  79   }
  80
  81   @Test
  82   public void nothing_to_traverse() {
  83     dbTester.prepareDbUnit(getClass(), "traverse.xml");
  84     ActivityResultSetIterator it = ActivityResultSetIterator.create(dbTester.getDbClient(), dbTester.getSession(), DateUtils.parseDate("2030-01-01").getTime());
  85
  86     assertThat(it.hasNext()).isFalse();
  87     it.close();
  88
  89     assertThat(it.getMaxRowDate()).isEqualTo(0L);
  90   }
  91
  92   private String formatLongDate(long dateInMs) {
  93     return DateUtils.formatDateTime(DateUtils.longToDate(dateInMs));
  94   }
  95 }