refactored_test.swift

class RefactoredCatalogExportingTests: XCTestCase {
  func test_Categories_Are_Exported_By_DisplayOrder() {
    let fooCategory = CatalogCategory2(displayOrder: 1, name: "Foo")
    let barCategory = CatalogCategory2(displayOrder: 2, name: "Bar")
    let stubCatalog = StubCatalog(stubs: [fooCategory, barCategory])

    let catalogExporter = CatalogExporter()
    let exportedCatalog = catalogExporter.export(stubCatalog)

    XCTAssertEqual(exportedCatalog.count, 2)
    XCTAssertEqual(exportedCatalog[0].name, "Foo")
    XCTAssertEqual(exportedCatalog[1].name, "Bar")
  }
}