mojom/generators/go/translator/names_test.go

// Copyright 2016 The Chromium 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 translator

import (
	"testing"

	"mojom/generated/mojom_files"
	"mojom/generated/mojom_types"
)

func TestFormatName(t *testing.T) {
	testCases := []struct {
		expected string
		name     string
	}{
		{"Hello", "hello"},
		{"Hello", "Hello"},
		{"HelloWorld", "HelloWorld"},
		{"HelloWorld", "hello_world"},
		{"HttpWorld", "HTTPWorld"},
		{"HttpWorld", "HTTP_World"},
	}

	for _, testCase := range testCases {
		checkEq(t, testCase.expected, formatName(testCase.name))
	}
}

func TestPrivateName(t *testing.T) {
	testCases := []struct {
		expected string
		public   string
	}{
		{"hello", "hello"},
		{"hello", "Hello"},
	}

	for _, testCase := range testCases {
		checkEq(t, testCase.expected, privateName(testCase.public))
	}
}

func TestGoTypeName(t *testing.T) {
	structName := "foo"
	s := mojom_types.MojomStruct{
		DeclData: &mojom_types.DeclarationData{ShortName: &structName},
	}

	graph := mojom_files.MojomFileGraph{}
	typeKey := "typeKey"
	graph.ResolvedTypes = map[string]mojom_types.UserDefinedType{
		typeKey: &mojom_types.UserDefinedTypeStructType{s},
	}

	translator := NewTranslator(&graph)

	checkEq(t, "Foo", translator.goTypeName(typeKey))
}

func TestGoTypeNameNestedEnum(t *testing.T) {
	structName := "foo"
	structTypeKey := "structTypeKey"
	s := mojom_types.MojomStruct{
		DeclData: &mojom_types.DeclarationData{ShortName: &structName},
	}

	enumName := "some_enum"
	enumTypeKey := "enumTypeKey"
	e := mojom_types.MojomEnum{
		DeclData: &mojom_types.DeclarationData{
			ShortName:        &enumName,
			ContainerTypeKey: &structTypeKey,
		},
	}

	graph := mojom_files.MojomFileGraph{}
	graph.ResolvedTypes = map[string]mojom_types.UserDefinedType{
		structTypeKey: &mojom_types.UserDefinedTypeStructType{s},
		enumTypeKey:   &mojom_types.UserDefinedTypeEnumType{e},
	}

	translator := NewTranslator(&graph)

	checkEq(t, "Foo_SomeEnum", translator.goTypeName(enumTypeKey))
}