diago/recording_test.go at main · emiago/diago · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package diago

import (
	"bytes"
	"os"
	"testing"

	"github.com/emiago/diago/audio"
	"github.com/emiago/diago/media"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestIntegrationRecordingStereoWav(t *testing.T) {
	fakePCMFrame := bytes.Repeat([]byte("0123456789"), 32)
	alawFrame := make([]byte, 160)
	_, err := audio.EncodeAlawTo(alawFrame, fakePCMFrame)
	require.NoError(t, err)
	encodedAudio := bytes.Repeat(alawFrame, 4)

	dialog := &DialogServerSession{
		DialogMedia: DialogMedia{
			mediaSession: &media.MediaSession{Codecs: []media.Codec{media.CodecAudioUlaw}},
			// audioReader:  bytes.NewBuffer(make([]byte, 9999)),
			audioReader:     bytes.NewBuffer(encodedAudio),
			audioWriter:     bytes.NewBuffer([]byte{}),
			RTPPacketWriter: media.NewRTPPacketWriter(nil, media.CodecAudioUlaw),
		},
	}

	recordFile, err := os.OpenFile("/tmp/diago_test_record_stereo.wav", os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0755)
	require.NoError(t, err)
	defer recordFile.Close()

	rec, err := dialog.AudioStereoRecordingCreate(recordFile)
	require.NoError(t, err)

	media.ReadAll(rec.AudioReader(), 160)
	media.WriteAll(rec.AudioWriter(), encodedAudio, 160)
	err = rec.Close()

	recordFile.Seek(0, 0)
	wav := audio.NewWavReader(recordFile)
	wav.ReadHeaders()
	// 2 channels, 4 frames Read, 4 frames Write
	assert.Equal(t, 2*4*320, wav.DataSize)
}