components/Cards/tests/unit/MDCCardShapeThemerTests.swift

// Copyright 2018-present the Material Components for iOS authors. All Rights Reserved.
//
// 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 MaterialComponents.MaterialCards
import MaterialComponents.MaterialCards_ShapeThemer
import MaterialComponents.MaterialShapeLibrary

class CardShapeThemerTests: XCTestCase {

  func testCardShapeThemer() {
    // Given
    let shapeScheme = MDCShapeScheme()
    let card = MDCCard()
    shapeScheme.mediumComponentShape = MDCShapeCategory(cornersWith: .cut, andSize: 10)
    shapeScheme.mediumComponentShape.topRightCorner = MDCCornerTreatment.corner(withRadius: 3)
    card.shapeGenerator = MDCRectangleShapeGenerator()

    // When
    MDCCardsShapeThemer.applyShapeScheme(shapeScheme, to: card)

    // Then
    XCTAssert(card.shapeGenerator is MDCRectangleShapeGenerator);
    XCTAssertEqual((card.shapeGenerator as! MDCRectangleShapeGenerator).topLeftCorner,
                   shapeScheme.mediumComponentShape.topLeftCorner)
    XCTAssertEqual((card.shapeGenerator as! MDCRectangleShapeGenerator).topRightCorner,
                   shapeScheme.mediumComponentShape.topRightCorner)
    XCTAssertEqual((card.shapeGenerator as! MDCRectangleShapeGenerator).bottomLeftCorner,
                   shapeScheme.mediumComponentShape.bottomLeftCorner)
    XCTAssertEqual((card.shapeGenerator as! MDCRectangleShapeGenerator).bottomRightCorner,
                   shapeScheme.mediumComponentShape.bottomRightCorner)
  }

  func testCardCollectionCellShapeThemer() {
    // Given
    let shapeScheme = MDCShapeScheme()
    let cardCell = MDCCardCollectionCell()
    shapeScheme.mediumComponentShape = MDCShapeCategory(cornersWith: .cut, andSize: 10)
    cardCell.shapeGenerator = MDCRectangleShapeGenerator()

    // When
    MDCCardsShapeThemer.applyShapeScheme(shapeScheme, toCardCell: cardCell)

    // Then
    XCTAssert(cardCell.shapeGenerator is MDCRectangleShapeGenerator);
    XCTAssertEqual((cardCell.shapeGenerator as! MDCRectangleShapeGenerator).topLeftCorner,
                   shapeScheme.mediumComponentShape.topLeftCorner)
    XCTAssertEqual((cardCell.shapeGenerator as! MDCRectangleShapeGenerator).topRightCorner,
                   shapeScheme.mediumComponentShape.topRightCorner)
    XCTAssertEqual((cardCell.shapeGenerator as! MDCRectangleShapeGenerator).bottomLeftCorner,
                   shapeScheme.mediumComponentShape.bottomLeftCorner)
    XCTAssertEqual((cardCell.shapeGenerator as! MDCRectangleShapeGenerator).bottomRightCorner,
                   shapeScheme.mediumComponentShape.bottomRightCorner)
  }
}