Ejemplo n.º 1
0
    def test_downloaded_file_is_NOT_re_extracted_after_modified(self):
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        # File is auto-extracted
        file_one = ModelFile("File.One", True)
        file_one.local_size = 100
        file_one.state = ModelFile.State.DOWNLOADED
        file_one.is_extractable = True
        self.model_listener.file_added(file_one)
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.EXTRACT, command.action)
        self.assertEqual("File.One", command.filename)
        self.controller.queue_command.reset_mock()

        # File is modified
        file_one_new = ModelFile("File.One", True)
        file_one_new.local_size = 101
        file_one_new.state = ModelFile.State.DOWNLOADED
        file_one_new.is_extractable = True
        self.model_listener.file_updated(file_one, file_one_new)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 2
0
    def test_partial_file_is_auto_queued_after_remote_discovery(self):
        # Test that a partial local file is auto-queued when discovered on remote some time later
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        # Local discovery
        file_one = ModelFile("File.One", True)
        file_one.local_size = 100
        self.model_listener.file_added(file_one)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        # Remote discovery
        file_one_new = ModelFile("File.One", True)
        file_one_new.local_size = 100
        file_one_new.remote_size = 200
        self.model_listener.file_updated(file_one, file_one_new)
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.One", command.filename)
Ejemplo n.º 3
0
    def test_downloaded_file_with_changed_remote_size_is_queued(self):
        # Disable auto-extract
        self.context.config.autoqueue.auto_extract = False

        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)
        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        file_one.local_size = 100
        file_one.state = ModelFile.State.DOWNLOADED
        self.model_listener.file_added(file_one)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        file_one_updated = ModelFile("File.One", True)
        file_one_updated.remote_size = 200
        file_one_updated.local_size = 100
        file_one_updated.state = ModelFile.State.DEFAULT
        self.model_listener.file_updated(file_one, file_one_updated)
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.One", command.filename)
Ejemplo n.º 4
0
 def test_listener_pattern_removed_non_existing(self):
     listener = TestAutoQueuePersistListener()
     listener.pattern_removed = MagicMock()
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     persist.add_pattern(AutoQueuePattern(pattern="two"))
     persist.add_pattern(AutoQueuePattern(pattern="three"))
     persist.add_listener(listener)
     persist.remove_pattern(AutoQueuePattern(pattern="four"))
     listener.pattern_removed.assert_not_called()
Ejemplo n.º 5
0
 def test_matching_queued_files_are_not_queued(self):
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="File.One"))
     # noinspection PyTypeChecker
     auto_queue = AutoQueue(self.context, persist, self.controller)
     file_one = ModelFile("File.One", True)
     file_one.remote_size = 100
     file_one.state = ModelFile.State.QUEUED
     self.model_listener.file_added(file_one)
     auto_queue.process()
     self.controller.queue_command.assert_not_called()
Ejemplo n.º 6
0
 def test_partial_matches(self):
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="One"))
     # noinspection PyTypeChecker
     auto_queue = AutoQueue(self.context, persist, self.controller)
     file_one = ModelFile("File.One", True)
     file_one.remote_size = 100
     self.model_listener.file_added(file_one)
     auto_queue.process()
     command = self.controller.queue_command.call_args[0][0]
     self.assertEqual(Controller.Command.Action.QUEUE, command.action)
     self.assertEqual("File.One", command.filename)
Ejemplo n.º 7
0
    def test_persist_read_error(self):
        # bad pattern
        content = """
        {
            "patterns": [
                "bad string"
            ]
        }
        """
        with self.assertRaises(PersistError):
            AutoQueuePersist.from_str(content)

        # empty json
        content = ""
        with self.assertRaises(PersistError):
            AutoQueuePersist.from_str(content)

        # missing keys
        content = "{}"
        with self.assertRaises(PersistError):
            AutoQueuePersist.from_str(content)

        # malformed
        content = "{"
        with self.assertRaises(PersistError):
            AutoQueuePersist.from_str(content)
Ejemplo n.º 8
0
    def test_matching_downloaded_files_are_not_queued(self):
        # Disable auto-extract
        self.context.config.autoqueue.auto_extract = False

        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)
        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        file_one.state = ModelFile.State.DOWNLOADED
        self.model_listener.file_added(file_one)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 9
