Tests.

Author: BB9z

README.md

@@ -6,9 +6,9 @@
 [![gitee 镜像](https://img.shields.io/badge/%E9%95%9C%E5%83%8F-gitee-C61E22.svg?style=flat-square)](https://gitee.com/b9swift/AssociatedObject)
 [![GitHub Source](https://img.shields.io/badge/Source-GitHub-24292F.svg?style=flat-square)](https://github.com/b9swift/AssociatedObject)
 
-Objective-C associated object 的 Swift 封装。
+方便地在 Swift 中利用 Objective-C 关联对象。
 
-Objective-C associated value wrapper.
+Objective-C associated value wrapper in Swift.
 
 ## 集成
 
@@ -24,9 +24,15 @@ Using Swift Package Manager or import manually.
 
 ## 使用
 
+定义扩展属性可以像下面这样，不仅支持 Objective-C 对象，各种 Swift 类型也支持。
+
 ## Usage
 
+You can define extended properties like below. All kinds of Swift types are also supported, not only Objective-C objects.
+
 ```swift
+import B9AssociatedObject
+
 private let fooAssociation = AssociatedObject<String>()
 extension SomeObject {
     var foo: String? {

Sources/B9AssociatedObject/B9AssociatedObject.swift

@@ -1,5 +1,5 @@
 /*
- AssociatedObject.swift
+ B9AssociatedObject.swift
 
  Copyright © 2020-2021 RFUI.
  https://github.com/b9swift/AssociatedObject
@@ -37,7 +37,11 @@ public final class AssociatedObject<T> {
     /// Accesses the associated value.
     /// - Parameter index: The source object for the association.
     public subscript(index: AnyObject) -> T? {
-        get { objc_getAssociatedObject(index, Unmanaged.passUnretained(self).toOpaque()) as? T }
-        set { objc_setAssociatedObject(index, Unmanaged.passUnretained(self).toOpaque(), newValue, policy) }
+        get {
+            objc_getAssociatedObject(index, Unmanaged.passUnretained(self).toOpaque()) as? T
+        }
+        set {
+            objc_setAssociatedObject(index, Unmanaged.passUnretained(self).toOpaque(), newValue, policy)
+        }
     }
 }

Tests/MainTests/MainTests.swift

@@ -1,6 +1,44 @@
 import XCTest
 @testable import B9AssociatedObject
 
-final class AssociatedObjectTests: XCTestCase {
+enum EnumType: Equatable, CustomDebugStringConvertible {
+    case one
+    case two
+
+    var debugDescription: String {
+        switch self {
+        case .one:
+            return "<EnumType.one>"
+        case .two:
+            return "<EnumType.two>"
+        }
+    }
+}
+
+let enumAssociation = AssociatedObject<EnumType>()
 
+class A {}
+
+extension A {
+    var enumValue: EnumType? {
+        get { enumAssociation[self] }
+        set { enumAssociation[self] = newValue }
+    }
+}
+
+final class AssociatedObjectTests: XCTestCase {
+    func testSwiftEnum() {
+        let a1 = A()
+        let a2 = A()
+        assert(a1.enumValue == nil)
+        a1.enumValue = .one
+        a2.enumValue = .two
+        debugPrint(a1.enumValue as Any, a2.enumValue as Any)
+        assert(a1.enumValue == .one)
+        assert(a2.enumValue == .two)
+        assert(a1.enumValue != a2.enumValue)
+        a1.enumValue = .two
+        assert(a1.enumValue == .two)
+        assert(a1.enumValue == a2.enumValue)
+    }
 }