xbmc/video/test/TestVideoFileItemClassify.cpp

   1 /*
   2  *  Copyright (C) 2005-2018 Team Kodi
   3  *  This file is part of Kodi - https://kodi.tv
   4  *
   5  *  SPDX-License-Identifier: GPL-2.0-or-later
   6  *  See LICENSES/README.md for more information.
   7  */
   8
   9 #include "FileItem.h"
  10 #include "ServiceBroker.h"
  11 #include "filesystem/Directory.h"
  12 #include "filesystem/File.h"
  13 #include "games/tags/GameInfoTag.h"
  14 #include "music/tags/MusicInfoTag.h"
  15 #include "pictures/PictureInfoTag.h"
  16 #include "test/TestUtils.h"
  17 #include "utils/FileExtensionProvider.h"
  18 #include "utils/FileUtils.h"
  19 #include "utils/URIUtils.h"
  20 #include "video/VideoFileItemClassify.h"
  21 #include "video/VideoInfoTag.h"
  22
  23 #include <array>
  24
  25 #include <gtest/gtest.h>
  26
  27 using namespace KODI;
  28
  29 struct StubDefinition
  30 {
  31   StubDefinition(const std::string& path,
  32                  bool res = true,
  33                  const std::string& tagPath = "",
  34                  bool isFolder = false)
  35     : item(path, isFolder), result(res)
  36   {
  37     if (!tagPath.empty())
  38     {
  39       if (isFolder)
  40         item.GetVideoInfoTag()->m_strPath = tagPath;
  41       else
  42         item.GetVideoInfoTag()->m_strFileNameAndPath = tagPath;
  43     }
  44   }
  45
  46   CFileItem item;
  47   bool result;
  48 };
  49
  50 class DiscStubTest : public testing::WithParamInterface<StubDefinition>, public testing::Test
  51 {
  52 };
  53
  54 TEST_P(DiscStubTest, IsDiscStub)
  55 {
  56   EXPECT_EQ(VIDEO::IsDiscStub(GetParam().item), GetParam().result);
  57 }
  58
  59 const auto discstub_tests = std::array{
  60     StubDefinition{"/home/user/test.disc"},
  61     StubDefinition{"videodb://foo/bar", true, "/home/user/test.disc"},
  62     StubDefinition{"videodb://foo/bar", false, "/home/user/test.disc", true},
  63     StubDefinition{"/home/user/test.avi", false},
  64 };
  65
  66 INSTANTIATE_TEST_SUITE_P(TestVideoFileItemClassify,
  67                          DiscStubTest,
  68                          testing::ValuesIn(discstub_tests));
  69
  70 struct VideoClassifyTest
  71 {
  72   VideoClassifyTest(const std::string& path,
  73                     bool res = true,
  74                     const std::string& mime = "",
  75                     int tag_type = 0)
  76     : item(path, false), result(res)
  77   {
  78     if (!mime.empty())
  79       item.SetMimeType(mime);
  80     switch (tag_type)
  81     {
  82       case 1:
  83         item.GetVideoInfoTag()->m_strFileNameAndPath = path;
  84         break;
  85       case 2:
  86         item.GetGameInfoTag()->SetGameClient("some_client");
  87         break;
  88       case 3:
  89         item.GetMusicInfoTag()->SetPlayCount(1);
  90         break;
  91       case 4:
  92         item.GetPictureInfoTag()->SetInfo("foo", "bar");
  93         ;
  94         break;
  95       default:
  96         break;
  97     }
  98   }
  99
 100   CFileItem item;
 101   bool result;
 102 };
 103
 104 class VideoTest : public testing::WithParamInterface<VideoClassifyTest>, public testing::Test
 105 {
 106 };
 107
 108 TEST_P(VideoTest, IsVideo)
 109 {
 110   EXPECT_EQ(VIDEO::IsVideo(GetParam().item), GetParam().result);
 111 }
 112
 113 const auto video_tests = std::array{
 114     VideoClassifyTest{"/home/user/video.avi", true, "video/avi"},
 115     VideoClassifyTest{"/home/user/video.avi", true, "", 1},
 116     VideoClassifyTest{"/home/user/video.gam", false, "", 2},
 117     VideoClassifyTest{"/home/user/video.mus", false, "", 3},
 118     VideoClassifyTest{"/home/user/video.pic", false, "", 4},
 119     VideoClassifyTest{"pvr://recordings/tv/1", true},
 120     VideoClassifyTest{"pvr://123", false},
 121     VideoClassifyTest{"dvd://VIDEO_TS/video_ts.ifo", true},
 122     VideoClassifyTest{"dvd://1", true},
 123     VideoClassifyTest{"/home/user/video.not", true, "application/ogg"},
 124     VideoClassifyTest{"/home/user/video.not", true, "application/mp4"},
 125     VideoClassifyTest{"/home/user/video.not", true, "application/mxf"},
 126 };
 127
 128 INSTANTIATE_TEST_SUITE_P(TestVideoFileItemClassify, VideoTest, testing::ValuesIn(video_tests));
 129
 130 TEST(TestVideoFileItemClassify, VideoExtensions)
 131 {
 132   const auto& exts = CServiceBroker::GetFileExtensionProvider().GetVideoExtensions();
 133   for (const auto& ext : StringUtils::Split(exts, "|"))
 134   {
 135     if (!ext.empty())
 136     {
 137       EXPECT_TRUE(VIDEO::IsVideo(CFileItem(ext, false)));
 138     }
 139   }
 140 }
 141
 142 TEST(TestVideoFileItemClassify, IsBDFile)
 143 {
 144   EXPECT_TRUE(VIDEO::IsBDFile(CFileItem("/home/foo/index.BDMV", false)));
 145   EXPECT_TRUE(VIDEO::IsBDFile(CFileItem("smb://foo/bar/index.bdm", false)));
 146   EXPECT_TRUE(VIDEO::IsBDFile(CFileItem("ftp://foo:bar@foobar.com/movieobject.BDMV", false)));
 147   EXPECT_TRUE(VIDEO::IsBDFile(CFileItem("https://foobar.com/movieobj.bdm", false)));
 148   EXPECT_FALSE(VIDEO::IsBDFile(CFileItem("https://foobar.com/movieobject.not", false)));
 149 }
 150
 151 TEST(TestVideoFileItemClassify, IsDVDFile)
 152 {
 153   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.vob", false), true, false));
 154   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.VOB", false), true, true));
 155   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_TS.vob", false), false, false));
 156   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.VOb", false), false, true));
 157
 158   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/vts_yo_0.vob", false), true, false));
 159   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/vts_0_ifo.vob", false), true, true));
 160   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/VTS_123456_0.vob", false), false, false));
 161   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/VTS_qwerty_0.VOB", false), false, true));
 162
 163   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.IFO", false), false, true));
 164   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.IFO", false), true, true));
 165   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.IFO", false), false, false));
 166   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/video_ts.IFO", false), true, false));
 167
 168   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/vts_ab_0.ifo", false), false, true));
 169   EXPECT_TRUE(VIDEO::IsDVDFile(CFileItem("/home/foo/vts_ab_0.ifo", false), false, true));
 170   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/VTS_ab_0.ifo", false), false, false));
 171   EXPECT_FALSE(VIDEO::IsDVDFile(CFileItem("/home/foo/VTS_ab_0.ifo", false), false, false));
 172 }
 173
 174 TEST(TestVideoFileItemClassify, IsProtectedBlurayDisc)
 175 {
 176   const auto temp_file = CXBMCTestUtils::Instance().CreateTempFile("bluraytest");
 177   const std::string dir = CXBMCTestUtils::Instance().TempFileDirectory(temp_file);
 178   CFileUtils::DeleteItem(URIUtils::AddFileToFolder(dir, "AACS", "Unit_Key_RO.inf"));
 179   EXPECT_FALSE(VIDEO::IsProtectedBlurayDisc(CFileItem(dir, true)));
 180   XFILE::CDirectory::Create(URIUtils::AddFileToFolder(dir, "AACS"));
 181   XFILE::CFile inf_file;
 182   inf_file.OpenForWrite(URIUtils::AddFileToFolder(dir, "AACS", "Unit_Key_RO.inf"));
 183   inf_file.Close();
 184   EXPECT_TRUE(VIDEO::IsProtectedBlurayDisc(CFileItem(dir, true)));
 185   CFileUtils::DeleteItem(URIUtils::AddFileToFolder(dir, "AACS", "Unit_Key_RO.inf"));
 186   CXBMCTestUtils::Instance().DeleteTempFile(temp_file);
 187 }
 188
 189 TEST(TestVideoFileItemClassify, IsSubtitle)
 190 {
 191   const auto& exts = CServiceBroker::GetFileExtensionProvider().GetSubtitleExtensions();
 192   for (const auto& ext : StringUtils::Split(exts, "|"))
 193   {
 194     if (!ext.empty())
 195     {
 196       EXPECT_TRUE(VIDEO::IsSubtitle(CFileItem("random" + ext, false)));
 197     }
 198   }
 199
 200   EXPECT_FALSE(VIDEO::IsSubtitle(CFileItem("random.notasub", false)));
 201 }
 202
 203 TEST(TestVideoFileItemClassify, IsVideoAssetsFile)
 204 {
 205   EXPECT_TRUE(VIDEO::IsVideoAssetFile(CFileItem("videodb://foo/bar?videoversionid=1", false)));
 206   EXPECT_FALSE(VIDEO::IsVideoAssetFile(CFileItem("videodb://foo/bar?videoversionid=1", true)));
 207   EXPECT_FALSE(VIDEO::IsVideoAssetFile(CFileItem("videodb://foo/bar", false)));
 208 }
 209
 210 TEST(TestVideoFileItemClassify, IsVideoDb)
 211 {
 212   EXPECT_TRUE(VIDEO::IsVideoDb(CFileItem("videodb://1/2/3", false)));
 213   EXPECT_TRUE(VIDEO::IsVideoDb(CFileItem("videodb://1/2/", true)));
 214   EXPECT_FALSE(VIDEO::IsVideoDb(CFileItem("/videodb/home/foo/Extraordinary/", true)));
 215 }
 216
 217 TEST(TestVideoFileItemClassify, IsVideoExtrasFolder)
 218 {
 219   EXPECT_TRUE(VIDEO::IsVideoExtrasFolder(CFileItem("/home/foo/Extras/", true)));
 220   EXPECT_TRUE(VIDEO::IsVideoExtrasFolder(CFileItem("/home/foo/extras/", true)));
 221   EXPECT_FALSE(VIDEO::IsVideoExtrasFolder(CFileItem("/home/foo/Extraordinary/", true)));
 222   EXPECT_FALSE(VIDEO::IsVideoExtrasFolder(CFileItem("/home/foo/Extras/abc.mkv", false)));
 223 }