From eec98b11b528ec4c9e7b1e78ec4ad04e9cd699e9 Mon Sep 17 00:00:00 2001
From: Yusuke Hosonuma <tobi462@gmail.com>
Date: Fri, 13 Mar 2020 11:44:25 +0900
Subject: [PATCH] fix: @available annotation

---
 Sources/DSL/BasicDSL.swift           | 31 ++++++++++++++++++++++++
 Sources/DSL/FunctionBuilderDSL.swift | 36 ----------------------------
 2 files changed, 31 insertions(+), 36 deletions(-)

diff --git a/Sources/DSL/BasicDSL.swift b/Sources/DSL/BasicDSL.swift
index 853e0b6..92fe6dc 100644
--- a/Sources/DSL/BasicDSL.swift
+++ b/Sources/DSL/BasicDSL.swift
@@ -5,6 +5,8 @@
 //  Created by Yusuke Hosonuma on 2020/03/12.
 //
 
+// MARK: - Basic API
+
 /// Assert to `target` function with `rows` parameters.
 /// - Parameters:
 ///   - target: test target function or operator
@@ -30,3 +32,32 @@ public func args<T, R>(_ args: T,
                        line: UInt = #line) -> Row<T, R> {
     Row(args: args, expect: expect, file: file, line: line)
 }
+
+// MARK: - Operator based API
+
+precedencegroup ExpectRowPrecedence {
+    lowerThan: FunctionArrowPrecedence
+}
+
+infix operator ==>: ExpectRowPrecedence
+
+/// Create expectation row.
+/// - Parameters:
+///   - lhs: parameters of test target function
+///   - rhs: expected value
+public func ==> <T, R>(lhs: T, rhs: R) -> ExpectRow<T, R> {
+    ExpectRow(args: lhs, expect: rhs)
+}
+
+public struct ExpectRow<T, R> {
+    var args: T
+    var expect: R
+}
+
+public func expect<T, R>(
+    _ row: ExpectRow<T, R>,
+    file: StaticString = #file,
+    line: UInt = #line
+) -> Row<T, R> {
+    Row(args: row.args, expect: row.expect, file: file, line: line)
+}
diff --git a/Sources/DSL/FunctionBuilderDSL.swift b/Sources/DSL/FunctionBuilderDSL.swift
index 2d065f8..3c74131 100644
--- a/Sources/DSL/FunctionBuilderDSL.swift
+++ b/Sources/DSL/FunctionBuilderDSL.swift
@@ -9,31 +9,6 @@
 // Note:
 // Function Builder is supported by Swift 5.1 or later.
 
-// MARK: - Operators
-
-precedencegroup ExpectRowPrecedence {
-    lowerThan: FunctionArrowPrecedence
-}
-
-infix operator ==>: ExpectRowPrecedence
-
-/// Create expectation row.
-/// - Parameters:
-///   - lhs: parameters of test target function
-///   - rhs: expected value
-@available(swift 5.1)
-public func ==> <T, R>(lhs: T, rhs: R) -> ExpectRow<T, R> {
-    ExpectRow(args: lhs, expect: rhs)
-}
-
-@available(swift 5.1)
-public struct ExpectRow<T, R> {
-    var args: T
-    var expect: R
-}
-
-// MARK: - Function Builders
-
 @available(swift 5.1)
 @_functionBuilder
 public struct ParameterBuilder<T, R> {
@@ -42,8 +17,6 @@ public struct ParameterBuilder<T, R> {
     }
 }
 
-// MARK: - DSL
-
 /// Create `Expect` for start parameterized-test.
 /// - Parameters:
 ///   - targetFunction: test target function
@@ -56,12 +29,3 @@ public func assert<T, R: Equatable>(
 ) {
     ParameterizedTest(target: targetFunction, customAssertion: customAssertion).execute(with: rows())
 }
-
-@available(swift 5.1)
-public func expect<T, R>(
-    _ row: ExpectRow<T, R>,
-    file: StaticString = #file,
-    line: UInt = #line
-) -> Row<T, R> {
-    Row(args: row.args, expect: row.expect, file: file, line: line)
-}