0
 def test_add_blank_pattern_fails(self):
     persist = AutoQueuePersist()
     with self.assertRaises(ValueError):
         persist.add_pattern(AutoQueuePattern(pattern=""))
     with self.assertRaises(ValueError):
         persist.add_pattern(AutoQueuePattern(pattern=" "))
     with self.assertRaises(ValueError):
         persist.add_pattern(AutoQueuePattern(pattern="   "))
Ejemplo n.º 10
0
    def test_new_matching_pattern_doesnt_queue_local_file(self):
        persist = AutoQueuePersist()

        file_one = ModelFile("File.One", True)
        file_one.local_size = 100

        self.initial_model = [file_one]

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 11
0
 def test_listener_pattern_added_duplicate(self):
     listener = TestAutoQueuePersistListener()
     listener.pattern_added = MagicMock()
     persist = AutoQueuePersist()
     persist.add_listener(listener)
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     listener.pattern_added.assert_called_once_with(
         AutoQueuePattern(pattern="one"))
     listener.pattern_added.reset_mock()
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     listener.pattern_added.assert_not_called()
Ejemplo n.º 12
0
    def test_new_matching_pattern_queues_existing_files(self):
        persist = AutoQueuePersist()

        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        file_two = ModelFile("File.Two", True)
        file_two.remote_size = 200
        file_three = ModelFile("File.Three", True)
        file_three.remote_size = 300
        file_four = ModelFile("File.Four", True)
        file_four.remote_size = 400
        file_five = ModelFile("File.Five", True)
        file_five.remote_size = 500

        self.initial_model = [
            file_one, file_two, file_three, file_four, file_five
        ]

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.One", command.filename)
        self.controller.queue_command.reset_mock()

        persist.add_pattern(AutoQueuePattern(pattern="File.Two"))
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.Two", command.filename)
        self.controller.queue_command.reset_mock()

        persist.add_pattern(AutoQueuePattern(pattern="File.Three"))
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.Three", command.filename)
        self.controller.queue_command.reset_mock()

        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 13
0
 def test_matching_downloading_files_are_not_queued(self):
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="File.One"))
     # noinspection PyTypeChecker
     auto_queue = AutoQueue(self.context, persist, self.controller)
     file_one = ModelFile("File.One", True)
     file_one.remote_size = 100
     file_one.local_size = 0
     file_one.state = ModelFile.State.DOWNLOADING
     self.model_listener.file_added(file_one)
     auto_queue.process()
     self.controller.queue_command.assert_not_called()
     file_one_new = ModelFile("File.One", True)
     file_one_new.remote_size = 100
     file_one_new.local_size = 50
     file_one_new.state = ModelFile.State.DOWNLOADING
     self.model_listener.file_updated(file_one, file_one_new)
     auto_queue.process()
     self.controller.queue_command.assert_not_called()
Ejemplo n.º 14
0
    def test_matching_initial_files_are_queued(self):
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Two"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Three"))

        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        file_two = ModelFile("File.Two", True)
        file_two.remote_size = 200
        file_three = ModelFile("File.Three", True)
        file_three.remote_size = 300
        file_four = ModelFile("File.Four", True)
        file_four.remote_size = 400
        file_five = ModelFile("File.Five", True)
        file_five.remote_size = 500

        self.initial_model = [
            file_one, file_two, file_three, file_four, file_five
        ]

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        auto_queue.process()

        calls = self.controller.queue_command.call_args_list
        self.assertEqual(3, len(calls))
        commands = [calls[i][0][0] for i in range(3)]
        self.assertEqual(set([Controller.Command.Action.QUEUE] * 3),
                         {c.action
                          for c in commands})
        self.assertEqual({"File.One", "File.Two", "File.Three"},
                         {c.filename
                          for c in commands})
