[SetupWizardLib] Add getNavigationBar

Add getNavigationBar method to SetupWizardLayout

Change-Id: I49eeb5cfee2917d5b4e3f552f9c3b68747dcd79a

1 files changed, 16 insertions, 0 deletions

diff --git a/library/test/src/com/android/setupwizardlib/test/SetupWizardLayoutTests.java b/library/test/src/com/android/setupwizardlib/test/SetupWizardLayoutTests.java
index ddaa391..f5cd287 100644
--- a/library/test/src/com/android/setupwizardlib/test/SetupWizardLayoutTests.java
+++ b/library/test/src/com/android/setupwizardlib/test/SetupWizardLayoutTests.java
@@ -102,6 +102,22 @@ public class SetupWizardLayoutTests extends InstrumentationTestCase {
         layout.setLayoutBackground(new ColorDrawable(Color.RED));
     }
 
+    @SmallTest
+    public void testGetNavigationBar() {
+        final SetupWizardLayout layout = new SetupWizardLayout(mContext);
+        final NavigationBar navigationBar = layout.getNavigationBar();
+        assertEquals("Navigation bar should have ID = @id/suw_layout_navigation_bar",
+                R.id.suw_layout_navigation_bar, navigationBar.getId());
+    }
+
+    @SmallTest
+    public void testGetNavigationBarNull() {
+        // test_template does not have navigation bar so getNavigationBar() should return null.
+        final SetupWizardLayout layout = new SetupWizardLayout(mContext, R.layout.test_template);
+        final NavigationBar navigationBar = layout.getNavigationBar();
+        assertNull("getNavigationBar() in test_template should return null", navigationBar);
+    }
+
     private void assertDefaultTemplateInflated(SetupWizardLayout layout) {
         View decorView = layout.findViewById(R.id.suw_layout_decor);
         View navbar = layout.findViewById(R.id.suw_layout_navigation_bar);