test_my_app.py

# Copyright 2022 Google LLC
#
# Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
# https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
# <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
# option. This file may not be copied, modified, or distributed
# except according to those terms.

# For more information on unittest, see:
#   https://docs.python.org/3/library/unittest.html

import unittest
from unittest.mock import patch

from apache_beam.testing.test_pipeline import TestPipeline
from apache_beam.testing.util import assert_that, equal_to

import my_app


@patch("apache_beam.Pipeline", TestPipeline)
@patch("builtins.print", lambda x: x)
class TestApp(unittest.TestCase):
    def test_run_direct_runner(self):
        # Note that the order of the elements doesn't matter.
        expected = ["Test", "Hello", "World!"]
        my_app.run(
            input_text="Test",
            test=lambda elements: assert_that(elements, equal_to(expected)),
        )


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