Add sources to topHeadlines

mtfum · Jul 31, 2021 · 1143721 · 1143721
1 parent 399c582
commit 1143721
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/Sources/NewsAPI/NewsAPI.swift b/Sources/NewsAPI/NewsAPI.swift
@@ -17,12 +17,12 @@ public actor NewsAPI {
   }
 
   public func getTopHeadlines(
-    sources: [NewsSource] = [],
+    sources: [String] = [],
     query: String? = nil,
     category: NewsSourceCategory? = nil,
     language: Language = Language.en
   ) async throws -> [NewsArticle] {
-    guard let url = Endpoint.topHeadlines(key: apiKey, query: query, category: category, language: language).url else {
+    guard let url = Endpoint.topHeadlines(key: apiKey, query: query, sources: sources, category: category, language: language).url else {
       throw ErrorResponse.invalidURL
     }
     do {

diff --git a/Tests/NewsAPITests/NewsAPITests.swift b/Tests/NewsAPITests/NewsAPITests.swift
@@ -23,6 +23,15 @@ final class NewsAPITests: XCTestCase {
     }
   }
 
+  func testGetTopHeadlinesWithSources() async throws {
+    do {
+      let articles = try await client.getTopHeadlines(sources: ["abc-news", "bbc-news"])
+      XCTAssertTrue(!articles.isEmpty)
+    } catch {
+      XCTFail(String(describing: error))
+    }
+  }
+
   func testGetSources() async throws {
     do {
       let sources = try await client.getSources()