Ejemplo n.º 15
0
    def test_non_extractable_files_are_not_extracted(self):
        persist = AutoQueuePersist()

        file_one = ModelFile("File.One", True)
        file_one.local_size = 100
        file_one.state = ModelFile.State.DOWNLOADED
        file_one.is_extractable = True
        file_two = ModelFile("File.Two", True)
        file_two.local_size = 200
        file_two.state = ModelFile.State.DOWNLOADED
        file_two.is_extractable = False

        self.initial_model = [file_one, file_two]

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.EXTRACT, command.action)
        self.assertEqual("File.One", command.filename)
        self.controller.queue_command.reset_mock()

        persist.add_pattern(AutoQueuePattern(pattern="File.Two"))
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 16
0
    def test_adding_then_removing_pattern_doesnt_queue_existing_file(self):
        persist = AutoQueuePersist()

        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        file_two = ModelFile("File.Two", True)
        file_two.remote_size = 200
        file_three = ModelFile("File.Three", True)
        file_three.remote_size = 300
        file_four = ModelFile("File.Four", True)
        file_four.remote_size = 400
        file_five = ModelFile("File.Five", True)
        file_five.remote_size = 500

        self.initial_model = [
            file_one, file_two, file_three, file_four, file_five
        ]

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        persist.remove_pattern(AutoQueuePattern(pattern="File.One"))
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 17
0
    def test_auto_queued_file_not_re_queued_after_stopping(self):
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)
        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        self.model_listener.file_added(file_one)
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.One", command.filename)

        file_one_updated = ModelFile("File.One", True)
        file_one_updated.remote_size = 100
        file_one_updated.local_size = 50
        self.model_listener.file_updated(file_one, file_one_updated)
        auto_queue.process()
        self.controller.queue_command.assert_called_once_with(
            unittest.mock.ANY)
Ejemplo n.º 18
0
 def test_add_pattern(self):
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     persist.add_pattern(AutoQueuePattern(pattern="two"))
     self.assertEqual(
         {AutoQueuePattern(pattern="one"),
          AutoQueuePattern(pattern="two")}, persist.patterns)
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     persist.add_pattern(AutoQueuePattern(pattern="three"))
     self.assertEqual(
         {
             AutoQueuePattern(pattern="one"),
             AutoQueuePattern(pattern="two"),
             AutoQueuePattern(pattern="three")
         }, persist.patterns)
Ejemplo n.º 19
0
    def test_matching_new_files_are_extracted(self):
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Two"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Three"))

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        file_one = ModelFile("File.One", True)
        file_one.state = ModelFile.State.DOWNLOADED
        file_one.local_size = 100
        file_one.is_extractable = True
        file_two = ModelFile("File.Two", True)
        file_two.state = ModelFile.State.DOWNLOADED
        file_two.local_size = 200
        file_two.is_extractable = True
        file_three = ModelFile("File.Three", True)
        file_three.state = ModelFile.State.DOWNLOADED
        file_three.local_size = 300
        file_three.is_extractable = True

        self.model_listener.file_added(file_one)
        auto_queue.process()
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.EXTRACT, command.action)
        self.assertEqual("File.One", command.filename)

        self.model_listener.file_added(file_two)
        auto_queue.process()
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.EXTRACT, command.action)
        self.assertEqual("File.Two", command.filename)

        self.model_listener.file_added(file_three)
        auto_queue.process()
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.EXTRACT, command.action)
        self.assertEqual("File.Three", command.filename)

        # All at once
        self.model_listener.file_added(file_one)
        self.model_listener.file_added(file_two)
        self.model_listener.file_added(file_three)
        auto_queue.process()
        calls = self.controller.queue_command.call_args_list[-3:]
        commands = [calls[i][0][0] for i in range(3)]
        self.assertEqual(set([Controller.Command.Action.EXTRACT] * 3),
                         {c.action
                          for c in commands})
        self.assertEqual({"File.One", "File.Two", "File.Three"},
                         {c.filename
                          for c in commands})
Ejemplo n.º 20
0
    def setUp(self):
        self.context = MagicMock()
        self.controller = MagicMock()

        # Mock the base logger
        logger = logging.getLogger()
        handler = logging.StreamHandler(sys.stdout)
        logger.addHandler(handler)
        logger.setLevel(logging.DEBUG)
        formatter = logging.Formatter(
            "%(asctime)s - %(levelname)s - %(name)s - %(message)s")
        handler.setFormatter(formatter)
        self.context.logger = logger

        # Model files
        self.model_files = []

        # Real status
        self.context.status = Status()

        # Real config
        self.context.config = Config()

        # Real auto-queue persist
        self.auto_queue_persist = AutoQueuePersist()

        # Capture the model listener
        def capture_listener(listener):
            self.model_listener = listener
            return self.model_files

        self.model_listener = None
        self.controller.get_model_files_and_add_listener = MagicMock()
        self.controller.get_model_files_and_add_listener.side_effect = capture_listener
        self.controller.remove_model_listener = MagicMock()

        # noinspection PyTypeChecker
        self.web_app_builder = WebAppBuilder(self.context, self.controller,
                                             self.auto_queue_persist)
        self.web_app = self.web_app_builder.build()
        self.test_app = TestApp(self.web_app)
