1/*
2 * Copyright 2013-present Facebook, Inc.
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 <folly/AtomicBitSet.h>
18
19#include <folly/portability/GTest.h>
20
21#include <glog/logging.h>
22
23namespace folly {
24namespace test {
25
26TEST(AtomicBitSet, Simple) {
27 constexpr size_t kSize = 1000;
28 AtomicBitSet<kSize> bs;
29
30 EXPECT_EQ(kSize, bs.size());
31
32 for (size_t i = 0; i < kSize; ++i) {
33 EXPECT_FALSE(bs[i]);
34 }
35
36 bs.set(42);
37 for (size_t i = 0; i < kSize; ++i) {
38 EXPECT_EQ(i == 42, bs[i]);
39 }
40
41 bs.set(43);
42 for (size_t i = 0; i < kSize; ++i) {
43 EXPECT_EQ((i == 42 || i == 43), bs[i]);
44 }
45
46 bs.reset(42);
47 for (size_t i = 0; i < kSize; ++i) {
48 EXPECT_EQ((i == 43), bs[i]);
49 }
50
51 bs.reset(43);
52 for (size_t i = 0; i < kSize; ++i) {
53 EXPECT_FALSE(bs[i]);
54 }
55}
56
57} // namespace test
58} // namespace folly
59
60int main(int argc, char* argv[]) {
61 testing::InitGoogleTest(&argc, argv);
62 gflags::ParseCommandLineFlags(&argc, &argv, true);
63 return RUN_ALL_TESTS();
64}
65