fix RootNamespace and introduce Namespace getter (#24)

Author: isopov

namespace_test.go

@@ -22,7 +22,7 @@ func TestNamespaceName(t *testing.T) {
 	require.EqualValues(t, "nsTest1.child", nsTest1Child.FullName())
 }
 
-func TestNamespace(t *testing.T) {
+func TestIsNamespaceOf(t *testing.T) {
 	require.True(t, nsTest0.IsNamespaceOf(nsTestET0))
 	require.False(t, nsTest1.IsNamespaceOf(nsTestET0))
 	require.False(t, nsTest0.IsNamespaceOf(nsTestET1))
@@ -39,3 +39,16 @@ func TestSubNamespace(t *testing.T) {
 	require.True(t, nsTest1Child.IsNamespaceOf(nsTestChild1ET))
 	require.True(t, nsTest1Child.IsNamespaceOf(nsTestChild1ETChild))
 }
+
+
+func TestRootNamespace(t *testing.T) {
+	require.Equal(t, nsTest1, nsTestChild1ET.NewWithNoMessage().Type().RootNamespace())
+}
+
+func TestNamespace(t *testing.T) {
+	require.Equal(t, nsTest1Child, nsTestChild1ET.NewWithNoMessage().Type().Namespace())
+}
+
+func TestSubTypeNamespaceFullName(t *testing.T) {
+	require.Equal(t, "nsTest1.child", nsTestChild1ETChild.Namespace().FullName())
+}

type.go

@@ -112,9 +112,18 @@ func (t *Type) FullName() string {
 	return t.fullName
 }
 
+// Namespace returns a namespace this type belongs to.
+func (t *Type) Namespace() Namespace {
+	return t.namespace
+}
+
 // RootNamespace returns a base namespace this type belongs to.
 func (t *Type) RootNamespace() Namespace {
-	return t.namespace
+	n := t.namespace
+	for n.parent != nil {
+		n = *n.parent
+	}
+	return n
 }
 
 func (t *Type) String() string {

type_test.go

@@ -15,14 +15,6 @@ func TestSubTypeName(t *testing.T) {
 	require.Equal(t, "foo.bar.internal.wat", testSubtype1.FullName())
 }
 
-func TestRootNamespace(t *testing.T) {
-	require.Equal(t, testNamespace, testType.NewWithNoMessage().Type().RootNamespace())
-}
-
-func TestSubTypeNamespace(t *testing.T) {
-	require.Equal(t, "foo", testSubtype1.RootNamespace().FullName())
-}
-
 func TestErrorTypeCheck(t *testing.T) {
 	require.True(t, testSubtype1.IsOfType(testSubtype1))
 	require.False(t, testSubtype1.IsOfType(NewNamespace("a").NewType("b")))