Merge pull request #91 from matej/matej/test-shape

Add HUD shape test
jdg · Nov 4, 2018 · 202d889 · 202d889
2 parents 2347f40 + b74d8e8
commit 202d889
Showing 1 changed file with 25 additions and 2 deletions.
diff --git a/Demo/HudTests/HudTests.m b/Demo/HudTests/HudTests.m
@@ -136,7 +136,6 @@ - (void)testCompletionBlock {
     [self waitForExpectationsWithTimeout:5. handler:nil];
 }
 
-///////////////////////////////////////////////////////////////////////////////////////////
 #pragma mark - Modes
 
 - (void)testRoundDeterminate {
@@ -463,6 +462,31 @@ - (void)testHideBeforeGraceTimeElapsed {
     MBTestHUDIsHidenAndRemoved(hud, rootView);
 }
 
+#pragma mark - Customization
+
+- (void)testShape {
+    UIViewController *rootViewController = UIApplication.sharedApplication.keyWindow.rootViewController;
+    UIView *rootView = rootViewController.view;
+
+    MBProgressHUD *hud = [[MBProgressHUD alloc] initWithView:rootView];
+    hud.removeFromSuperViewOnHide = YES;
+    hud.offset = CGPointMake(50, 50);
+    hud.square = YES;
+    hud.label.text = @"Some long text...";
+    [rootView addSubview:hud];
+    [hud showAnimated:NO];
+
+    [hud setNeedsLayout];
+    [hud layoutIfNeeded];
+
+    CGRect frame = hud.bezelView.frame;
+    XCTAssertEqual(frame.size.width, frame.size.height);
+
+    [hud hideAnimated:NO];
+
+    MBTestHUDIsHidenAndRemoved(hud, rootView);
+}
+
 #pragma mark - MBProgressHUDDelegate
 
 - (void)hudWasHidden:(MBProgressHUD *)hud {
@@ -473,7 +497,6 @@ - (void)hudWasHidden:(MBProgressHUD *)hud {
     self.hideExpectation = nil;
 }
 
-///////////////////////////////////////////////////////////////////////////////////////////
 #pragma mark - Helpers
 
 - (nullable UIView *)view:(UIView *)view firstSubviewOfClass:(Class)clazz {