Ejemplo n.º 21
0
 def test_from_str(self):
     content = """
     {{
         "patterns": [
             "{}",
             "{}",
             "{}",
             "{}",
             "{}",
             "{}"
         ]
     }}
     """.format(
         AutoQueuePattern(pattern="one").to_str().replace("\\",
                                                          "\\\\").replace(
                                                              "\"", "\\\""),
         AutoQueuePattern(pattern="two").to_str().replace("\\",
                                                          "\\\\").replace(
                                                              "\"", "\\\""),
         AutoQueuePattern(pattern="th ree").to_str().replace(
             "\\", "\\\\").replace("\"", "\\\""),
         AutoQueuePattern(pattern="fo.ur").to_str().replace(
             "\\", "\\\\").replace("\"", "\\\""),
         AutoQueuePattern(pattern="fi\"ve").to_str().replace(
             "\\", "\\\\").replace("\"", "\\\""),
         AutoQueuePattern(pattern="si'x").to_str().replace(
             "\\", "\\\\").replace("\"", "\\\""))
     print(content)
     print(AutoQueuePattern(pattern="fi\"ve").to_str())
     persist = AutoQueuePersist.from_str(content)
     golden_patterns = {
         AutoQueuePattern(pattern="one"),
         AutoQueuePattern(pattern="two"),
         AutoQueuePattern(pattern="th ree"),
         AutoQueuePattern(pattern="fo.ur"),
         AutoQueuePattern(pattern="fi\"ve"),
         AutoQueuePattern(pattern="si'x")
     }
     self.assertEqual(golden_patterns, persist.patterns)
Ejemplo n.º 22
0
    def test_all_files_are_extracted_when_patterns_only_disabled(self):
        self.context.config.autoqueue.patterns_only = False

        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Two"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Three"))

        file_one = ModelFile("File.One", True)
        file_one.local_size = 100
        file_one.state = ModelFile.State.DOWNLOADED
        file_one.is_extractable = True
        file_two = ModelFile("File.Two", True)
        file_two.local_size = 200
        file_two.state = ModelFile.State.DOWNLOADED
        file_two.is_extractable = True
        file_three = ModelFile("File.Three", True)
        file_three.local_size = 300
        file_three.state = ModelFile.State.DOWNLOADED
        file_three.is_extractable = True
        file_four = ModelFile("File.Four", True)
        file_four.local_size = 400
        file_four.state = ModelFile.State.DOWNLOADED
        file_four.is_extractable = True
        file_five = ModelFile("File.Five", True)
        file_five.local_size = 500
        file_five.state = ModelFile.State.DOWNLOADED
        file_five.is_extractable = True

        self.initial_model = [
            file_one, file_two, file_three, file_four, file_five
        ]

        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)
        auto_queue.process()
        calls = self.controller.queue_command.call_args_list
        self.assertEqual(5, len(calls))
        commands = [calls[i][0][0] for i in range(5)]
        self.assertEqual(set([Controller.Command.Action.EXTRACT] * 5),
                         {c.action
                          for c in commands})
        self.assertEqual(
            {"File.One", "File.Two", "File.Three", "File.Four", "File.Five"},
            {c.filename
             for c in commands})
