test_protocol_reorder_requirements.swift

// RUN: %target-resilience-test
// REQUIRES: executable_test

import StdlibUnittest
import protocol_reorder_requirements


var ProtocolReorderRequirementsTest = TestSuite("ProtocolReorderRequirements")

var log = [String]()

struct MyBassinet : Bed {
  func squiggle() {
    log.append("nap time")
  }
}

struct MyOnesie : Outfit {
  let size = 3
}

struct SillyBaby : Baby {
  func eat() {
    log.append("hangry!")
  }

  func sleep(in bassinet: MyBassinet) {
    bassinet.squiggle()
  }

  func wear(outfit: MyOnesie) {
    log.append("wearing outfit size \(outfit.size)")
  }

  func poop() {
    log.append("change the diaper")
  }

  func cry() {
    log.append("waaaaah!")
  }

  func wiggle() {
    log.append("time to wiggle!")
  }

  let outfitSize = 3
}

func typicalDay<B : Baby>(for baby: B,
                          sleepingIn bed: B.Bassinet,
                          wearing outfit: B.Onesie) {
  baby.wear(outfit: outfit)
  baby.sleep(in: bed)
  baby.cry()
  baby.poop()
  baby.cry()
  baby.sleep(in: bed)
  baby.eat()
  baby.cry()
}

ProtocolReorderRequirementsTest.test("ReorderProtocolRequirements") {
  let baby = SillyBaby()
  let bed = MyBassinet()
  let outfit = MyOnesie()

  typicalDay(for: baby, sleepingIn: bed, wearing: outfit)
  expectEqual(log, [
      "wearing outfit size 3",
      "nap time",
      "waaaaah!",
      "change the diaper",
      "waaaaah!",
      "nap time",
      "hangry!",
      "waaaaah!"
    ])
  log = []

  goodDay(for: baby, sleepingIn: bed, wearing: outfit)
  expectEqual(log, [
      "wearing outfit size 3",
      "nap time",
      "change the diaper",
      "nap time",
      "hangry!",
      "nap time",
      "time to wiggle!"
    ])
}

struct Adult<Child: Baby> { }

extension Adult: Adoring
where Child.Onesie == MyOnesie, Child.Bassinet == MyBassinet {
  func adore() -> String { return "awwwwwww" }
}

struct MyDiaper : Outfit {
  let size = 2
}

struct GrumpyBaby : Baby {
  func eat() {
    log.append("waaaaaa!")
  }

  func sleep(in bassinet: MyBassinet) {
    bassinet.squiggle()
  }

  func wear(outfit: MyDiaper) {
    log.append("waaaaaa!")
  }

  func poop() {
    log.append("waaaaah!")
  }

  func cry() {
    log.append("waaaaah!")
  }

  func wiggle() {
    log.append("waaaaah!")
  }

  let outfitSize = 2
}

func adoreIfYouCan(_ value: Any) -> String {
  if let adoring = value as? Adoring {
    return adoring.adore()
  }

  return "bah humbug"
}

ProtocolReorderRequirementsTest.test("ReorderProtocolRequirements") {
  let adult1 = Adult<SillyBaby>()
  let exclamation1 = adoreIfYouCan(adult1)
  expectEqual(exclamation1, "awwwwwww")
  let adult2 = Adult<GrumpyBaby>()
  let exclamation2 = adoreIfYouCan(adult2)
  expectEqual(exclamation2, "bah humbug")
  
}

runAllTests()