Source code for seriesmarker.test.gui.search_dialog_test

# ==============================================================================
# -*- coding: utf-8 -*-
#
# Copyright (C) 2013 - 2016 Tobias Röttger <toroettg@gmail.com>
#
# This file is part of SeriesMarker.
#
# SeriesMarker is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3 as
# published by the Free Software Foundation.
#
# SeriesMarker is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with SeriesMarker.  If not, see <http://www.gnu.org/licenses/>.
# ==============================================================================

import unittest
from collections import namedtuple

from PySide.QtCore import Qt

from PySide.QtGui import QTableView

from PySide.QtTest import QTest

from seriesmarker.gui.search_dialog import SearchDialog
from seriesmarker.test.gui.base.gui_test_case import GUITestCase

Search = namedtuple("Search", "string,expected_row,expected_count")



[docs]class SearchDialogTest(GUITestCase):

[docs]    def setUp(self):


        self.search = Search(string="how i", expected_row=2, expected_count=6)

        self.dialog = SearchDialog()

        self.dialog.show()

        QTest.qWaitForWindowShown(self.dialog)

        self.assertEqual(self.dialog.ui.search_text_field.text(), "",
                         "Default Search Text is not empty")
        self.assertEqual(self.dialog.model.rowCount(), 0,
                         "Result model not cleared")




[docs]    def test_search_by_keyhit(self):
        self.type(self.dialog.ui.search_text_field, self.search.string)
        self.keyhit(self.dialog.ui.search_text_field, Qt.Key_Enter)

        self.assertEqual(self.dialog.model.rowCount(),
                         self.search.expected_count,
                         "Search did not return expected results")
        self.assertEqual(self.dialog.model.data(
            self.dialog.model.index(self.search.expected_row, 0)),
                         "How I Met Your Mother",
                         "First result not as expected")





[docs]    def test_reject(self):
        self.click(self.dialog.ui.cancel_button)

        self.assertEqual(self.dialog.result(), self.dialog.Rejected,
                         "Dialog did not reject")



[docs]    def test_accept(self):
        self.click(self.dialog.ui.ok_button)
        self.assertEqual(self.dialog.result(), self.dialog.Rejected,
                         "Dialog did accept without search or selection")
        self.assertEqual(self.dialog.result_value(), None,
                         "Accept without search changed result")

        self.type(self.dialog.ui.search_text_field, "how i met")
        self.click(self.dialog.ui.search_button)

        self.click(self.dialog.ui.ok_button)
        self.assertEqual(self.dialog.result(), self.dialog.Rejected,
                         "Dialog did accept without selection")
        self.assertEqual(self.dialog.result_value(), None,
                         "Accept without selection changed result")

        view = self.dialog.findChild(QTableView, "result_view")
        viewport = view.viewport()
        item = view.model().index(0, 1)
        item_rect = view.visualRect(item)
        self.click(viewport, item_rect.center())
        self.click(self.dialog.ui.ok_button)

        self.assertEqual(self.dialog.result(), self.dialog.Accepted,
                         "Dialog did not accept")
        self.assertEqual(self.dialog.result_value().SeriesName,
                         "How I Met Your Mother",
                         "Accept with selection did not return correct search result")



[docs]    def tearDown(self):
        QTest.mouseMove(self.dialog,
                        delay=2000)  # Emulates waiting, can be removed

        self.dialog.close()




[docs]def get_suit():
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(SearchDialogTest))
    return suite



if __name__ == "__main__":
    unittest.main()