Ejemplo n.º 23
0
    def test_no_files_are_extracted_when_auto_extract_disabled(self):
        self.context.config.autoqueue.enabled = True
        self.context.config.autoqueue.auto_extract = False

        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="File.One"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Two"))
        persist.add_pattern(AutoQueuePattern(pattern="File.Three"))

        file_one = ModelFile("File.One", True)
        file_one.local_size = 100
        file_one.state = ModelFile.State.DOWNLOADED
        file_two = ModelFile("File.Two", True)
        file_two.local_size = 200
        file_two.state = ModelFile.State.DOWNLOADED
        file_three = ModelFile("File.Three", True)
        file_three.local_size = 300
        file_three.state = ModelFile.State.DOWNLOADED
        file_four = ModelFile("File.Four", True)
        file_four.local_size = 400
        file_four.state = ModelFile.State.DOWNLOADED
        file_five = ModelFile("File.Five", True)
        file_five.local_size = 500
        file_five.state = ModelFile.State.DOWNLOADED

        self.initial_model = [
            file_one, file_two, file_three, file_four, file_five
        ]

        # First with patterns_only ON
        self.context.config.autoqueue.patterns_only = True
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()

        # Second with patterns_only OFF
        self.context.config.autoqueue.patterns_only = False
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 24
0
    def test_removed_pattern_doesnt_queue_new_file(self):
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="One"))
        persist.add_pattern(AutoQueuePattern(pattern="Two"))
        # noinspection PyTypeChecker
        auto_queue = AutoQueue(self.context, persist, self.controller)

        file_one = ModelFile("File.One", True)
        file_one.remote_size = 100
        self.model_listener.file_added(file_one)
        auto_queue.process()
        command = self.controller.queue_command.call_args[0][0]
        self.assertEqual(Controller.Command.Action.QUEUE, command.action)
        self.assertEqual("File.One", command.filename)
        self.controller.queue_command.reset_mock()

        persist.remove_pattern(AutoQueuePattern(pattern="Two"))

        file_two = ModelFile("File.Two", True)
        file_two.remote_size = 100
        self.model_listener.file_added(file_two)
        auto_queue.process()
        self.controller.queue_command.assert_not_called()
Ejemplo n.º 25
0
 def test_listener_pattern_removed(self):
     listener = TestAutoQueuePersistListener()
     listener.pattern_removed = MagicMock()
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     persist.add_pattern(AutoQueuePattern(pattern="two"))
     persist.add_pattern(AutoQueuePattern(pattern="three"))
     persist.add_listener(listener)
     persist.remove_pattern(AutoQueuePattern(pattern="one"))
     listener.pattern_removed.assert_called_once_with(
         AutoQueuePattern(pattern="one"))
     listener.pattern_removed.reset_mock()
     persist.remove_pattern(AutoQueuePattern(pattern="two"))
     listener.pattern_removed.assert_called_once_with(
         AutoQueuePattern(pattern="two"))
     listener.pattern_removed.reset_mock()
Ejemplo n.º 26
0
 def test_to_str(self):
     persist = AutoQueuePersist()
     persist.add_pattern(AutoQueuePattern(pattern="one"))
     persist.add_pattern(AutoQueuePattern(pattern="two"))
     persist.add_pattern(AutoQueuePattern(pattern="th ree"))
     persist.add_pattern(AutoQueuePattern(pattern="fo.ur"))
     persist.add_pattern(AutoQueuePattern(pattern="fi\"ve"))
     persist.add_pattern(AutoQueuePattern(pattern="si'x"))
     print(persist.to_str())
     dct = json.loads(persist.to_str())
     self.assertTrue("patterns" in dct)
     self.assertEqual([
         AutoQueuePattern(pattern="one").to_str(),
         AutoQueuePattern(pattern="two").to_str(),
         AutoQueuePattern(pattern="th ree").to_str(),
         AutoQueuePattern(pattern="fo.ur").to_str(),
         AutoQueuePattern(pattern="fi\"ve").to_str(),
         AutoQueuePattern(pattern="si'x").to_str()
     ], dct["patterns"])
Ejemplo n.º 27
0
    def test_to_and_from_str(self):
        persist = AutoQueuePersist()
        persist.add_pattern(AutoQueuePattern(pattern="one"))
        persist.add_pattern(AutoQueuePattern(pattern="two"))
        persist.add_pattern(AutoQueuePattern(pattern="th ree"))
        persist.add_pattern(AutoQueuePattern(pattern="fo.ur"))
        persist.add_pattern(AutoQueuePattern(pattern="fi\"ve"))
        persist.add_pattern(AutoQueuePattern(pattern="si'x"))

        persist_actual = AutoQueuePersist.from_str(persist.to_str())
        self.assertEqual(persist.patterns, persist_actual.patterns)