From 9288b6ac97948df2877cc0a4316f83abb215a63c Mon Sep 17 00:00:00 2001 From: Evgeny Kuznetsov Date: Mon, 27 Jan 2020 21:17:01 +0300 Subject: add TestCleanText --- main_test.go | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 main_test.go (limited to 'main_test.go') diff --git a/main_test.go b/main_test.go new file mode 100644 index 0000000..b572fd2 --- /dev/null +++ b/main_test.go @@ -0,0 +1,52 @@ +// Copyright (C) 2020 Evgeny Kuznetsov (evgeny@kuznetsov.md) +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along tihe this program. If not, see . + +package main + +import ( + "bytes" + "flag" + "io/ioutil" + "path/filepath" + "testing" +) + +var update = flag.Bool("update", false, "update .golden files") + +func helperLoadBytes(t *testing.T, name string) []byte { + t.Helper() + path := filepath.Join("testdata", name) + bytes, err := ioutil.ReadFile(path) + if err != nil { + t.Fatal(err) + } + return bytes +} + +func TestCleanText(t *testing.T) { + about := helperLoadBytes(t, "about") + actual := cleanText(about) + golden := filepath.Join("testdata", t.Name()+".golden") + if *update { + if err := ioutil.WriteFile(golden, actual, 0644); err != nil { + t.Fatal(err) + } + } + expected, _ := ioutil.ReadFile(golden) + + if !bytes.Equal(actual, expected) { + t.Fail() + } +} -- cgit v1.2.3