def test_container_should_return_wrapped_object_if_possible(self):
mock1 = mock.Mock(WebDriver)
mock2 = mock.Mock(WebElement)
iterable = [mock1, mock2]
container = SeleniumContainerWrapper(iterable)
wrapped1 = container.pop()
self.assertTrue(isinstance(wrapped1, SeleniumWrapper))
self.assertTrue(hasattr(wrapped1, 'waitfor'))
wrapped2 = container.pop()
self.assertTrue(isinstance(wrapped2, SeleniumWrapper))
self.assertTrue(hasattr(wrapped2, 'waitfor'))
def test_container_should_return_wrapped_object_if_possible(self):
mock1 = mock.Mock(WebDriver)
mock2 = mock.Mock(WebElement)
iterable = [mock1, mock2]
container = SeleniumContainerWrapper(iterable)
wrapped1 = container.pop()
self.assertTrue(isinstance(wrapped1, SeleniumWrapper))
self.assertTrue(hasattr(wrapped1, 'waitfor'))
wrapped2 = container.pop()
self.assertTrue(isinstance(wrapped2, SeleniumWrapper))
self.assertTrue(hasattr(wrapped2, 'waitfor'))
def test_container_should_delegate_unknown_attribute_access_to_wrapped_container(self):
container = SeleniumContainerWrapper([])
container.append(1)
container.append(1)
self.assertEquals(container.count(1), 2)
self.assertEquals(container.pop(), 1)
self.assertEquals(container.count(1), 1)
def test_container_should_delegate_unknown_attribute_access_to_wrapped_container(
self):
container = SeleniumContainerWrapper([])
container.append(1)
container.append(1)
self.assertEqual(container.count(1), 2)
self.assertEqual(container.pop(), 1)
self.assertEqual(container.count(1), 1)
