space-ios/Carthage/Checkouts/realm-cocoa/RealmSwift/Tests/SwiftLinkTests.swift

////////////////////////////////////////////////////////////////////////////
//
// Copyright 2014 Realm 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.
//
////////////////////////////////////////////////////////////////////////////

import XCTest
import RealmSwift

class SwiftLinkTests: TestCase {

    func testBasicLink() {
        let realm = realmWithTestPath()

        let owner = SwiftOwnerObject()
        owner.name = "Tim"
        owner.dog = SwiftDogObject()
        owner.dog!.dogName = "Harvie"

        try! realm.write { realm.add(owner) }

        let owners = realm.objects(SwiftOwnerObject.self)
        let dogs = realm.objects(SwiftDogObject.self)
        XCTAssertEqual(owners.count, Int(1), "Expecting 1 owner")
        XCTAssertEqual(dogs.count, Int(1), "Expecting 1 dog")
        XCTAssertEqual(owners[0].name, "Tim", "Tim is named Tim")
        XCTAssertEqual(dogs[0].dogName, "Harvie", "Harvie is named Harvie")

        XCTAssertEqual(owners[0].dog!.dogName, "Harvie", "Tim's dog should be Harvie")
    }

    func testMultipleOwnerLink() {
        let realm = realmWithTestPath()

        let owner = SwiftOwnerObject()
        owner.name = "Tim"
        owner.dog = SwiftDogObject()
        owner.dog!.dogName = "Harvie"

        try! realm.write { realm.add(owner) }

        XCTAssertEqual(realm.objects(SwiftOwnerObject.self).count, Int(1), "Expecting 1 owner")
        XCTAssertEqual(realm.objects(SwiftDogObject.self).count, Int(1), "Expecting 1 dog")

        realm.beginWrite()
        let fiel = realm.create(SwiftOwnerObject.self, value: ["Fiel", NSNull()])
        fiel.dog = owner.dog
        try! realm.commitWrite()

        XCTAssertEqual(realm.objects(SwiftOwnerObject.self).count, Int(2), "Expecting 2 owners")
        XCTAssertEqual(realm.objects(SwiftDogObject.self).count, Int(1), "Expecting 1 dog")
    }

    func testLinkRemoval() {
        let realm = realmWithTestPath()

        let owner = SwiftOwnerObject()
        owner.name = "Tim"
        owner.dog = SwiftDogObject()
        owner.dog!.dogName = "Harvie"

        try! realm.write { realm.add(owner) }

        XCTAssertEqual(realm.objects(SwiftOwnerObject.self).count, Int(1), "Expecting 1 owner")
        XCTAssertEqual(realm.objects(SwiftDogObject.self).count, Int(1), "Expecting 1 dog")

        try! realm.write { realm.delete(owner.dog!) }

        XCTAssertNil(owner.dog, "Dog should be nullified when deleted")

        // refresh owner and check
        let owner2 = realm.objects(SwiftOwnerObject.self).first!
        XCTAssertNotNil(owner2, "Should have 1 owner")
        XCTAssertNil(owner2.dog, "Dog should be nullified when deleted")
        XCTAssertEqual(realm.objects(SwiftDogObject.self).count, Int(0), "Expecting 0 dogs")
    }

    func testLinkingObjects() {
        let realm = realmWithTestPath()

        let owner = SwiftOwnerObject()
        owner.name = "Tim"
        owner.dog = SwiftDogObject()
        owner.dog!.dogName = "Harvie"

        XCTAssertEqual(0, owner.dog!.owners.count, "Linking objects are not available until the object is persisted")

        try! realm.write {
            realm.add(owner)
        }

        let owners = owner.dog!.owners
        XCTAssertEqual(1, owners.count)
        XCTAssertEqual(owner.name, owners.first!.name)

        try! realm.write {
            owner.dog = nil
        }

        XCTAssertEqual(0, owners.count)
    }
}