001/*
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *     http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.hadoop.hbase.hbtop.screen.top;
019
020import static org.hamcrest.CoreMatchers.is;
021import static org.hamcrest.MatcherAssert.assertThat;
022import static org.mockito.ArgumentMatchers.argThat;
023import static org.mockito.Mockito.verify;
024
025import java.util.ArrayList;
026import java.util.List;
027import java.util.stream.Collectors;
028import org.apache.hadoop.hbase.HBaseClassTestRule;
029import org.apache.hadoop.hbase.hbtop.RecordFilter;
030import org.apache.hadoop.hbase.hbtop.field.Field;
031import org.apache.hadoop.hbase.hbtop.field.FieldInfo;
032import org.apache.hadoop.hbase.hbtop.mode.Mode;
033import org.apache.hadoop.hbase.testclassification.SmallTests;
034import org.junit.Before;
035import org.junit.ClassRule;
036import org.junit.Test;
037import org.junit.experimental.categories.Category;
038import org.junit.runner.RunWith;
039import org.mockito.Mock;
040import org.mockito.junit.MockitoJUnitRunner;
041
042@Category(SmallTests.class)
043@RunWith(MockitoJUnitRunner.class)
044public class TestFilterDisplayModeScreenPresenter {
045
046  @ClassRule
047  public static final HBaseClassTestRule CLASS_RULE =
048    HBaseClassTestRule.forClass(TestFilterDisplayModeScreenPresenter.class);
049
050  @Mock
051  private FilterDisplayModeScreenView filterDisplayModeScreenView;
052
053  @Mock
054  private TopScreenView topScreenView;
055
056  private FilterDisplayModeScreenPresenter filterDisplayModeScreenPresenter;
057
058  @Before
059  public void setup() {
060    List<Field> fields =
061      Mode.REGION.getFieldInfos().stream().map(FieldInfo::getField).collect(Collectors.toList());
062
063    List<RecordFilter> filters = new ArrayList<>();
064    filters.add(RecordFilter.parse("NAMESPACE==namespace", fields, true));
065    filters.add(RecordFilter.parse("TABLE==table", fields, true));
066
067    filterDisplayModeScreenPresenter =
068      new FilterDisplayModeScreenPresenter(filterDisplayModeScreenView, filters, topScreenView);
069  }
070
071  @Test
072  public void testInit() {
073    filterDisplayModeScreenPresenter.init();
074    verify(filterDisplayModeScreenView).showFilters(argThat(
075      filters -> filters.size() == 2 && filters.get(0).toString().equals("NAMESPACE==namespace")
076        && filters.get(1).toString().equals("TABLE==table")));
077  }
078
079  @Test
080  public void testReturnToTopScreen() {
081    assertThat(filterDisplayModeScreenPresenter.returnToNextScreen(), is(topScreenView));
082  }
083}