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  }