MediaTests.cpp 16.4 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
/*****************************************************************************
 * Media Library
 *****************************************************************************
 * Copyright (C) 2015 Hugo Beauzée-Luyssen, Videolabs
 *
 * Authors: Hugo Beauzée-Luyssen<hugo@beauzee.fr>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

23
24
25
26
#if HAVE_CONFIG_H
# include "config.h"
#endif

27
#include "Tests.h"
28

29
#include "medialibrary/IMediaLibrary.h"
30
#include "File.h"
31
#include "Media.h"
32
33
34
#include "Artist.h"
#include "Album.h"
#include "AlbumTrack.h"
35
#include "mocks/FileSystem.h"
36
#include "mocks/DiscovererCbMock.h"
37
#include "compat/Thread.h"
38

39
class Medias : public Tests
40
41
42
43
{
};


44
TEST_F( Medias, Init )
45
46
47
48
{
    // only test for correct test fixture behavior
}

49
TEST_F( Medias, Create )
50
{
51
52
53
54
55
56
57
58
    auto m = ml->addFile( "media.avi" );
    ASSERT_NE( m, nullptr );

    ASSERT_EQ( m->playCount(), 0 );
    ASSERT_EQ( m->albumTrack(), nullptr );
    ASSERT_EQ( m->showEpisode(), nullptr );
    ASSERT_EQ( m->duration(), -1 );
    ASSERT_NE( 0u, m->insertionDate() );
59
60
61
62
63

    auto files = m->files();
    ASSERT_EQ( 1u, files.size() );
    auto f = files[0];
    ASSERT_FALSE( f->isExternal() );
64
    ASSERT_EQ( File::Type::Main, f->type() );
65
66
}

67
TEST_F( Medias, Fetch )
68
{
69
    auto f = ml->addMedia( "media.avi" );
70
71
    auto f2 = ml->media( f->id() );
    ASSERT_EQ( f->id(), f2->id() );
72
    ASSERT_EQ( f, f2 );
73
74

    // Flush cache and fetch from DB
75
    Reload();
76

77
    f2 = std::static_pointer_cast<Media>( ml->media( f->id() ) );
78
    ASSERT_EQ( f->id(), f2->id() );
79
}
80

81
TEST_F( Medias, Duration )
82
{
83
    auto f = std::static_pointer_cast<Media>( ml->addMedia( "media.avi" ) );
84
    ASSERT_EQ( f->duration(), -1 );
85

86
87
88
89
    // Use a value that checks we're using a 64bits value
    int64_t d = int64_t(1) << 40;

    f->setDuration( d );
90
    f->save();
91
    ASSERT_EQ( f->duration(), d );
92
93
94

    Reload();

95
    auto f2 = ml->media( f->id() );
96
    ASSERT_EQ( f2->duration(), d );
97
}
98

Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
99
TEST_F( Medias, Thumbnail )
100
{
101
    auto f = std::static_pointer_cast<Media>( ml->addMedia( "media.avi" ) );
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
102
    ASSERT_EQ( f->thumbnail(), "" );
103

Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
104
    std::string newThumbnail( "/path/to/thumbnail" );
105

Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
106
    f->setThumbnail( newThumbnail );
107
    f->save();
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
108
    ASSERT_EQ( f->thumbnail(), newThumbnail );
109
110
111

    Reload();

112
    auto f2 = ml->media( f->id() );
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
113
    ASSERT_EQ( f2->thumbnail(), newThumbnail );
114
}
115
116
117

TEST_F( Medias, PlayCount )
{
118
    auto f = std::static_pointer_cast<Media>( ml->addMedia( "media.avi" ) );
119
120
121
122
123
124
    ASSERT_EQ( 0, f->playCount() );
    f->increasePlayCount();
    ASSERT_EQ( 1, f->playCount() );

    Reload();

125
    f = std::static_pointer_cast<Media>( ml->media( f->id() ) );
126
127
    ASSERT_EQ( 1, f->playCount() );
}
128
129
130

TEST_F( Medias, Progress )
{
131
    auto f = std::static_pointer_cast<Media>( ml->addMedia( "media.avi" ) );
132
133
134
    ASSERT_EQ( 0, f->metadata( Media::MetadataType::Progress ).integer() );
    f->setMetadata( Media::MetadataType::Progress, 123 );
    ASSERT_EQ( 123, f->metadata( Media::MetadataType::Progress ).integer() );
135
    ASSERT_TRUE( f->metadata( Media::MetadataType::Progress ).isSet() );
136
137
138
139

    Reload();

    f = ml->media( f->id() );
140
    ASSERT_EQ( 123, f->metadata( Media::MetadataType::Progress ).integer() );
141
142
143
144
}

TEST_F( Medias, Rating )
{
145
    auto f = std::static_pointer_cast<Media>( ml->addMedia( "media.avi" ) );
146
147
148
    ASSERT_FALSE( f->metadata( Media::MetadataType::Rating ).isSet() );
    f->setMetadata( Media::MetadataType::Rating, 12345 );
    ASSERT_EQ( 12345, f->metadata( Media::MetadataType::Rating ).integer() );
149
    ASSERT_TRUE( f->metadata( Media::MetadataType::Rating ).isSet() );
150
151
152
153

    Reload();

    f = ml->media( f->id() );
154
    ASSERT_EQ( 12345, f->metadata( Media::MetadataType::Rating ).integer() );
155
}
156
157
158
159
160

TEST_F( Medias, Search )
{
    for ( auto i = 1u; i <= 10u; ++i )
    {
161
        ml->addMedia( "track " + std::to_string( i ) + ".mp3" );
162
    }
163
    auto media = ml->searchMedia( "tra" ).others;
164
165
    ASSERT_EQ( 10u, media.size() );

166
    media = ml->searchMedia( "track 1" ).others;
167
168
    ASSERT_EQ( 2u, media.size() );

169
    media = ml->searchMedia( "grouik" ).others;
170
171
    ASSERT_EQ( 0u, media.size() );

172
    media = ml->searchMedia( "rack" ).others;
173
174
175
176
177
    ASSERT_EQ( 0u, media.size() );
}

TEST_F( Medias, SearchAfterEdit )
{
178
    auto m = std::static_pointer_cast<Media>( ml->addMedia( "media.mp3" ) );
179

180
    auto media = ml->searchMedia( "media" ).others;
181
182
    ASSERT_EQ( 1u, media.size() );

Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
183
    m->setTitleBuffered( "otters are awesome" );
184
185
    m->save();

186
    media = ml->searchMedia( "media" ).others;
187
188
    ASSERT_EQ( 0u, media.size() );

189
    media = ml->searchMedia( "otters" ).others;
190
191
192
193
194
    ASSERT_EQ( 1u, media.size() );
}

TEST_F( Medias, SearchAfterDelete )
{
195
    auto m = std::static_pointer_cast<Media>( ml->addMedia( "media.mp3" ) );
196

197
    auto media = ml->searchMedia( "media" ).others;
198
199
200
201
202
    ASSERT_EQ( 1u, media.size() );

    auto f = m->files()[0];
    m->removeFile( static_cast<File&>( *f ) );

203
    media = ml->searchMedia( "media" ).others;
204
205
    ASSERT_EQ( 0u, media.size() );
}
206
207
208

TEST_F( Medias, SearchByLabel )
{
209
    auto m = std::static_pointer_cast<Media>( ml->addMedia( "media.mkv" ) );
210
    auto media = ml->searchMedia( "otter" ).others;
211
212
213
214
215
    ASSERT_EQ( 0u, media.size() );

    auto l = ml->createLabel( "otter" );
    m->addLabel( l );

216
    media = ml->searchMedia( "otter" ).others;
217
218
219
220
221
    ASSERT_EQ( 1u, media.size() );

    auto l2 = ml->createLabel( "pangolins" );
    m->addLabel( l2 );

222
    media = ml->searchMedia( "otter" ).others;
223
224
    ASSERT_EQ( 1u, media.size() );

225
    media = ml->searchMedia( "pangolin" ).others;
226
227
228
229
    ASSERT_EQ( 1u, media.size() );

    m->removeLabel( l );

230
    media = ml->searchMedia( "otter" ).others;
231
232
    ASSERT_EQ( 0u, media.size() );

233
    media = ml->searchMedia( "pangolin" ).others;
234
235
236
237
    ASSERT_EQ( 1u, media.size() );

    m->addLabel( l );

238
    media = ml->searchMedia( "otter" ).others;
239
240
    ASSERT_EQ( 1u, media.size() );

241
    media = ml->searchMedia( "pangolin" ).others;
242
243
244
245
    ASSERT_EQ( 1u, media.size() );

    ml->deleteLabel( l );

246
    media = ml->searchMedia( "otter" ).others;
247
248
    ASSERT_EQ( 0u, media.size() );

249
    media = ml->searchMedia( "pangolin" ).others;
250
251
    ASSERT_EQ( 1u, media.size() );
}
252
253
254
255
256
257

TEST_F( Medias, SearchTracks )
{
    auto a = ml->createAlbum( "album" );
    for ( auto i = 1u; i <= 10u; ++i )
    {
258
       auto m = std::static_pointer_cast<Media>( ml->addMedia( "track " + std::to_string( i ) + ".mp3" ) );
259
       a->addTrack( m, i, 1, 0, 0 );
260
261
262
263
264
265
266
267
268
269
270
271
272
    }
    auto tracks = ml->searchMedia( "tra" ).tracks;
    ASSERT_EQ( 10u, tracks.size() );

    tracks = ml->searchMedia( "track 1" ).tracks;
    ASSERT_EQ( 2u, tracks.size() );

    tracks = ml->searchMedia( "grouik" ).tracks;
    ASSERT_EQ( 0u, tracks.size() );

    tracks = ml->searchMedia( "rack" ).tracks;
    ASSERT_EQ( 0u, tracks.size() );
}
273
274
275

TEST_F( Medias, Favorite )
{
276
    auto m = ml->addMedia( "media.mkv" );
277
278
279
280
281
282
283
284
285
286
    ASSERT_FALSE( m->isFavorite() );

    m->setFavorite( true );
    ASSERT_TRUE( m->isFavorite() );

    Reload();

    m = ml->media( m->id() );
    ASSERT_TRUE( m->isFavorite() );
}
287
288
289

TEST_F( Medias, History )
{
290
    auto m = std::static_pointer_cast<Media>( ml->addMedia( "media.mkv" ) );
291
292
293
294
295
296
297
298
299
300

    auto history = ml->lastMediaPlayed();
    ASSERT_EQ( 0u, history.size() );

    m->increasePlayCount();
    m->save();
    history = ml->lastMediaPlayed();
    ASSERT_EQ( 1u, history.size() );
    ASSERT_EQ( m->id(), history[0]->id() );

301
    compat::this_thread::sleep_for( std::chrono::seconds{ 1 } );
302
    auto m2 = std::static_pointer_cast<Media>( ml->addMedia( "media2.mkv" ) );
303
304
305
306
307
308
309
    m2->increasePlayCount();

    history = ml->lastMediaPlayed();
    ASSERT_EQ( 2u, history.size() );
    ASSERT_EQ( m2->id(), history[0]->id() );
    ASSERT_EQ( m->id(), history[1]->id() );
}
310

311
312
TEST_F( Medias, ClearHistory )
{
313
    auto m = std::static_pointer_cast<Media>( ml->addMedia( "media.mkv" ) );
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333

    auto history = ml->lastMediaPlayed();
    ASSERT_EQ( 0u, history.size() );

    m->increasePlayCount();
    m->save();
    history = ml->lastMediaPlayed();
    ASSERT_EQ( 1u, history.size() );

    ASSERT_TRUE( ml->clearHistory() );

    history = ml->lastMediaPlayed();
    ASSERT_EQ( 0u, history.size() );

    Reload();

    history = ml->lastMediaPlayed();
    ASSERT_EQ( 0u, history.size() );
}

334
335
TEST_F( Medias, SetReleaseDate )
{
336
    auto m = std::static_pointer_cast<Media>( ml->addMedia( "movie.mkv" ) );
337
338
339
340
341
342
343
344
345
346
347

    ASSERT_EQ( m->releaseDate(), 0u );
    m->setReleaseDate( 1234 );
    m->save();
    ASSERT_EQ( m->releaseDate(), 1234u );

    Reload();

    auto m2 = ml->media( m->id() );
    ASSERT_EQ( m2->releaseDate(), 1234u );
}
348
349
350

TEST_F( Medias, SortByAlpha )
{
351
    auto m1 = std::static_pointer_cast<Media>( ml->addMedia( "media1.mp3" ) );
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
352
    m1->setTitleBuffered( "Abcd" );
353
    m1->setType( Media::Type::Audio );
354
355
    m1->save();

356
    auto m2 = std::static_pointer_cast<Media>( ml->addMedia( "media2.mp3" ) );
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
357
    m2->setTitleBuffered( "Zyxw" );
358
    m2->setType( Media::Type::Audio );
359
360
    m2->save();

361
    auto m3 = std::static_pointer_cast<Media>( ml->addMedia( "media3.mp3" ) );
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
362
    m3->setTitleBuffered( "afterA-beforeZ" );
363
    m3->setType( Media::Type::Audio );
364
365
    m3->save();

366
    auto media = ml->audioFiles( SortingCriteria::Alpha, false );
367
    ASSERT_EQ( 3u, media.size() );
368
    ASSERT_EQ( m1->id(), media[0]->id() );
369
370
    ASSERT_EQ( m3->id(), media[1]->id() );
    ASSERT_EQ( m2->id(), media[2]->id() );
371

372
    media = ml->audioFiles( SortingCriteria::Alpha, true );
373
    ASSERT_EQ( 3u, media.size() );
374
    ASSERT_EQ( m2->id(), media[0]->id() );
375
376
    ASSERT_EQ( m3->id(), media[1]->id() );
    ASSERT_EQ( m1->id(), media[2]->id() );
377
378
379
380
381
382
}

TEST_F( Medias, SortByLastModifDate )
{
    auto file1 = std::make_shared<mock::NoopFile>( "media.mkv" );
    file1->setLastModificationDate( 666 );
383
    auto m1 = ml->addFile( file1 );
384
    m1->setType( Media::Type::Video );
385
    m1->save();
386
387
388

    auto file2 = std::make_shared<mock::NoopFile>( "media2.mkv" );
    file2->setLastModificationDate( 111 );
389
    auto m2 = ml->addFile( file2 );
390
    m2->setType( Media::Type::Video );
391
    m2->save();
392

393
    auto media = ml->videoFiles( SortingCriteria::LastModificationDate, false );
394
395
396
397
    ASSERT_EQ( 2u, media.size() );
    ASSERT_EQ( m2->id(), media[0]->id() );
    ASSERT_EQ( m1->id(), media[1]->id() );

398
    media = ml->videoFiles( SortingCriteria::LastModificationDate, true );
399
400
401
402
    ASSERT_EQ( 2u, media.size() );
    ASSERT_EQ( m2->id(), media[1]->id() );
    ASSERT_EQ( m1->id(), media[0]->id() );
}
403

404
405
406
407
TEST_F( Medias, SortByFileSize )
{
    auto file1 = std::make_shared<mock::NoopFile>( "media.mkv" );
    file1->setSize( 666 );
408
    auto m1 = ml->addFile( file1 );
409
    m1->setType( Media::Type::Video );
410
    m1->save();
411
412
413

    auto file2 = std::make_shared<mock::NoopFile>( "media2.mkv" );
    file2->setSize( 111 );
414
    auto m2 = ml->addFile( file2 );
415
    m2->setType( Media::Type::Video );
416
    m2->save();
417
418
419
420
421
422
423
424
425
426
427
428

    auto media = ml->videoFiles( SortingCriteria::FileSize, false );
    ASSERT_EQ( 2u, media.size() );
    ASSERT_EQ( m2->id(), media[0]->id() );
    ASSERT_EQ( m1->id(), media[1]->id() );

    media = ml->videoFiles( SortingCriteria::FileSize, true );
    ASSERT_EQ( 2u, media.size() );
    ASSERT_EQ( m2->id(), media[1]->id() );
    ASSERT_EQ( m1->id(), media[0]->id() );
}

429
430
TEST_F( Medias, SetType )
{
431
    auto m1 = std::static_pointer_cast<Media>( ml->addMedia( "media1.mp3" ) );
432
    ASSERT_EQ( IMedia::Type::External, m1->type() );
433

434
    m1->setType( IMedia::Type::Video );
435
436
    m1->save();

437
    ASSERT_EQ( IMedia::Type::Video, m1->type() );
438
439
440
441

    Reload();

    auto m2 = ml->media( m1->id() );
442
    ASSERT_EQ( IMedia::Type::Video, m2->type() );
443
}
444

445
446
TEST_F( Medias, Metadata )
{
447
    auto m = ml->addMedia( "media.mp3" );
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470

    {
        const auto& md = m->metadata( Media::MetadataType::Speed );
        ASSERT_FALSE( md.isSet() );
    }

    auto res = m->setMetadata( Media::MetadataType::Speed, "foo" );
    ASSERT_TRUE( res );

    {
        const auto& md = m->metadata( Media::MetadataType::Speed );
        ASSERT_EQ( "foo", md.str() );
    }

    Reload();

    m = ml->media( m->id() );
    const auto& md = m->metadata( Media::MetadataType::Speed );
    ASSERT_EQ( "foo", md.str() );
}

TEST_F( Medias, MetadataOverride )
{
471
    auto m = ml->addMedia( "media.mp3" );
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
    auto res = m->setMetadata( Media::MetadataType::Speed, "foo" );
    ASSERT_TRUE( res );

    m->setMetadata( Media::MetadataType::Speed, "otter" );
    {
        const auto& md = m->metadata( Media::MetadataType::Speed );
        ASSERT_EQ( "otter", md.str() );
    }

    Reload();

    m = ml->media( m->id() );
    const auto& md = m->metadata( Media::MetadataType::Speed );
    ASSERT_EQ( "otter", md.str() );
}

488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
TEST_F( Medias, ExternalMrl )
{
    auto m = ml->addMedia( "https://foo.bar/sea-otters.mkv" );
    ASSERT_NE( nullptr, m );

    ASSERT_EQ( m->title(), "sea-otters.mkv" );

    // External files shouldn't appear in listings
    auto videos = ml->videoFiles( medialibrary::SortingCriteria::Default, false );
    ASSERT_EQ( 0u, videos.size() );

    auto audios = ml->audioFiles( medialibrary::SortingCriteria::Default, false );
    ASSERT_EQ( 0u, audios.size() );

    Reload();

    auto m2 = ml->media( "https://foo.bar/sea-otters.mkv" );
    ASSERT_NE( nullptr, m2 );
    ASSERT_EQ( m->id(), m2->id() );
507
508
509
510
511
512

    auto files = m2->files();
    ASSERT_EQ( 1u, files.size() );
    auto f = files[0];
    ASSERT_TRUE( f->isExternal() );
    ASSERT_EQ( File::Type::Main, f->type() );
513
514
}

515
516
517
518
519
520
521
522
TEST_F( Medias, DuplicatedExternalMrl )
{
    auto m = ml->addMedia( "http://foo.bar" );
    auto m2 = ml->addMedia( "http://foo.bar" );
    ASSERT_NE( nullptr, m );
    ASSERT_EQ( nullptr, m2 );
}

523
524
525
526
527
528
529
530
531
532
533
534
535
536
TEST_F( Medias, SetTitle )
{
    auto m = ml->addMedia( "media" );
    ASSERT_EQ( "media", m->title() );
    auto res = m->setTitle( "sea otters" );
    ASSERT_TRUE( res );
    ASSERT_EQ( "sea otters", m->title() );

    Reload();

    m = ml->media( m->id() );
    ASSERT_EQ( "sea otters", m->title() );
}

537
538
539
540
541
542
543
544
545
546
547
548
549
class FetchMedia : public Tests
{
protected:
    static const std::string RemovableDeviceUuid;
    static const std::string RemovableDeviceMountpoint;
    std::shared_ptr<mock::FileSystemFactory> fsMock;
    std::unique_ptr<mock::WaitForDiscoveryComplete> cbMock;

    virtual void SetUp() override
    {
        unlink( "test.db" );
        fsMock.reset( new mock::FileSystemFactory );
        cbMock.reset( new mock::WaitForDiscoveryComplete );
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
550
        fsMock->addFolder( "file:///a/mnt/" );
551
552
553
554
555
556
557
558
        auto device = fsMock->addDevice( RemovableDeviceMountpoint, RemovableDeviceUuid );
        device->setRemovable( true );
        fsMock->addFile( RemovableDeviceMountpoint + "removablefile.mp3" );
        Reload();
    }

    virtual void InstantiateMediaLibrary() override
    {
559
        ml.reset( new MediaLibraryTester );
560
561
562
563
564
565
566
567
568
    }

    virtual void Reload()
    {
        Tests::Reload( fsMock, cbMock.get() );
    }
};

const std::string FetchMedia::RemovableDeviceUuid = "{fake-removable-device}";
Hugo Beauzée-Luyssen's avatar
Hugo Beauzée-Luyssen committed
569
const std::string FetchMedia::RemovableDeviceMountpoint = "file:///a/mnt/fake-device/";
570
571
572
573

TEST_F( FetchMedia, FetchNonRemovable )
{
    ml->discover( mock::FileSystemFactory::Root );
574
    bool discovered = cbMock->waitDiscovery();
575
576
577
578
579
580
581
582
583
    ASSERT_TRUE( discovered );

    auto m = ml->media( mock::FileSystemFactory::SubFolder + "subfile.mp4" );
    ASSERT_NE( nullptr, m );
}

TEST_F( FetchMedia, FetchRemovable )
{
    ml->discover( mock::FileSystemFactory::Root );
584
    bool discovered = cbMock->waitDiscovery();
585
586
587
588
589
590
591
592
593
    ASSERT_TRUE( discovered );

    auto m = ml->media( RemovableDeviceMountpoint + "removablefile.mp3" );
    ASSERT_NE( nullptr, m );
}

TEST_F( FetchMedia, FetchRemovableUnplugged )
{
    ml->discover( mock::FileSystemFactory::Root );
594
    bool discovered = cbMock->waitDiscovery();
595
596
597
598
599
    ASSERT_TRUE( discovered );

    fsMock->unmountDevice( RemovableDeviceUuid );

    Reload();
600
    bool reloaded = cbMock->waitReload();
601
602
603
604
605
    ASSERT_TRUE( reloaded );

    auto m = ml->media( RemovableDeviceMountpoint + "removablefile.mp3" );
    ASSERT_EQ( nullptr, m );
}