Python validate_sql_select Examples

Example #1

File: test_sensor.py Project: crazyfish1111/home

    def test_invalid_query(self):
        """Test the SQL sensor for invalid queries."""
        with pytest.raises(vol.Invalid):
            validate_sql_select("DROP TABLE *")

        config = {
            'sensor': {
                'platform':
                'sql',
                'db_url':
                'sqlite://',
                'queries': [{
                    'name': 'count_tables',
                    'query': 'SELECT * value FROM sqlite_master;',
                    'column': 'value',
                }]
            }
        }

        assert setup_component(self.hass, 'sensor', config)

        state = self.hass.states.get('sensor.count_tables')
        assert state.state == STATE_UNKNOWN

Example #2

File: test_sensor.py Project: pkrolkgp/home-assistant

    def test_invalid_query(self):
        """Test the SQL sensor for invalid queries."""
        with pytest.raises(vol.Invalid):
            validate_sql_select("DROP TABLE *")

        config = {
            "sensor": {
                "platform":
                "sql",
                "db_url":
                "sqlite://",
                "queries": [{
                    "name": "count_tables",
                    "query": "SELECT * value FROM sqlite_master;",
                    "column": "value",
                }],
            }
        }

        assert setup_component(self.hass, "sensor", config)

        state = self.hass.states.get("sensor.count_tables")
        assert state.state == STATE_UNKNOWN

Example #3

File: test_sensor.py Project: dmulcahey/home-assistant

async def test_invalid_query(hass: HomeAssistant) -> None:
    """Test the SQL sensor for invalid queries."""
    with pytest.raises(vol.Invalid):
        validate_sql_select("DROP TABLE *")

    config = {
        "sensor": {
            "platform": "sql",
            "db_url": "sqlite://",
            "queries": [
                {
                    "name": "count_tables",
                    "query": "SELECT * value FROM sqlite_master;",
                    "column": "value",
                }
            ],
        }
    }

    assert await async_setup_component(hass, "sensor", config)
    await hass.async_block_till_done()

    state = hass.states.get("sensor.count_tables")
    assert state.state == STATE_UNKNOWN