spider/internal/telegram/sessionimport/dcaddr_test.go

package sessionimport

import "testing"

func TestTGDCAddr(t *testing.T) {
	cases := []struct {
		dc   int
		want string
	}{
		{1, "149.154.175.53:443"},
		{2, "149.154.167.51:443"},
		{3, "149.154.175.100:443"},
		{4, "149.154.167.91:443"},
		{5, "91.108.56.130:443"},
	}
	for _, tc := range cases {
		got, err := TGDCAddr(tc.dc)
		if err != nil {
			t.Fatalf("dc=%d unexpected err: %v", tc.dc, err)
		}
		if got != tc.want {
			t.Errorf("dc=%d got %q want %q", tc.dc, got, tc.want)
		}
	}
	if _, err := TGDCAddr(99); err == nil {
		t.Error("expected err for unknown dc=99")
	}
}