feat(api): (Integ) SocialNote from codegen (#469)

aws-amplify · May 20, 2020 · 79c6482 · 79c6482
1 parent 4ab2740
commit 79c6482
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 35 deletions.
diff --git a/...UserPoolIntegrationTests/AuthDirective/GraphQLAuthDirectiveIntegrationTests+Support.swift b/...UserPoolIntegrationTests/AuthDirective/GraphQLAuthDirectiveIntegrationTests+Support.swift
@@ -157,17 +157,9 @@ extension GraphQLAuthDirectiveIntegrationTests {
             XCTFail("underlying error should be AuthError, but instead was \(underlyingError ?? "nil")")
             return
         }
-        guard case let .service(_, _, underlyingAuthError) = authError else {
+        guard case .signedOut = authError else {
             XCTFail("Error should be AuthError.service")
             return
         }
-        guard let awsCognitoAuthError = underlyingAuthError as? AWSCognitoAuthError else {
-            XCTFail("Error should be AWSCognitoAuthError")
-            return
-        }
-        guard case .signedOut = awsCognitoAuthError else {
-            XCTFail("Error should be .signedOut")
-            return
-        }
     }
 }
diff --git a/...tegrationTests/GraphQL/GraphQLWithUserPoolIntegrationTests/AuthDirective/SocialNote.swift b/...tegrationTests/GraphQL/GraphQLWithUserPoolIntegrationTests/AuthDirective/SocialNote.swift
@@ -8,31 +8,43 @@
 import Amplify
 
 public struct SocialNote: Model {
-    public let id: String
-    public var content: String
-    public var owner: String?
-    public init(id: String = UUID().uuidString,
-                content: String,
-                owner: String?) {
-        self.id = id
-        self.content = content
-        self.owner = owner
-    }
-    public enum CodingKeys: String, ModelKey {
-        case id
-        case content
-        case owner
-    }
-    public static let keys = CodingKeys.self
-
-    public static let schema = defineSchema { model in
-        let socialNote = SocialNote.keys
-        model.authRules = [
-            rule(allow: .owner, operations: [.create, .update, .delete])
-        ]
-        model.fields(
-            .id(),
-            .field(socialNote.content, is: .required, ofType: .string),
-            .field(socialNote.owner, is: .optional, ofType: .string))
+  public let id: String
+  public var content: String
+  public var owner: String?
+
+  public init(id: String = UUID().uuidString,
+      content: String,
+      owner: String? = nil) {
+      self.id = id
+      self.content = content
+      self.owner = owner
+  }
+}
+
+extension SocialNote {
+  // MARK: - CodingKeys
+   public enum CodingKeys: String, ModelKey {
+    case id
+    case content
+    case owner
+  }
+
+  public static let keys = CodingKeys.self
+  //  MARK: - ModelSchema
+
+  public static let schema = defineSchema { model in
+    let socialNote = SocialNote.keys
+
+    model.authRules = [
+      rule(allow: .owner, ownerField: "owner", identityClaim: "cognito:username", operations: [.create, .update, .delete])
+    ]
+
+    model.pluralName = "SocialNotes"
+
+    model.fields(
+      .id(),
+      .field(socialNote.content, is: .required, ofType: .string),
+      .field(socialNote.owner, is: .optional, ofType: .string)
+    )
     }
 }