Unit Test for Ansible Module and Plugin

test_module.py

import pytest
from ansible.module_utils.basic import AnsibleModule
from my_plugin import main

@pytest.fixture
def fake_module():
    module = ansibleModule(
        argument_spec=dict(
            name=dict(type='str'),
            age=dict(type='int')
        )
    )
    return module

def test_main(fake_module):
    # Set the module's parameters
    fake_module.params = dict(name='John', age=30)
    # Call the plugin's main function
    main()
    # Assert that the plugin returned the expected results
    assert fake_module.exit_json.called
    assert fake_module.exit_json.call_args[0][0]['changed'] == False
    assert fake_module.exit_json.call_args[0][0]['message'] == 'Hello John, you are 30 years old.'

test_plugin.py

import pytest
from ansible.module_utils.basic import AnsibleModule
from my_plugin import main

@pytest.fixture
def fake_module():
    module = ansibleModule(
        argument_spec=dict(
            name=dict(type='str'),
            age=dict(type='int')
        )
    )
    return module

def test_main(fake_module):
    # Set the module's parameters
    fake_module.params = dict(name='John', age=30)
    # Call the plugin's main function
    main()
    # Assert that the plugin returned the expected results
    assert fake_module.exit_json.called
    assert fake_module.exit_json.call_args[0][0]['changed'] == False
    assert fake_module.exit_json.call_args[0][0]['message'] == 'Hello John, you are 30 years old.'