Added a test for the Printer class. Note that this only checks that a PDF has actually been created. It does not check the contents of the PDF file.

pyqso/logbook.py

@@ -491,7 +491,7 @@ class Logbook:
         if(page_index == 0):  # If we are on the Summary page...
             logging.debug("No log currently selected!")
             return
-        page = self.notebook.get_nth_page(page_index)  # Gets the Gtk.VBox of the selected tab in the logbook.
+        page = self.notebook.get_nth_page(page_index)  # Get the Gtk.VBox of the selected tab in the logbook.
         old_log_name = page.get_name()
 
         log_index = self.get_log_index(name=old_log_name)

pyqso/printer.py

@@ -53,9 +53,8 @@ class Printer(object):
         for r in records:
             self.text_to_print += str(r["CALL"]) + "\t---\t" + str(r["QSO_DATE"]) + "\t---\t" + str(r["TIME_ON"]) + "\t---\t" + str(r["FREQ"]) + "\t---\t" + str(r["MODE"]) + "\n"
 
-        self.operation.run(self.action, parent=self.application.window)
-
-        return
+        result = self.operation.run(self.action, parent=self.application.window)
+        return result
 
     def begin_print(self, operation, context):
         """ Specify the layout/position/font of the text on the pages to be printed.

tests/test_logbook.py

@@ -47,7 +47,7 @@ class TestLogbook(unittest.TestCase):
         assert(self.logbook.logs[1].name == "test2")
 
     def tearDown(self):
-        """ Disconnect from the test database. """
+        """ Close the logbook and disconnect from the test database. """
         self.logbook.notebook.get_n_pages.return_value = 0
         closed = self.logbook.close()
         assert(closed)

tests/test_printer.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+
+#    Copyright (C) 2017 Christian Thomas Jacobs.
+
+#    This file is part of PyQSO.
+
+#    PyQSO is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    PyQSO 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 PyQSO.  If not, see <http://www.gnu.org/licenses/>.
+
+from gi.repository import Gtk
+import unittest
+try:
+    import unittest.mock as mock
+except ImportError:
+    import mock
+import os
+from pyqso.printer import *
+
+
+class TestPrinter(unittest.TestCase):
+
+    """ The unit tests for the Printer class. """
+
+    def setUp(self):
+        """ Set up the Printer object. """
+        PyQSO = mock.MagicMock()
+        self.printer = Printer(application=PyQSO())
+        self.printer.application.window = Gtk.Window()
+
+    def tearDown(self):
+        """ Destroy any unit test resources. """
+        return
+
+    def test_print_records(self):
+        """ Check that a list of records can be printed to a PDF file. """
+        self.printer.action = Gtk.PrintOperationAction.EXPORT
+        pdf = "Printer.test_print_records.pdf"
+        self.printer.operation.set_export_filename(pdf)
+        records = [{"CALL":"MYCALL", "QSO_DATE":"24062017", "TIME_ON":"1519", "FREQ":"145.550", "MODE":"FM"}]
+        result = self.printer.print_records(records)
+        assert(result != Gtk.PrintOperationResult.ERROR)
+        assert(result == Gtk.PrintOperationResult.APPLY)
+        assert(os.path.exists(pdf))
+
+if(__name__ == '__main__'):
+    unittest.main()