SwiftLint/Tests/SwiftLintFrameworkTests/ObjectLiteralRuleTests.swift

//
//  ObjectLiteralRuleTests.swift
//  SwiftLint
//
//  Created by Cihat Gündüz on 06/13/17.
//  Copyright © 2017 Realm. All rights reserved.
//

import SwiftLintFramework
import XCTest

class ObjectLiteralRuleTests: XCTestCase {
    // MARK: - Instance Properties
    private let imageLiteralTriggeringExamples = ["", ".init"].flatMap { (method: String) -> [String] in
        ["UI", "NS"].flatMap { (prefix: String) -> [String] in
            [
                "let image = ↓\(prefix)Image\(method)(named: \"foo\")"
            ]
        }
    }

    private let colorLiteralTriggeringExamples = ["", ".init"].flatMap { (method: String) -> [String] in
        ["UI", "NS"].flatMap { (prefix: String) -> [String] in
            [
                "let color = ↓\(prefix)Color\(method)(red: 0.3, green: 0.3, blue: 0.3, alpha: 1)",
                "let color = ↓\(prefix)Color\(method)(red: 100 / 255.0, green: 50 / 255.0, blue: 0, alpha: 1)",
                "let color = ↓\(prefix)Color\(method)(white: 0.5, alpha: 1)"
            ]
        }
    }

    private var allTriggeringExamples: [String] {
        return imageLiteralTriggeringExamples + colorLiteralTriggeringExamples
    }

    // MARK: - Test Methods
    func testObjectLiteralWithDefaultConfiguration() {
        verifyRule(ObjectLiteralRule.description)
    }

    func testObjectLiteralWithImageLiteral() {
        // Verify ObjectLiteral rule for when image_literal is true.
        let baseDescription = ObjectLiteralRule.description
        let nonTriggeringColorLiteralExamples = colorLiteralTriggeringExamples.map { example in
            example.replacingOccurrences(of: "↓", with: "")
        }

        let description = RuleDescription(
            identifier: baseDescription.identifier,
            name: baseDescription.name,
            description: baseDescription.description,
            nonTriggeringExamples: baseDescription.nonTriggeringExamples + nonTriggeringColorLiteralExamples,
            triggeringExamples: imageLiteralTriggeringExamples
        )

        verifyRule(description, ruleConfiguration: ["image_literal": true, "color_literal": false])
    }

    func testObjectLiteralWithColorLiteral() {
        // Verify ObjectLiteral rule for when color_literal is true.
        let baseDescription = ObjectLiteralRule.description
        let nonTriggeringImageLiteralExamples = imageLiteralTriggeringExamples.map { example in
            example.replacingOccurrences(of: "↓", with: "")
        }

        let description = RuleDescription(
            identifier: baseDescription.identifier,
            name: baseDescription.name,
            description: baseDescription.description,
            nonTriggeringExamples: baseDescription.nonTriggeringExamples + nonTriggeringImageLiteralExamples,
            triggeringExamples: colorLiteralTriggeringExamples
        )

        verifyRule(description, ruleConfiguration: ["image_literal": false, "color_literal": true])
    }

    func testObjectLiteralWithImageAndColorLiteral() {
        // Verify ObjectLiteral rule for when image_literal & color_literal are true.
        let baseDescription = ObjectLiteralRule.description

        let description = RuleDescription(
            identifier: baseDescription.identifier,
            name: baseDescription.name,
            description: baseDescription.description,
            nonTriggeringExamples: baseDescription.nonTriggeringExamples,
            triggeringExamples: allTriggeringExamples
        )

        verifyRule(description, ruleConfiguration: ["image_literal": true, "color_literal": true])
    }
}