safe-ios/MultisigIntegrationTests/CoreData/SafeTests.swift

//
//  SafeTests.swift
//  MultisigTests
//
//  Created by Andrey Scherbovich on 21.04.20.
//  Copyright © 2020 Gnosis Ltd. All rights reserved.
//

import XCTest
@testable import Multisig

class SafeTests: CoreDataTestCase {
    func test_removeSafe() throws {
        let network1 = try makeChain(id: "1")
        let network2 = try makeChain(id: "2")

        Safe.create(address: "0x0000000000000000000000000000000000000000", version: "1.2.0", name: "0", chain: network1, selected: false)
        Safe.create(address: "0x0000000000000000000000000000000000000001", version: "1.2.0", name: "1", chain: network1, selected: true)
        Safe.create(address: "0x0000000000000000000000000000000000000002", version: "1.2.0", name: "2", chain: network2, selected: false)

        var safesResult = try context.fetch(Safe.fetchRequest().all())
        var networksResult = try context.fetch(Chain.fetchRequest().all())
        XCTAssertEqual(safesResult.count, 3)
        XCTAssertEqual(networksResult.count, 2)

        var safe = safesResult.first!
        Safe.remove(safe: safe)
        safesResult = try context.fetch(Safe.fetchRequest().all())
        networksResult = try context.fetch(Chain.fetchRequest().all())
        XCTAssertEqual(safesResult.count, 2)
        XCTAssertEqual(networksResult.count, 2)

        safe = safesResult.first!
        XCTAssertTrue(safe.isSelected)
        Safe.remove(safe: safe)
        safesResult = try context.fetch(Safe.fetchRequest().all())
        networksResult = try context.fetch(Chain.fetchRequest().all())
        XCTAssertEqual(safesResult.count, 1)
        XCTAssertEqual(networksResult.count, 2)

        safe = safesResult.first!
        XCTAssertNotNil(safe.selection)
        Safe.remove(safe: safe)
        safesResult = try context.fetch(Safe.fetchRequest().all())
        networksResult = try context.fetch(Chain.fetchRequest().all())
        XCTAssertEqual(safesResult.count, 0)
        XCTAssertEqual(networksResult.count, 2)
    }

    func test_allSafes() throws {
        let safe0 = createSafe(name: "1", address: "0x1")
        let safe1 = createSafe(name: "0", address: "0x0")
        let allSafes = try Safe.getAll()
        XCTAssertEqual(allSafes.count, 2)
        // should be sorted by creation date
        XCTAssertEqual(allSafes[0], safe0)
        XCTAssertEqual(allSafes[1], safe1)
    }

    func test_safeBy() throws {
        let safe = createSafe(name: "0", address: "0x0")
        createSafe(name: "1", address: "0x1")
        let result = Safe.by(address: "0x0", chainId: Chain.ChainID.ethereumMainnet)
        XCTAssertNotNil(result)
        XCTAssertEqual(result, safe)
    }

    func test_update() {
        let safe = createSafe(name: "0", address: Address.zero.checksummed)
        safe.update(name: "1")
        let result = Safe.by(address: Address.zero.checksummed, chainId: Chain.ChainID.ethereumMainnet)
        XCTAssertEqual(result!.name, "1")
    }

    func test_select() throws {
        let testNetwork1 = try makeChain(id: "1")
        let testNetwork2 = try makeChain(id: "2")
        let testNetwork3 = try makeChain(id: "3")

        Safe.create(address: "0x0000000000000000000000000000000000000001", version: "1.2.0", name: "1", chain: testNetwork1, selected: false)
        Safe.create(address: "0x0000000000000000000000000000000000000001", version: "1.2.0", name: "2", chain: testNetwork2, selected: false)
        Safe.create(address: "0x0000000000000000000000000000000000000001", version: "1.2.0", name: "3", chain: testNetwork3, selected: true)

        Safe.select(address: "0x0000000000000000000000000000000000000001", chainId: "2")

        let result = try Safe.getSelected()!
        XCTAssertEqual(result.name, "2")
    }
}