def test_next_image_last_image(self, mock_view): args = mock.Mock() args.input_dir = None model = Model(args) controller = Controller(model, mock_view) model.images = ["abc.png", "test.png", "ababaa.png"] model.current_file = 2 controller.next_image() self.assertEqual(2, model.current_file)
def test_next_image(self, mock_view, mock_load_image): args = mock.Mock() args.input_dir = None model = Model(args) controller = Controller(model, mock_view) model.images = ["abc.png", "test.png", "ababaa.png"] model.current_file = 1 controller.next_image() mock_load_image.assert_called_with = model.images[2] self.assertEqual(2, model.current_file)