Add validations to 'CanToggleData' Test (#528)

appium · Oct 26, 2022 · ca08924 · ca08924
1 parent 46b0cdf
commit ca08924
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/test/integration/Android/Device/NetworkTests.cs b/test/integration/Android/Device/NetworkTests.cs
@@ -26,7 +26,10 @@ public void SetUp()
         [OneTimeTearDown]
         public void TearDown()
         {
-            _driver.Dispose();
+            if (_driver != null)
+            {
+                _driver.Dispose();
+            }            
         }
 
         [Test]
@@ -35,7 +38,11 @@ public void CanToggleDataTest()
             var androidDriver = (AndroidDriver) _driver;
 
             androidDriver.ToggleData();
+            ConnectionType currentConnectionType = androidDriver.ConnectionType;
+            Assert.That(currentConnectionType, Is.EqualTo(ConnectionType.WifiOnly));
             androidDriver.ToggleData();
+            currentConnectionType = androidDriver.ConnectionType;
+            Assert.That(currentConnectionType, Is.EqualTo(ConnectionType.AllNetworkOn));
         }
 
         [Test]