1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
- /*
- * Copyright 2013-present Facebook, Inc.
- *
- * Licensed 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.
- */
- #include <folly/synchronization/AtomicStruct.h>
- #include <folly/portability/GTest.h>
- using namespace folly;
- struct TwoBy32 {
- uint32_t left;
- uint32_t right;
- };
- TEST(AtomicStruct, two_by_32) {
- AtomicStruct<TwoBy32> a(TwoBy32{10, 20});
- TwoBy32 av = a;
- EXPECT_EQ(av.left, 10);
- EXPECT_EQ(av.right, 20);
- EXPECT_TRUE(a.compare_exchange_strong(av, TwoBy32{30, 40}));
- EXPECT_FALSE(a.compare_exchange_weak(av, TwoBy32{31, 41}));
- EXPECT_EQ(av.left, 30);
- EXPECT_TRUE(a.is_lock_free());
- auto b = a.exchange(TwoBy32{50, 60});
- EXPECT_EQ(b.left, 30);
- EXPECT_EQ(b.right, 40);
- EXPECT_EQ(a.load().left, 50);
- a = TwoBy32{70, 80};
- EXPECT_EQ(a.load().right, 80);
- a.store(TwoBy32{90, 100});
- av = a;
- EXPECT_EQ(av.left, 90);
- AtomicStruct<TwoBy32> c;
- c = b;
- EXPECT_EQ(c.load().right, 40);
- }
- template <size_t I>
- struct S {
- char x[I];
- };
- TEST(AtomicStruct, size_selection) {
- EXPECT_EQ(sizeof(AtomicStruct<S<1>>), 1);
- EXPECT_EQ(sizeof(AtomicStruct<S<2>>), 2);
- EXPECT_EQ(sizeof(AtomicStruct<S<3>>), 4);
- EXPECT_EQ(sizeof(AtomicStruct<S<4>>), 4);
- EXPECT_EQ(sizeof(AtomicStruct<S<5>>), 8);
- EXPECT_EQ(sizeof(AtomicStruct<S<6>>), 8);
- EXPECT_EQ(sizeof(AtomicStruct<S<7>>), 8);
- EXPECT_EQ(sizeof(AtomicStruct<S<8>>), 8);
- }
|