def test_extract_column_7(self, is_instance_mock):
"""Verify isinstance() is called with correct parameters.
"""
is_instance_mock.return_value = True
querying.extract_column(self.column, check=self.check_type)
is_instance_mock.assert_any_call(self.column, self.check_type)
def test_extract_column_1(self, IsInstance):
with self.assertRaises(TypeError):
IsInstance.return_value = False
querying.extract_column(self.column)
IsInstance.assert_any_call(self.column, Column)
IsInstance.assert_any_call(self.column, functions.count)
IsInstance.assert_any_call(self.column, UnaryExpression)
IsInstance.assert_any_call(self.column, BinaryExpression)
def test_extract_column_1(self, is_instance_mock):
"""Verify extract_column() raises a TypeError from innate type checks.
extract_column() should throw an exception when the object is not
any expected Column-related SQLAlchemy object.
"""
with self.assertRaises(TypeError):
is_instance_mock.return_value = False
querying.extract_column(self.column)
is_instance_mock.assert_any_call(self.column, Column)
is_instance_mock.assert_any_call(self.column, functions.count)
is_instance_mock.assert_any_call(self.column, UnaryExpression)
is_instance_mock.assert_any_call(self.column, BinaryExpression)
def test_extract_column_5(self):
binary_expression = Mock(spec=BinaryExpression)
type(binary_expression).left = self.column
self.assertEqual(querying.extract_column(binary_expression),
self.column)
def test_extract_column_4(self):
unary_expression = Mock(spec=UnaryExpression)
type(unary_expression).element = self.column
self.assertEqual(querying.extract_column(unary_expression),
self.column)
def test_extract_column_5(self):
"""Verify extract_column() converts received BinaryExpression object.
"""
binary_expression = Mock(spec=BinaryExpression)
type(binary_expression).left = self.column
self.assertEqual(querying.extract_column(binary_expression),
self.column)
def test_extract_column_4(self):
"""Verify extract_column() converts received UnaryExpression object.
"""
unary_expression = Mock(spec=UnaryExpression)
type(unary_expression).element = self.column
self.assertEqual(querying.extract_column(unary_expression),
self.column)
def test_extract_column_3(self):
column_list = PropertyMock(return_value=self.columns)
clause_list = Mock()
count = Mock(spec=functions.count)
type(count).clauses = clause_list
type(clause_list).clauses = column_list
self.assertEqual(querying.extract_column(count), self.column)
def test_extract_column_3(self):
"""Verify extract_column() converts received func.count object.
"""
column_list = PropertyMock(return_value=self.columns)
clause_list = Mock()
count = Mock(spec=functions.count)
type(count).clauses = clause_list
type(clause_list).clauses = column_list
self.assertEqual(querying.extract_column(count), self.column)
def test_extract_column_6(self):
"""Verify extract_column() converts received BinaryExpression object.
extract_column() should detect embedded UnaryExpression objects.
"""
unary_expression = Mock(spec=UnaryExpression)
binary_expression = Mock(spec=BinaryExpression)
type(unary_expression).element = self.column
type(binary_expression).left = unary_expression
self.assertEqual(querying.extract_column(binary_expression),
self.column)
def test_extract_column_8(self):
"""Verify extract_column() raises TypeError from imposed type check.
"""
with self.assertRaises(TypeError):
querying.extract_column(self.column, check=str)
def test_extract_column_2(self):
"""Verify extract_column() returns received Column object.
"""
self.assertEqual(querying.extract_column(self.column), self.column)
def test_extract_column_8(self):
with self.assertRaises(TypeError):
querying.extract_column(self.column, check=str)
def test_extract_column_7(self, IsInstance):
IsInstance.return_value = True
querying.extract_column(self.column, check=self.check_type)
IsInstance.assert_any_call(self.column, self.check_type)
def test_extract_column_2(self):
self.assertEqual(querying.extract_column(self.column), self.column)