Initial open source release

internal/rpc/upload_test.go

@@ -0,0 +1,34 @@
+package rpc
+
+import (
+	"testing"
+
+	"github.com/gotd/td/tg"
+)
+
+func TestStorageFileTypePrefersMagicOverMime(t *testing.T) {
+	webp := []byte{'R', 'I', 'F', 'F', 0, 0, 0, 0, 'W', 'E', 'B', 'P'}
+	if _, ok := storageFileType("image/jpeg", webp).(*tg.StorageFileWebp); !ok {
+		t.Fatalf("webp bytes mislabeled as jpeg should return StorageFileWebp")
+	}
+}
+
+func TestStorageFileTypeFallsBackToMime(t *testing.T) {
+	if _, ok := storageFileType("image/png", nil).(*tg.StorageFilePng); !ok {
+		t.Fatalf("png mime without bytes should return StorageFilePng")
+	}
+}
+
+func TestFileLocationKeyUsesDocumentID(t *testing.T) {
+	key, ok := fileLocationKey(&tg.InputDocumentFileLocation{
+		ID:        1382305375846410902,
+		ThumbSize: "m",
+	})
+	if !ok {
+		t.Fatal("fileLocationKey returned !ok")
+	}
+	const want = "doc:1382305375846410902:m"
+	if key != want {
+		t.Fatalf("key = %q, want %q", key, want)
+	}
+}