import os
import unittest

from src.transcriber import Transcriber


class TranscriberTest(unittest.TestCase):
    def setUp(self):
        self.transcriber = Transcriber()

    def test_transcribe(self):
        wav_file = '<Your wav file>'  # Replace with path to a .wav file on your machine
        result = self.transcriber.transcribe(wav_file)
        self.assertIsNotNone(result)

    def test_save_transcript(self):
        wav_file = '<Your wav file>'  # Replace with path to a .wav file on your machine
        result = self.transcriber.transcribe(wav_file)

        json_output_file = '<Path to your temporary JSON output file>'  # Temporary output file for testing
        txt_output_file = '<Path to your temporary TXT output file>'  # Temporary output file for testing
        self.transcriber.save_transcript(json_output_file, txt_output_file)

        self.assertTrue(os.path.exists(json_output_file))
        self.assertTrue(os.path.exists(txt_output_file))

        os.remove(json_output_file)  # clean up after test
        os.remove(txt_output_file)  # clean up after test


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