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")))