Update testing variables to reflect switch to delegate

facebookarchive · Jun 23, 2016 · 587ee7e · 587ee7e
1 parent 14fdafd
commit 587ee7e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/AsyncDisplayKitTests/ASTableViewTests.m b/AsyncDisplayKitTests/ASTableViewTests.m
@@ -139,10 +139,10 @@ - (ASCellNodeBlock)tableView:(ASTableView *)tableView nodeBlockForRowAtIndexPath
 
 @end
 
-@interface ASTableViewFilledDataSourceWithRowSize : ASTableViewFilledDataSource
+@interface ASTableViewFilledDelegate : NSObject <ASTableViewDelegate>
 @end
 
-@implementation ASTableViewFilledDataSourceWithRowSize
+@implementation ASTableViewFilledDelegate
 
 - (ASSizeRange)tableView:(ASTableView *)tableView constrainedSizeForRowAtIndexPath:(NSIndexPath *)indexPath
 {
@@ -165,9 +165,10 @@ - (void)testConstrainedSizeForRowAtIndexPath
   ASTestTableView *tableView = [[ASTestTableView alloc] __initWithFrame:CGRectMake(0, 0, 100, 400)
                                                                   style:UITableViewStylePlain];
 
-  ASTableViewFilledDataSourceWithRowSize *dataSource = [ASTableViewFilledDataSourceWithRowSize new];
+  ASTableViewFilledDelegate *delegate = [ASTableViewFilledDelegate new];
+  ASTableViewFilledDataSource *dataSource = [ASTableViewFilledDataSource new];
 
-  tableView.asyncDelegate = dataSource;
+  tableView.asyncDelegate = delegate;
   tableView.asyncDataSource = dataSource;
 
   [tableView reloadDataImmediately];