/* * Copyright (C) 2014 Red Hat Inc. * * Author: Bastien Nocera * * This library 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; version 2.1 of * the License, or (at your option) any later version. * * This library 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 library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA * 02110-1301 USA * */ #include #define LUA_FACTORY_ID "grl-lua-factory" #define VIDEO_TITLE_PARSING_ID "grl-video-title-parsing" static void test_setup (void) { GError *error = NULL; GrlRegistry *registry; registry = grl_registry_get_default (); grl_registry_load_all_plugins (registry, TRUE, &error); g_assert_no_error (error); } static char * get_show_for_title (GrlSource *source, const char *title, char **new_title, int *season, int *episode) { GrlMedia *media; GrlOperationOptions *options; GList *keys; char *show; const gchar *str; media = grl_media_video_new (); grl_media_set_title (media, title); grl_data_set_boolean (GRL_DATA (media), GRL_METADATA_KEY_TITLE_FROM_FILENAME, TRUE); keys = grl_metadata_key_list_new (GRL_METADATA_KEY_SHOW, GRL_METADATA_KEY_SEASON, GRL_METADATA_KEY_EPISODE, GRL_METADATA_KEY_EPISODE_TITLE, NULL); options = grl_operation_options_new (NULL); grl_operation_options_set_resolution_flags (options, GRL_RESOLVE_NORMAL); grl_source_resolve_sync (source, media, keys, options, NULL); g_list_free (keys); g_object_unref (options); *season = grl_data_get_int (GRL_DATA (media), GRL_METADATA_KEY_SEASON); *episode = grl_data_get_int (GRL_DATA (media), GRL_METADATA_KEY_EPISODE); show = g_strdup (grl_data_get_string (GRL_DATA (media), GRL_METADATA_KEY_SHOW)); str = grl_media_get_episode_title (media); *new_title = (str && str[0] == '\0') ? NULL : g_strdup (str); g_object_unref (media); return show; } static void test_episodes (void) { GrlRegistry *registry; GrlSource *source; guint i; struct { char *title; char *show; char *episode_title; int season; int episode; } episode_tests[] = { { "The.Slap.S01E01.Hector.WS.PDTV.XviD-BWB.avi", "The Slap", "Hector", 1, 1 }, { "metalocalypse.s02e01.dvdrip.xvid-ffndvd.avi", "metalocalypse", NULL, 2, 1 }, { "Boardwalk.Empire.S04E01.HDTV.x264-2HD.mp4", "Boardwalk Empire", NULL, 4, 1 }, { "My super series.S01E01.mp4", "My super series", NULL, 1, 1 }, { "Adventure Time - 2x01 - It Came from the Nightosphere.mp4", "Adventure Time", "It Came from the Nightosphere", 2, 1 }, { "Real Humans S01 EP01 [X264] [HD 720p] [FR] [SWE] [SRT FR] [MRPHU].mkv", "Real Humans", NULL, 1, 1 }, /* Episode and Series separated by '.' and Title inside parenthesis */ { "Felicity 2.05 (Crash).avi", "Felicity", "Crash", 2, 5 }, { "Felicity 4.08 (Last Thanksgiving).avi", "Felicity", "Last Thanksgiving", 4, 8 }, /* These below should not be detected as an episode of a series. */ { "My.Neighbor.Totoro.1988.1080p.BluRay.X264.mkv", NULL, NULL, 0, 0 }, { "140127Mata-16x9 (bug 723166).mp4", "140127Mata", "bug 723166", 16, 9 } }; registry = grl_registry_get_default (); source = grl_registry_lookup_source (registry, VIDEO_TITLE_PARSING_ID); g_assert (source); for (i = 0; i < G_N_ELEMENTS(episode_tests); i++) { char *show, *new_title; int season, episode; show = get_show_for_title (source, episode_tests[i].title, &new_title, &season, &episode); g_assert_cmpstr (episode_tests[i].show, ==, show); if (show != NULL) { g_assert_cmpstr (episode_tests[i].episode_title, ==, new_title); g_assert_cmpint (episode_tests[i].season, ==, season); g_assert_cmpint (episode_tests[i].episode, ==, episode); } g_free (show); g_clear_pointer (&new_title, g_free); } } static void test_title_override (void) { GrlRegistry *registry; GrlSource *source; guint i; struct { char *title; gboolean from_filename; char *expected; } filename_tests[] = { { "Test.mp4", TRUE, "Test" }, { "Boardwalk.Empire.S04E01.HDTV.x264-2HD.mp4", FALSE, "Boardwalk.Empire.S04E01.HDTV.x264-2HD.mp4" }, { "wsb.wmv", TRUE, "wsb" }, { "[XVID-ITA] Lo Svarione Degli anelli - I DUE PORRI - Clistere.Org - rip by Shella.avi", FALSE, "[XVID-ITA] Lo Svarione Degli anelli - I DUE PORRI - Clistere.Org - rip by Shella" }, }; registry = grl_registry_get_default (); source = grl_registry_lookup_source (registry, VIDEO_TITLE_PARSING_ID); g_assert (source); for (i = 0; i < G_N_ELEMENTS(filename_tests); i++) { GrlMedia *media; GrlOperationOptions *options; GList *keys; const gchar *title; media = grl_media_video_new (); grl_media_set_title (media, filename_tests[i].title); grl_data_set_boolean (GRL_DATA (media), GRL_METADATA_KEY_TITLE_FROM_FILENAME, filename_tests[i].from_filename); keys = grl_metadata_key_list_new (GRL_METADATA_KEY_TITLE, GRL_METADATA_KEY_SHOW, NULL); options = grl_operation_options_new (NULL); grl_operation_options_set_resolution_flags (options, GRL_RESOLVE_FULL); grl_source_resolve_sync (source, media, keys, options, NULL); g_list_free (keys); g_object_unref (options); title = grl_media_get_title(media); g_assert_cmpstr (filename_tests[i].expected, ==, title); g_object_unref (media); } } int main(int argc, char **argv) { g_setenv ("GRL_PLUGIN_PATH", LUA_FACTORY_PLUGIN_PATH, TRUE); g_setenv ("GRL_PLUGIN_LIST", LUA_FACTORY_ID, TRUE); g_setenv ("GRL_LUA_SOURCES_PATH", LUA_SOURCES_PATH, TRUE); grl_init (&argc, &argv); g_test_init (&argc, &argv, NULL); test_setup (); g_test_add_func ("/lua-factory/video-title-parsing/resolve/episodes", test_episodes); g_test_add_func ("/lua-factory/video-title-parsing/resolve/title-override", test_title_override); gint result = g_test_run (); grl_deinit (); return result; }