Added unit test for sizeThatFits when using a custom truncation token…

… with a dynamic height label.
Alterplay · Oct 26, 2016 · 0763af6 · 0763af6
1 parent 9a8cf93
commit 0763af6
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/Example/TTTAttributedLabelTests/TTTAttributedLabelTests.m b/Example/TTTAttributedLabelTests/TTTAttributedLabelTests.m
@@ -26,6 +26,11 @@
                                            attributes:TTTAttributedTestAttributesDictionary()];
 }
 
+static inline NSAttributedString * TTTAttributedTruncationTokenString() {
+    return [[NSAttributedString alloc] initWithString:@"+++"
+                                           attributes:TTTAttributedTestAttributesDictionary()];
+}
+
 static inline void TTTSizeAttributedLabel(TTTAttributedLabel *label) {
     CGSize size = [TTTAttributedLabel sizeThatFitsAttributedString:label.attributedText
                                                    withConstraints:kTestLabelSize
@@ -197,6 +202,19 @@ - (void)testMultilineLabelSizing {
     XCTAssertGreaterThan(size.height, font.pointSize, @"Text should size to more than one line");
 }
 
+- (void)testMultilineLabelSizeThatFitsWithTruncationToken {
+    NSAttributedString *testString = TTTAttributedTestString();
+    label.text = testString;
+
+    NSAttributedString *tokenString = TTTAttributedTruncationTokenString();
+    label.attributedTruncationToken = tokenString;
+
+    CGSize size = [label sizeThatFits:kTestLabelSize];
+
+    UIFont *font = [testString attribute:NSFontAttributeName atIndex:0 effectiveRange:NULL];
+    XCTAssertGreaterThan(size.height, font.pointSize, @"Text should size to more than one line");
+}
+
 - (void)testContainsLinkAtPoint {
     label.text = TTTAttributedTestString();
     [label addLinkToURL:testURL withRange:NSMakeRange(0, 4)];