local helpers = require('helpers')

local function assert_equals(expected, actual)
    if expected ~= actual then
        error(string.format("TEST FAILED: Expected '%s', got '%s'", expected, actual))
    end
end

local function test_get_episode_number()
    local test_cases = {
      { nil, "A Whisker Away.mkv" },
      { nil, "[Placeholder] Gekijouban SHIROBAKO [Ma10p_1080p][x265_flac]" },
      { "06", "[Placeholder] Sono Bisque Doll wa Koi wo Suru - 06 [54E495D0]" },
      { "02", "(Hi10)_Kobayashi-san_Chi_no_Maid_Dragon_-_02_(BD_1080p)_(Placeholder)_(12C5D2B4)" },
      { "01", "[Placeholder] Koi to Yobu ni wa Kimochi Warui - 01 (1080p) [D517C9F0]" },
      { "01", "[Placeholder] Tsukimonogatari 01 [BD 1080p x264 10-bit FLAC] [5CD88145]" },
      { "01", "[Placeholder] 86 - Eighty Six - 01 (1080p) [1B13598F]" },
      { "00", "[Placeholder] Fate Stay Night - Unlimited Blade Works - 00 (BD 1080p Hi10 FLAC) [95590B7F]" },
      { "01", "House, M.D. S01E01 Pilot - Everybody Lies (1080p x265 Placeholder)" },
      { "165", "A Generic Episode-165" }
    }

    for _, case in pairs(test_cases) do
      local _, _, episode_num = helpers.get_episode_number(case[2])
      assert_equals(case[1], episode_num)
    end
end

-- Runs tests
test_get_episode_number()

os.exit(print("Tests passed"))