Subversion Repositories openrisc

// Copyright 2010 The Go Authors. All rights reserved.

// Use of this source code is governed by a BSD-style

// license that can be found in the LICENSE file.

package mime

import "testing"

var typeTests = initMimeForTests()

func TestTypeByExtension(t *testing.T) {

        for ext, want := range typeTests {

                val := TypeByExtension(ext)

                if val != want {

                        t.Errorf("TypeByExtension(%q) = %q, want %q", ext, val, want)

                }

        }

}

func TestCustomExtension(t *testing.T) {

        custom := "text/xml; charset=iso-8859-1"

        if error := AddExtensionType(".xml", custom); error != nil {

                t.Fatalf("error %s for AddExtension(%s)", error, custom)

        }

        if registered := TypeByExtension(".xml"); registered != custom {

                t.Fatalf("registered %s instead of %s", registered, custom)

        }

}