BookmarksPanelTests.swift

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

@testable import Client
@testable import Storage
import Shared
import Foundation
import WebKit
import EarlGrey

class BookmarksPanelTests: KIFTestCase {
    
    override func setUp() {
        super.setUp()
        BrowserUtils.configEarlGrey()
        BrowserUtils.dismissFirstRunUI()
	}
	
    override func tearDown() {
        super.tearDown()
		BrowserUtils.resetToAboutHome() 
    }

    private func getAppBookmarkStorage() -> BookmarkBufferStorage? {
        let application = UIApplication.shared

        guard let delegate = application.delegate as? TestAppDelegate else {
            XCTFail("Couldn't get app delegate.")
            return nil
        }
        
        let profile = delegate.getProfile(application)
        
        guard let bookmarks = profile.bookmarks as? BookmarkBufferStorage else {
            XCTFail("Couldn't get buffer storage.")
            return nil
        }
        
        return bookmarks
    }
    
    private func createSomeBufferBookmarks() {
        // Set up the buffer.
        let bufferDate = Date.now()
        let changedBufferRecords = [
            BookmarkMirrorItem.folder(BookmarkRoots.ToolbarFolderGUID, dateAdded: Date.now(), modified: bufferDate, hasDupe: false, parentID: BookmarkRoots.RootGUID, parentName: nil, title: "Bookmarks Toolbar", description: nil, children: ["aaa", "bbb"]),
            BookmarkMirrorItem.bookmark("aaa", dateAdded: Date.now(), modified: Date.now(), hasDupe: false, parentID: BookmarkRoots.ToolbarFolderGUID, parentName: nil, title: "AAA", description: nil, URI: "http://getfirefox.com", tags: "[]", keyword: nil),
            BookmarkMirrorItem.livemark("bbb", dateAdded: Date.now(), modified: Date.now(), hasDupe: false, parentID: BookmarkRoots.ToolbarFolderGUID, parentName: nil, title: "Some Livemark", description: nil, feedURI: "https://www.google.ca", siteURI: "https://www.google.ca") ]
        
        if let bookmarks = getAppBookmarkStorage() {
            XCTAssert(bookmarks.applyRecords(changedBufferRecords).value.isSuccess)
        }
    }

    func testBookmarkPanelBufferOnly() {
        // Insert some data into the buffer. There will be nothing in the mirror, but we can still
        // show Desktop Bookmarks.

        createSomeBufferBookmarks()

        EarlGrey.selectElement(with: grey_accessibilityID("url")).perform(grey_tap())
        EarlGrey.selectElement(with: grey_accessibilityLabel("Bookmarks")).perform(grey_tap())
        EarlGrey.selectElement(with: grey_accessibilityLabel("Desktop Bookmarks")).perform(grey_tap())
        EarlGrey.selectElement(with: grey_accessibilityLabel("Bookmarks Toolbar")).perform(grey_tap())
        EarlGrey.selectElement(with: grey_accessibilityLabel("AAA"))
            .assert(grey_sufficientlyVisible())
        EarlGrey.selectElement(with: grey_accessibilityLabel("Some Livemark"))
            .assert(grey_sufficientlyVisible())
        
        // When we tap the livemark, we load the siteURI.
        EarlGrey.selectElement(with: grey_accessibilityLabel("Some Livemark")).perform(grey_tap())
        
        // … so we show the truncated URL.
        // Strangely, earlgrey cannot find the label in buddybuild, but passes locally.
        // Using KIF for this check for now.
        EarlGrey.selectElement(with: grey_accessibilityValue("www.google.ca/")).assert(grey_sufficientlyVisible())
    }

    private func navigateBackInTableView() {
        EarlGrey.selectElement(with: grey_accessibilityLabel("Bookmarks")).inRoot(grey_kindOfClass(UITableView.self)).perform(grey_tap())
    }
    
    private func navigateFolder(withTitle title: String) {
        EarlGrey.selectElement(with: grey_accessibilityLabel(title)).perform(grey_tap())
    }

    private func makeBookmark(guid: GUID, parentID: GUID, title: String) -> BookmarkMirrorItem {
        return BookmarkMirrorItem.bookmark(guid, dateAdded: Date.now(), modified: Date.now(), hasDupe: false, parentID: parentID, parentName: nil, title: title, description: nil, URI: "http://unused.com", tags: "[]", keyword: nil)
    }
    
    private func makeFolder(guid: GUID, parentID: GUID, title: String, childrenGuids: [GUID]) -> BookmarkMirrorItem {
        return BookmarkMirrorItem.folder(guid, dateAdded: Date.now(), modified: Date.now(), hasDupe: false, parentID: parentID, parentName: nil, title: title, description: nil, children: childrenGuids)
    }
    
    private func assertRowExists(withTitle title: String) {
        EarlGrey.selectElement(with: grey_accessibilityLabel(title)).inRoot(grey_kindOfClass(UITableView.self)).assert(grey_notNil())
    }

    func testMobileBookmarks() {
        verifyRootHasLocalAndBufferBookmarks()

        // Rather than an elaborate setup() step, just continue from the previous test
        // as the view state is now setup for testing deletion.
        verifyMobileBookmarkDelete()
    }

