def test_validates_enum_parameters(self): schema = [{"name": "bar", "type": "enum", "enumOptions": ["baz", "qux"]}] query = ParameterizedQuery("foo {{bar}}", schema) query.apply({"bar": "baz"}) self.assertEquals("foo baz", query.text)
def test_validates_date_parameters(self): schema = [{"name": "bar", "type": "date"}] query = ParameterizedQuery("foo {{bar}}", schema) query.apply({"bar": "2000-01-01 12:00:00"}) self.assertEquals("foo 2000-01-01 12:00:00", query.text)
def test_coerces_number_parameters(self): schema = [{"name": "bar", "type": "number"}] query = ParameterizedQuery("foo {{bar}}", schema) query.apply({"bar": "3.14"}) self.assertEquals("foo 3.14", query.text)
def test_validation_accepts_integer_values_for_dropdowns(self, _): schema = [{"name": "bar", "type": "query", "queryId": 1}] query = ParameterizedQuery("foo {{bar}}", schema) query.apply({"bar": 1}) self.assertEquals("foo 1", query.text)
def test_validates_date_range_parameters(self): schema = [{"name": "bar", "type": "date-range"}] query = ParameterizedQuery("foo {{bar.start}} {{bar.end}}", schema) query.apply({"bar": {"start": "2000-01-01 12:00:00", "end": "2000-12-31 12:00:00"}}) self.assertEquals("foo 2000-01-01 12:00:00 2000-12-31 12:00:00", query.text)
def test_validates_text_parameters(self): schema = [{"name": "bar", "type": "text"}] query = ParameterizedQuery("foo {{bar}}", schema) query.apply({"bar": u"baz"}) self.assertEquals("foo baz", query.text)
def test_validates_query_parameters(self, _): schema = [{"name": "bar", "type": "query", "queryId": 1}] query = ParameterizedQuery("foo {{bar}}", schema) query.apply({"bar": "baz"}) self.assertEquals("foo baz", query.text)
def test_raises_on_invalid_text_parameters(self): schema = [{"name": "bar", "type": "text"}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": 7})
def test_raises_on_parameters_not_in_schema(self): schema = [{"name": "bar", "type": "text"}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"qux": 7})
def test_raises_on_unexpected_param_types(self): schema = [{"name": "bar", "type": "burrito"}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": "baz"})
def test_raises_on_unlisted_query_value_parameters(self, _): schema = [{"name": "bar", "type": "query", "queryId": 1}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": "shlomo"})
def test_raises_on_invalid_date_parameters(self): schema = [{"name": "bar", "type": "date"}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": "baz"})
def test_raises_on_none_for_date_parameters(self): schema = [{"name": "bar", "type": "date"}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": None})
def test_is_not_safe_if_expecting_text_parameter(self): schema = [{"name": "bar", "type": "text"}] query = ParameterizedQuery("foo", schema) self.assertFalse(query.is_safe)
def test_raises_on_unlisted_enum_value_parameters(self): schema = [{"name": "bar", "type": "enum", "enumOptions": ["baz", "qux"]}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": "shlomo"})
def test_raises_on_invalid_date_range_parameters(self): schema = [{"name": "bar", "type": "date-range"}] query = ParameterizedQuery("foo", schema) with pytest.raises(InvalidParameterError): query.apply({"bar": "baz"})
def test_finds_all_params_when_missing(self): query = ParameterizedQuery(u"SELECT {{param}} FROM {{table}}") self.assertEqual(set(['param', 'table']), query.missing_params)
def test_returns_empty_list_for_regular_query(self): query = ParameterizedQuery(u"SELECT 1") self.assertEqual(set([]), query.missing_params)
def test_is_safe_if_not_expecting_any_parameters(self): schema = [] query = ParameterizedQuery("foo", schema) self.assertTrue(query.is_safe)
def parameterized(self): return ParameterizedQuery(self.query_text, self.parameters, self.org)