package ytid

import (
	"testing"
)

func TestGetID(t *testing.T) {
	inputs := []string{
		"https://www.youtube.com/watch?v=m4U232MuTG4",
		"https://youtu.be/dVtehSbLO8M",
		"https://www.youtube.com/embed/Kt-tLuszKBA",
	}

	for n, i := range inputs {
		switch n {
		case 0:
			if GetID(i) != "m4U232MuTG4" {
				t.Fatal("failed", n)
			}
		case 1:
			if GetID(i) != "dVtehSbLO8M" {
				t.Fatal("failed", n)
			}
		case 2:
			if GetID(i) != "Kt-tLuszKBA" {
				t.Fatal("failed", n)
			}

		}
	}
}