ADD: 添加测试代码

Author: sgxiang

YTKKeyValueStore_Swift.xcodeproj/project.xcworkspace/xcuserdata/YSQ.xcuserdatad/UserInterfaceState.xcuserstate

@@ -55,12 +55,12 @@ public class YTKKeyValueStore_Swift: NSObject {
 
     //MARK: - 初始化
 
-    override init(){
+    public override init(){
         super.init()
         self.setupDB(DEFAULT_DB_NAME)
     }
 
-    init(dbName : String!){
+    public init(dbName : String!){
         super.init()
         self.setupDB(dbName)
     }

YTKKeyValueStore_Swift/YTKKeyValueStore_Swift.swift

@@ -8,29 +8,76 @@
 
 import UIKit
 import XCTest
+import YTKKeyValueStore_Swift
 
 class YTKKeyValueStore_SwiftTests: XCTestCase {
 
+    private var _store : YTKKeyValueStore_Swift?
+    private var _tableName : String?
+    
     override func setUp() {
         super.setUp()
-        // Put setup code here. This method is called before the invocation of each test method in the class.
+        _tableName = "test_table"
+        _store = YTKKeyValueStore_Swift()
+        _store?.createTable(tableName: _tableName!)
+        _store?.clearTable(tableName: _tableName!)
     }
 
     override func tearDown() {
-        // Put teardown code here. This method is called after the invocation of each test method in the class.
+        _store?.clearTable(tableName: _tableName)
+        _store?.close()
+        _store = nil
         super.tearDown()
     }
 
-    func testExample() {
-        // This is an example of a functional test case.
-        XCTAssert(true, "Pass")
+    func testSaveString(){
+       
+        let str1 = "abc"
+        let key1 = "key1"
+        let str2 = "abc2"
+        let key2 = "key2"
+        
+        _store?.putString(str1, withId: key1, intoTable: _tableName)
+        _store?.putString(str2, withId: key2, intoTable: _tableName)
+        
+        var result : String?
+        
+        result = _store?.getStringById(key1, fromTable: _tableName)
+        XCTAssertNotNil(result)
+        XCTAssertEqual(str1, result!)
+        
+        result = _store?.getStringById(key2, fromTable: _tableName)
+        XCTAssertNotNil(result)
+        XCTAssertEqual(str2, result!)
+        
+        result = _store?.getStringById("key3", fromTable: _tableName)
+        XCTAssertNil(result)
+        
     }
 
-    func testPerformanceExample() {
-        // This is an example of a performance test case.
-        self.measureBlock() {
-            // Put the code you want to measure the time of here.
-        }
+    func testSaveCGFloat(){
+        
+        let num1 : CGFloat = 1
+        let key1 = "key1"
+        let num2 : CGFloat = 2
+        let key2 = "key2"
+        
+        _store?.putNumber(num1, withId: key1, intoTable: _tableName)
+        _store?.putNumber(num2, withId: key2, intoTable: _tableName)
+        
+        var result : CGFloat?
+        
+        result = _store?.getNumberById(key1, fromTable: _tableName)
+        XCTAssertNotNil(result)
+        XCTAssertEqual(num1, result!)
+        
+        result = _store?.getNumberById(key2, fromTable: _tableName)
+        XCTAssertNotNil(result)
+        XCTAssertEqual(num2, result!)
+        
+        result = _store?.getNumberById("key3", fromTable: _tableName)
+        XCTAssertNil(result)
+    
     }
 
 }