Home | History | Annotate | Download | only in filesystem 
      1 /*
      2  * Copyright (C) 2018 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #include "src/traced/probes/filesystem/range_tree.h"
     18 #include "src/traced/probes/filesystem/prefix_finder.h"
     19 
     20 #include <fcntl.h>
     21 #include <stdio.h>
     22 #include <stdlib.h>
     23 #include <sys/stat.h>
     24 
     25 #include "gmock/gmock.h"
     26 #include "gtest/gtest.h"
     27 #include "perfetto/base/build_config.h"
     28 #include "perfetto/base/scoped_file.h"
     29 #include "perfetto/base/temp_file.h"
     30 #include "perfetto/base/utils.h"
     31 
     32 namespace perfetto {
     33 namespace {
     34 
     35 using ::testing::Contains;
     36 using ::testing::Not;
     37 
     38 TEST(RangeTreeTest, Basic) {
     39   PrefixFinder pr(1);
     40   pr.AddPath("/a/foo");
     41   pr.AddPath("/b/foo");
     42   pr.AddPath("/c/foo");
     43   pr.AddPath("/d/foo");
     44   pr.Finalize();
     45 
     46   auto a = pr.GetPrefix("/a/foo");
     47   auto b = pr.GetPrefix("/b/foo");
     48   auto c = pr.GetPrefix("/c/foo");
     49   auto d = pr.GetPrefix("/d/foo");
     50 
     51   ASSERT_EQ(a->ToString(), "/a");
     52   ASSERT_EQ(b->ToString(), "/b");
     53   ASSERT_EQ(c->ToString(), "/c");
     54   ASSERT_EQ(d->ToString(), "/d");
     55 
     56   // This test needs to be changed for other kSetSize.
     57   ASSERT_EQ(kSetSize, 3u);
     58 
     59   RangeTree t;
     60   t.Insert(1, a);
     61   t.Insert(2, a);
     62   t.Insert(20, b);
     63   t.Insert(24, a);
     64   t.Insert(25, c);
     65   t.Insert(27, d);
     66 
     67   EXPECT_THAT(t.Get(1), Contains("/a"));
     68   EXPECT_THAT(t.Get(2), Contains("/a"));
     69   EXPECT_THAT(t.Get(20), Contains("/b"));
     70   EXPECT_THAT(t.Get(24), Contains("/a"));
     71   EXPECT_THAT(t.Get(25), Contains("/c"));
     72   EXPECT_THAT(t.Get(27), Contains("/d"));
     73   // 27 will have overflowed kSetSize = 3;
     74   EXPECT_THAT(t.Get(27), Not(Contains("/a")));
     75   EXPECT_THAT(t.Get(27), Not(Contains("/b")));
     76   EXPECT_THAT(t.Get(27), Not(Contains("/c")));
     77 }
     78 
     79 }  // namespace
     80 }  // namespace perfetto
     81