    func verifyRootHasLocalAndBufferBookmarks() {
        // Add buffer data, then later in the test verify that the buffer mobile folder is not shown in there anymore.
        createSomeBufferBookmarks()
        
        guard let bookmarks = getAppBookmarkStorage() else {
            return
        }
        
        // TEST: Create remote mobile bookmark, and verify the bookmark appears in the root and the remote mobile folder is not shown
        var applyResult = bookmarks.applyRecords([
            makeBookmark(guid: "bm-guid0", parentID: BookmarkRoots.MobileFolderGUID, title: "xyz"),
            makeFolder(guid: BookmarkRoots.MobileFolderGUID, parentID: BookmarkRoots.RootGUID, title: "", childrenGuids: ["bm-guid0"])
            ])
        XCTAssert(applyResult.value.isSuccess)
        EarlGrey.selectElement(with: grey_accessibilityID("url")).perform(grey_tap())
        EarlGrey.selectElement(with: grey_accessibilityLabel("Bookmarks")).perform(grey_tap())
        
        // is this in the root?
        assertRowExists(withTitle: "xyz")
        
        navigateFolder(withTitle: "Desktop Bookmarks")
        // this should be missing, they are shown in the root
        EarlGrey.selectElement(with: grey_accessibilityLabel("Mobile Bookmarks")).assert(grey_nil())
        
        // TEST: Add a local bookmark, and then navigate back to the root view (the navigation will refresh the table).
        let isAdded = (bookmarks as! MergedSQLiteBookmarks).local.addToMobileBookmarks(URL(string: "http://another-unused")!, title: "123", favicon: nil)
        XCTAssert(isAdded.value.isSuccess)
        navigateBackInTableView()
        // is this in the root?
        assertRowExists(withTitle: "123")
        
        // TEST: Add sub-folder to MobileFolderGUID, ensure it is navigable
        applyResult = bookmarks.applyRecords([
            makeBookmark(guid: "bm-guid1", parentID: "folder-guid0", title: "item-in-remote-subfolder"),
            makeFolder(guid: "folder-guid0", parentID: BookmarkRoots.MobileFolderGUID, title: "remote-subfolder", childrenGuids: ["bm-guid1"]),
            makeFolder(guid: BookmarkRoots.MobileFolderGUID, parentID: BookmarkRoots.RootGUID, title: "", childrenGuids: ["bm-guid0", "folder-guid0"])
            ])
        XCTAssert(applyResult.value.isSuccess)
        
        // refresh view
        navigateFolder(withTitle: "Desktop Bookmarks")
        navigateBackInTableView()
        
        // subfolder should now be in the root
        assertRowExists(withTitle: "remote-subfolder")
        
        // navigate remote mobile subfolder, ensure bookmark is shown in the subfolder
        navigateFolder(withTitle: "remote-subfolder")
        assertRowExists(withTitle: "item-in-remote-subfolder")
    }

    // This is a continuation of the above test
    func verifyMobileBookmarkDelete() {
        // Assert precondition that this test continues from previous state
        assertRowExists(withTitle: "item-in-remote-subfolder")

        let deleteAction = GREYMatchers.matcher(forText: "Remove Bookmark")

        // Test deletion not available

        var row = EarlGrey.selectElement(with: grey_accessibilityLabel("item-in-remote-subfolder"))
        row.perform(grey_longPress())
        // verify the context menu does not have deletion option
        EarlGrey.selectElement(with: deleteAction).assert(grey_nil())
        // close the context menu
        EarlGrey.selectElement(with: GREYMatchers.matcher(forText: "Pin to Top Sites")).perform(grey_tap())

        navigateBackInTableView()

        // Test successful deletion

        assertRowExists(withTitle: "xyz")

        let tv = tester().waitForView(withAccessibilityIdentifier: "Bookmarks List") as! UITableView
        let rowCount = tv.numberOfRows(inSection: 0)

        row = EarlGrey.selectElement(with: grey_accessibilityLabel("xyz"))
        row.perform(grey_longPress())
        EarlGrey.selectElement(with: deleteAction).perform(grey_tap())

        XCTAssert(tv.numberOfRows(inSection: 0) == rowCount - 1)
        
        EarlGrey.selectElement(with: grey_accessibilityID("goBack")).perform(grey_tap())
    }
  
    func testRefreshBookmarks() {
        createSomeBufferBookmarks()
        guard let bookmarks = getAppBookmarkStorage() else {
            return
        }
        let applyResult = bookmarks.applyRecords([
            makeBookmark(guid: "bm-guid0", parentID: BookmarkRoots.MobileFolderGUID, title: "xyz"),
            makeFolder(guid: BookmarkRoots.MobileFolderGUID, parentID: BookmarkRoots.RootGUID, title: "", childrenGuids: ["bm-guid0"])
            ])
        XCTAssert(applyResult.value.isSuccess)
        EarlGrey.selectElement(with: grey_accessibilityID("url")).perform(grey_tap())
        EarlGrey.selectElement(with: grey_accessibilityLabel("Bookmarks")).perform(grey_tap())
        
        // Add a bookmark
        let isAdded = (bookmarks as! MergedSQLiteBookmarks).local.addToMobileBookmarks(URL(string: "http://new-bookmark")!, title: "NewBookmark", favicon: nil)
        XCTAssert(isAdded.value.isSuccess)
        
        // Pull to refresh
        assertRowExists(withTitle: "xyz")
        EarlGrey.selectElement(with: grey_accessibilityLabel("xyz")).inRoot(grey_kindOfClass(NSClassFromString("UITableView")!)).perform(grey_swipeSlowInDirectionWithStartPoint(.down, 0.7, 0.7))
        
        // Verify new bookmark exists
        assertRowExists(withTitle: "NewBookmark")

        EarlGrey.selectElement(with: grey_accessibilityID("goBack")).perform(grey_tap())
    }
}