github.com/benoitkugler/goacve@v0.0.0-20201217100549-151ce6e55dc8/server/documents/compress_test.go (about) 1 package documents 2 3 import ( 4 "fmt" 5 "io/ioutil" 6 "os" 7 "testing" 8 ) 9 10 func TestPdf(t *testing.T) { 11 fileContent, err := ioutil.ReadFile("../../ressources/pdf_multi_pages.pdf") 12 if err != nil { 13 t.Fatal(err) 14 } 15 16 nb, err := countPdfPages(fileContent) 17 if err != nil { 18 t.Fatal(err) 19 } 20 if nb != 3 { 21 t.Fatalf("expected 3 pages, got %d", nb) 22 } 23 24 fileContent, err = ioutil.ReadFile("../../ressources/RIB.pdf") 25 if err != nil { 26 t.Fatal(err) 27 } 28 29 nb, err = countPdfPages(fileContent) 30 if err != nil { 31 t.Fatal(err) 32 } 33 if nb != 1 { 34 t.Fatalf("expected 1 page, got %d", nb) 35 } 36 37 asJPEG, err := CompressPdf(fileContent) 38 if err != nil { 39 t.Fatal(err) 40 } 41 fmt.Println(len(fileContent), len(asJPEG)) 42 43 fileContent, err = ioutil.ReadFile("../../ressources/pdf_heavy.pdf") 44 if err != nil { 45 t.Fatal(err) 46 } 47 48 nb, err = countPdfPages(fileContent) 49 if err != nil { 50 t.Fatal(err) 51 } 52 if nb != 1 { 53 t.Fatalf("expected 1 page, got %d", nb) 54 } 55 56 asJPEG, err = CompressPdf(fileContent) 57 if err != nil { 58 t.Fatal(err) 59 } 60 fmt.Println(len(fileContent), len(asJPEG)) 61 if err := ioutil.WriteFile("../../ressources/pdf_heavy.jpeg", asJPEG, os.ModePerm); err != nil { 62 t.Fatal(err) 63 } 64 }