1
0
forked from um/cli
cli/algo/qmc/qmc_test.go

102 lines
2.2 KiB
Go
Raw Normal View History

2021-12-13 12:42:07 +00:00
package qmc
import (
"bytes"
"fmt"
2021-12-13 12:42:07 +00:00
"io"
"os"
"reflect"
"testing"
2022-12-04 15:14:06 +00:00
"unlock-music.dev/cli/algo/common"
2021-12-13 12:42:07 +00:00
)
func loadTestDataQmcDecoder(filename string) ([]byte, []byte, error) {
encBody, err := os.ReadFile(fmt.Sprintf("./testdata/%s_raw.bin", filename))
2021-12-13 12:42:07 +00:00
if err != nil {
return nil, nil, err
}
encSuffix, err := os.ReadFile(fmt.Sprintf("./testdata/%s_suffix.bin", filename))
2021-12-13 12:42:07 +00:00
if err != nil {
return nil, nil, err
}
target, err := os.ReadFile(fmt.Sprintf("./testdata/%s_target.bin", filename))
2021-12-13 12:42:07 +00:00
if err != nil {
return nil, nil, err
}
return bytes.Join([][]byte{encBody, encSuffix}, nil), target, nil
}
func TestMflac0Decoder_Read(t *testing.T) {
tests := []struct {
name string
2022-12-04 15:14:06 +00:00
fileExt string
wantErr bool
}{
2022-12-04 15:14:06 +00:00
{"mflac0_rc4", ".mflac0", false},
{"mflac_rc4", ".mflac", false},
{"mflac_map", ".mflac", false},
{"mgg_map", ".mgg", false},
{"qmc0_static", ".qmc0", false},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
raw, target, err := loadTestDataQmcDecoder(tt.name)
if err != nil {
t.Fatal(err)
}
2022-12-04 15:14:06 +00:00
d := NewDecoder(&common.DecoderParams{
Reader: bytes.NewReader(raw),
Extension: tt.fileExt,
})
2022-11-19 18:53:04 +00:00
if err := d.Validate(); err != nil {
t.Errorf("validate file error = %v", err)
}
2022-11-19 18:53:04 +00:00
buf := make([]byte, len(target))
if _, err := io.ReadFull(d, buf); err != nil {
t.Errorf("read bytes from decoder error = %v", err)
return
}
if !reflect.DeepEqual(buf, target) {
t.Errorf("Decrypt() got = %v, want %v", buf[:32], target[:32])
}
})
2021-12-13 12:42:07 +00:00
}
}
func TestMflac0Decoder_Validate(t *testing.T) {
tests := []struct {
name string
fileExt string
wantErr bool
}{
{"mflac0_rc4", ".flac", false},
{"mflac_map", ".flac", false},
{"mgg_map", ".ogg", false},
{"qmc0_static", ".mp3", false},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
raw, _, err := loadTestDataQmcDecoder(tt.name)
if err != nil {
t.Fatal(err)
}
2022-12-04 15:14:06 +00:00
d := NewDecoder(&common.DecoderParams{
Reader: bytes.NewReader(raw),
Extension: tt.fileExt,
})
if err := d.Validate(); err != nil {
t.Errorf("read bytes from decoder error = %v", err)
return
}
})
}
}