From 34dd1dbfd96bde4fcde7f51a97ccacc149142317 Mon Sep 17 00:00:00 2001
From: Benoit St-Pierre <bstpierre@google.com>
Date: Mon, 26 Feb 2018 13:25:26 -0500
Subject: Add tests for useragent

Tests a variety of simple use cases.
---
 Example/Core/Tests/FIRAppTest.m | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'Example')

diff --git a/Example/Core/Tests/FIRAppTest.m b/Example/Core/Tests/FIRAppTest.m
index 8466488..707c763 100644
--- a/Example/Core/Tests/FIRAppTest.m
+++ b/Example/Core/Tests/FIRAppTest.m
@@ -596,6 +596,28 @@ NSString *const kFIRTestAppName2 = @"test-app-name-2";
   XCTAssertFalse([FIRApp isDefaultAppConfigured]);
 }
 
+- (void)testIllegalLibraryName {
+  [FIRApp registerLibrary:@"Oops>" withVersion:@"1.0.0"];
+  XCTAssertTrue([[FIRApp firebaseUserAgent] isEqualToString:@""]);
+}
+
+- (void)testIllegalLibraryVersion {
+  [FIRApp registerLibrary:@"LegalName" withVersion:@"1.0.0+"];
+  XCTAssertTrue([[FIRApp firebaseUserAgent] isEqualToString:@""]);
+}
+
+- (void)testSingleLibrary {
+  [FIRApp registerLibrary:@"LegalName" withVersion:@"1.0.0"];
+  XCTAssertTrue([[FIRApp firebaseUserAgent] isEqualToString:@"LegalName/1.0.0"]);
+}
+
+- (void)testMultipleLibraries {
+  [FIRApp registerLibrary:@"LegalName" withVersion:@"1.0.0"];
+  [FIRApp registerLibrary:@"LegalName2" withVersion:@"2.0.0"];
+  XCTAssertTrue([[FIRApp firebaseUserAgent] containsString:@"LegalName/1.0.0"]);
+  XCTAssertTrue([[FIRApp firebaseUserAgent] containsString:@"LegalName2/2.0.0"]);
+}
+
 #pragma mark - private
 
 - (NSDictionary<NSString *, NSObject *> *)expectedUserInfoWithAppName:(NSString *)name
-- 
cgit v1.2.3