123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252 |
- // Copyright 2020 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package base
-
- import (
- "testing"
-
- "github.com/stretchr/testify/assert"
- )
-
- func TestEncodeMD5(t *testing.T) {
- assert.Equal(t,
- "3858f62230ac3c915f300c664312c63f",
- EncodeMD5("foobar"),
- )
- }
-
- func TestEncodeSha1(t *testing.T) {
- assert.Equal(t,
- "8843d7f92416211de9ebb963ff4ce28125932878",
- EncodeSha1("foobar"),
- )
- }
-
- func TestEncodeSha256(t *testing.T) {
- assert.Equal(t,
- "c3ab8ff13720e8ad9047dd39466b3c8974e592c2fa383d4a3960714caef0c4f2",
- EncodeSha256("foobar"),
- )
- }
-
- func TestShortSha(t *testing.T) {
- assert.Equal(t, "veryverylo", ShortSha("veryverylong"))
- }
-
- func TestBasicAuthDecode(t *testing.T) {
- _, _, err := BasicAuthDecode("?")
- assert.Equal(t, "illegal base64 data at input byte 0", err.Error())
-
- user, pass, err := BasicAuthDecode("Zm9vOmJhcg==")
- assert.NoError(t, err)
- assert.Equal(t, "foo", user)
- assert.Equal(t, "bar", pass)
-
- _, _, err = BasicAuthDecode("aW52YWxpZA==")
- assert.Error(t, err)
-
- _, _, err = BasicAuthDecode("invalid")
- assert.Error(t, err)
- }
-
- func TestBasicAuthEncode(t *testing.T) {
- assert.Equal(t, "Zm9vOmJhcg==", BasicAuthEncode("foo", "bar"))
- }
-
- // TODO: Test PBKDF2()
- // TODO: Test VerifyTimeLimitCode()
- // TODO: Test CreateTimeLimitCode()
-
- func TestFileSize(t *testing.T) {
- var size int64 = 512
- assert.Equal(t, "512 B", FileSize(size))
- size *= 1024
- assert.Equal(t, "512 KiB", FileSize(size))
- size *= 1024
- assert.Equal(t, "512 MiB", FileSize(size))
- size *= 1024
- assert.Equal(t, "512 GiB", FileSize(size))
- size *= 1024
- assert.Equal(t, "512 TiB", FileSize(size))
- size *= 1024
- assert.Equal(t, "512 PiB", FileSize(size))
- size *= 4
- assert.Equal(t, "2.0 EiB", FileSize(size))
- }
-
- func TestSubtract(t *testing.T) {
- toFloat64 := func(n interface{}) float64 {
- switch v := n.(type) {
- case int:
- return float64(v)
- case int8:
- return float64(v)
- case int16:
- return float64(v)
- case int32:
- return float64(v)
- case int64:
- return float64(v)
- case float32:
- return float64(v)
- case float64:
- return v
- default:
- return 0.0
- }
- }
- values := []interface{}{
- int(-3),
- int8(14),
- int16(81),
- int32(-156),
- int64(1528),
- float32(3.5),
- float64(-15.348),
- }
- for _, left := range values {
- for _, right := range values {
- expected := toFloat64(left) - toFloat64(right)
- sub := Subtract(left, right)
- assert.InDelta(t, expected, sub, 1e-3)
- }
- }
- }
-
- func TestEllipsisString(t *testing.T) {
- assert.Equal(t, "...", EllipsisString("foobar", 0))
- assert.Equal(t, "...", EllipsisString("foobar", 1))
- assert.Equal(t, "...", EllipsisString("foobar", 2))
- assert.Equal(t, "...", EllipsisString("foobar", 3))
- assert.Equal(t, "f...", EllipsisString("foobar", 4))
- assert.Equal(t, "fo...", EllipsisString("foobar", 5))
- assert.Equal(t, "foobar", EllipsisString("foobar", 6))
- assert.Equal(t, "foobar", EllipsisString("foobar", 10))
- }
-
- func TestTruncateString(t *testing.T) {
- assert.Equal(t, "", TruncateString("foobar", 0))
- assert.Equal(t, "f", TruncateString("foobar", 1))
- assert.Equal(t, "fo", TruncateString("foobar", 2))
- assert.Equal(t, "foo", TruncateString("foobar", 3))
- assert.Equal(t, "foob", TruncateString("foobar", 4))
- assert.Equal(t, "fooba", TruncateString("foobar", 5))
- assert.Equal(t, "foobar", TruncateString("foobar", 6))
- assert.Equal(t, "foobar", TruncateString("foobar", 7))
- }
-
- func TestStringsToInt64s(t *testing.T) {
- testSuccess := func(input []string, expected []int64) {
- result, err := StringsToInt64s(input)
- assert.NoError(t, err)
- assert.Equal(t, expected, result)
- }
- testSuccess([]string{}, []int64{})
- testSuccess([]string{"-1234"}, []int64{-1234})
- testSuccess([]string{"1", "4", "16", "64", "256"},
- []int64{1, 4, 16, 64, 256})
-
- _, err := StringsToInt64s([]string{"-1", "a", "$"})
- assert.Error(t, err)
- }
-
- func TestInt64sToStrings(t *testing.T) {
- assert.Equal(t, []string{}, Int64sToStrings([]int64{}))
- assert.Equal(t,
- []string{"1", "4", "16", "64", "256"},
- Int64sToStrings([]int64{1, 4, 16, 64, 256}),
- )
- }
-
- func TestInt64sToMap(t *testing.T) {
- assert.Equal(t, map[int64]bool{}, Int64sToMap([]int64{}))
- assert.Equal(t,
- map[int64]bool{1: true, 4: true, 16: true},
- Int64sToMap([]int64{1, 4, 16}),
- )
- }
-
- func TestIsLetter(t *testing.T) {
- assert.True(t, IsLetter('a'))
- assert.True(t, IsLetter('e'))
- assert.True(t, IsLetter('q'))
- assert.True(t, IsLetter('z'))
- assert.True(t, IsLetter('A'))
- assert.True(t, IsLetter('E'))
- assert.True(t, IsLetter('Q'))
- assert.True(t, IsLetter('Z'))
- assert.True(t, IsLetter('_'))
- assert.False(t, IsLetter('-'))
- assert.False(t, IsLetter('1'))
- assert.False(t, IsLetter('$'))
- }
-
- func TestDetectContentTypeLongerThanSniffLen(t *testing.T) {
- // Pre-condition: Shorter than sniffLen detects SVG.
- assert.Equal(t, "image/svg+xml", DetectContentType([]byte(`<!-- Comment --><svg></svg>`)))
- // Longer than sniffLen detects something else.
- assert.Equal(t, "text/plain; charset=utf-8", DetectContentType([]byte(`<!--
- Comment Comment Comment Comment Comment Comment Comment Comment Comment Comment
- Comment Comment Comment Comment Comment Comment Comment Comment Comment Comment
- Comment Comment Comment Comment Comment Comment Comment Comment Comment Comment
- Comment Comment Comment Comment Comment Comment Comment Comment Comment Comment
- Comment Comment Comment Comment Comment Comment Comment Comment Comment Comment
- Comment Comment Comment Comment Comment Comment Comment Comment Comment Comment
- Comment Comment Comment --><svg></svg>`)))
- }
-
- func TestIsTextFile(t *testing.T) {
- assert.True(t, IsTextFile([]byte{}))
- assert.True(t, IsTextFile([]byte("lorem ipsum")))
- }
-
- func TestIsSVGImageFile(t *testing.T) {
- assert.True(t, IsSVGImageFile([]byte("<svg></svg>")))
- assert.True(t, IsSVGImageFile([]byte(" <svg></svg>")))
- assert.True(t, IsSVGImageFile([]byte(`<svg width="100"></svg>`)))
- assert.True(t, IsSVGImageFile([]byte("<svg/>")))
- assert.True(t, IsSVGImageFile([]byte(`<?xml version="1.0" encoding="UTF-8"?><svg></svg>`)))
- assert.True(t, IsSVGImageFile([]byte(`<!-- Comment -->
- <svg></svg>`)))
- assert.True(t, IsSVGImageFile([]byte(`<!-- Multiple -->
- <!-- Comments -->
- <svg></svg>`)))
- assert.True(t, IsSVGImageFile([]byte(`<!-- Multiline
- Comment -->
- <svg></svg>`)))
- assert.True(t, IsSVGImageFile([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- Comment -->
- <svg></svg>`)))
- assert.True(t, IsSVGImageFile([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- Multiple -->
- <!-- Comments -->
- <svg></svg>`)))
- assert.True(t, IsSVGImageFile([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- Multline
- Comment -->
- <svg></svg>`)))
- assert.False(t, IsSVGImageFile([]byte{}))
- assert.False(t, IsSVGImageFile([]byte("svg")))
- assert.False(t, IsSVGImageFile([]byte("<svgfoo></svgfoo>")))
- assert.False(t, IsSVGImageFile([]byte("text<svg></svg>")))
- assert.False(t, IsSVGImageFile([]byte("<html><body><svg></svg></body></html>")))
- assert.False(t, IsSVGImageFile([]byte(`<script>"<svg></svg>"</script>`)))
- assert.False(t, IsSVGImageFile([]byte(`<!-- <svg></svg> inside comment -->
- <foo></foo>`)))
- assert.False(t, IsSVGImageFile([]byte(`<?xml version="1.0" encoding="UTF-8"?>
- <!-- <svg></svg> inside comment -->
- <foo></foo>`)))
- }
-
- func TestFormatNumberSI(t *testing.T) {
- assert.Equal(t, "125", FormatNumberSI(int(125)))
- assert.Equal(t, "1.3k", FormatNumberSI(int64(1317)))
- assert.Equal(t, "21.3M", FormatNumberSI(21317675))
- assert.Equal(t, "45.7G", FormatNumberSI(45721317675))
- assert.Equal(t, "", FormatNumberSI("test"))
- }
-
- // TODO: IsImageFile(), currently no idea how to test
- // TODO: IsPDFFile(), currently no idea how to test
|