test_handler.py

import unittest
from unittest.mock import patch, mock_open
import handler

class TestHandler(unittest.TestCase):
    @patch('builtins.open', new_callable=mock_open)
    @patch('subprocess.Popen')
    def test_process_data_success(self, mock_popen, mock_file):
        mock_popen_instance = mock_popen.return_value
        mock_popen_instance.communicate.return_value = (b'', b'')
        mock_popen_instance.returncode = 0
        mock_file().read.return_value = "encoded_value1,encoded_value2\n"

        result = handler.process_data({"itemKeyArray": ["1"], "ip": "127.0.0.1", "port": "5000"}, "input.txt", "output.txt")
        self.assertEqual(result["status"], "success")

    @patch('builtins.open', new_callable=mock_open)
    @patch('subprocess.Popen')
    def test_process_data_failure(self, mock_popen, mock_file):
        mock_popen_instance = mock_popen.return_value
        mock_popen_instance.communicate.return_value = (b'', b'')
        mock_popen_instance.returncode = 1

        result = handler.process_data({"itemKeyArray": ["1"], "ip": "127.0.0.1", "port": "5000"}, "input.txt", "output.txt")
        self.assertEqual(result["status"], "fail")

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