Update sidebar tests

Uncommon · Jun 27, 2018 · 2bb9e39 · 2bb9e39
1 parent f07d899
commit 2bb9e39
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/XitTests/XTSideBarDataSourceTests.swift b/XitTests/XTSideBarDataSourceTests.swift
@@ -266,16 +266,21 @@ class MockSidebarOutline: NSOutlineView
                                                 width: 16, height: 16))
     let statusButton = NSButton(frame: NSRect(x: 171, y: 2,
                                               width: 16, height: 16))
+    let buttonContainer = NSView(frame: NSRect(x: 0, y: 0, width: 20, height: 16))
+    let missingImage = NSImageView(frame: NSRect(x: 0, y: 0, width: 20, height: 16))
     let statusText = NSButton(title: "10", target: nil, action: nil)
 
-    for view in [textField, imageView, statusImage, statusButton, statusText] {
+    for view in [textField, imageView, statusImage, statusButton, statusText,
+                 buttonContainer, missingImage] {
       result.addSubview(view)
     }
     result.textField = textField
     result.imageView = imageView
     result.statusImage = statusImage
     result.statusButton = statusButton
     result.statusText = statusText
+    result.buttonContainer = buttonContainer
+    result.missingImage = missingImage
 
     return result
   }