1 files changed, 111 insertions, 0 deletions
diff --git a/third_party/libwebrtc/sdk/objc/unittests/scoped_cftyperef_tests.mm b/third_party/libwebrtc/sdk/objc/unittests/scoped_cftyperef_tests.mm
new file mode 100644
index 0000000000..a354410ede
--- /dev/null
+++ b/third_party/libwebrtc/sdk/objc/unittests/scoped_cftyperef_tests.mm
@@ -0,0 +1,111 @@
+/*
+ *  Copyright 2017 The WebRTC project authors. All Rights Reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#import <XCTest/XCTest.h>
+
+#include "sdk/objc/helpers/scoped_cftyperef.h"
+
+namespace {
+struct TestType {
+  TestType() : has_value(true) {}
+  TestType(bool b) : has_value(b) {}
+  explicit operator bool() { return has_value; }
+  bool has_value;
+  int retain_count = 0;
+};
+
+typedef TestType* TestTypeRef;
+
+struct TestTypeTraits {
+  static TestTypeRef InvalidValue() { return TestTypeRef(false); }
+  static void Release(TestTypeRef t) { t->retain_count--; }
+  static TestTypeRef Retain(TestTypeRef t) {
+    t->retain_count++;
+    return t;
+  }
+};
+}  // namespace
+
+using ScopedTestType = rtc::internal::ScopedTypeRef<TestTypeRef, TestTypeTraits>;
+
+// In these tests we sometime introduce variables just to
+// observe side-effects. Ignore the compilers complaints.
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Wunused-variable"
+
+@interface ScopedTypeRefTests : XCTestCase
+@end
+
+@implementation ScopedTypeRefTests
+
+- (void)testShouldNotRetainByDefault {
+  TestType a;
+  ScopedTestType ref(&a);
+  XCTAssertEqual(0, a.retain_count);
+}
+
+- (void)testShouldRetainWithPolicy {
+  TestType a;
+  ScopedTestType ref(&a, rtc::RetainPolicy::RETAIN);
+  XCTAssertEqual(1, a.retain_count);
+}
+
+- (void)testShouldReleaseWhenLeavingScope {
+  TestType a;
+  XCTAssertEqual(0, a.retain_count);
+  {
+    ScopedTestType ref(&a, rtc::RetainPolicy::RETAIN);
+    XCTAssertEqual(1, a.retain_count);
+  }
+  XCTAssertEqual(0, a.retain_count);
+}
+
+- (void)testShouldBeCopyable {
+  TestType a;
+  XCTAssertEqual(0, a.retain_count);
+  {
+    ScopedTestType ref1(&a, rtc::RetainPolicy::RETAIN);
+    XCTAssertEqual(1, a.retain_count);
+    ScopedTestType ref2 = ref1;
+    XCTAssertEqual(2, a.retain_count);
+  }
+  XCTAssertEqual(0, a.retain_count);
+}
+
+- (void)testCanReleaseOwnership {
+  TestType a;
+  XCTAssertEqual(0, a.retain_count);
+  {
+    ScopedTestType ref(&a, rtc::RetainPolicy::RETAIN);
+    XCTAssertEqual(1, a.retain_count);
+    TestTypeRef b = ref.release();
+  }
+  XCTAssertEqual(1, a.retain_count);
+}
+
+- (void)testShouldBeTestableForTruthiness {
+  ScopedTestType ref;
+  XCTAssertFalse(ref);
+  TestType a;
+  ref = &a;
+  XCTAssertTrue(ref);
+  ref.release();
+  XCTAssertFalse(ref);
+}
+
+- (void)testShouldProvideAccessToWrappedType {
+  TestType a;
+  ScopedTestType ref(&a);
+  XCTAssertEqual(&(a.retain_count), &(ref->retain_count));
+}
+
+@end
+
+#pragma clang diagnostic pop