Python MssqlCli._should_show_limit_prompt Examples

Example #1

File: test_rowlimit.py Project: ekkarat-w-gmail-com/mssql-cli

    def test_default_row_limit(self):
        cli = MssqlCli(self.DEFAULT_OPTIONS)
        result = cli._should_show_limit_prompt(['row'] * self.low_count)
        assert not result

        result = cli._should_show_limit_prompt(['row'] * self.over_default)
        assert result

Example #2

def test_default_row_limit():
    cli = MssqlCli(DEFAULT_OPTIONS)
    stmt = "SELECT * FROM students"
    result = cli._should_show_limit_prompt(stmt, ['row'] * low_count)
    assert result is False

    result = cli._should_show_limit_prompt(stmt, ['row'] * over_default)
    assert result is True

Example #3

    def test_default_row_limit(self):
        cli = MssqlCli(self.DEFAULT_OPTIONS)
        stmt = "SELECT * FROM students"
        result = cli._should_show_limit_prompt(stmt, ['row']*self.low_count)
        assert not result

        result = cli._should_show_limit_prompt(stmt, ['row']*self.over_default)
        assert result

Example #4

def test_set_row_limit():
    cli_options = create_mssql_cli_options(row_limit=LIMIT)
    cli = MssqlCli(cli_options)
    stmt = "SELECT * FROM students"
    result = cli._should_show_limit_prompt(stmt, ['row'] * over_default)
    assert result is False

    result = cli._should_show_limit_prompt(stmt, ['row'] * over_limit)
    assert result is True

Example #5

File: test_rowlimit.py Project: ekkarat-w-gmail-com/mssql-cli

    def test_row_limit_on_non_select(self):
        cli = MssqlCli(self.DEFAULT_OPTIONS)
        result = cli._should_show_limit_prompt(None)
        assert not result

        cli_options = create_mssql_cli_options(row_limit=0)
        assert cli_options.row_limit == 0
        cli = MssqlCli(cli_options)
        result = cli._should_show_limit_prompt(['row'] * self.over_default)
        assert cli.row_limit == 0
        assert not result

Example #6

def test_row_limit_on_non_select():
    cli = MssqlCli(DEFAULT_OPTIONS)
    stmt = "UPDATE students set name='Boby'"
    result = cli._should_show_limit_prompt(stmt, None)
    assert result is False

    cli_options = create_mssql_cli_options(row_limit=0)
    assert cli_options.row_limit is 0
    cli = MssqlCli(cli_options)
    result = cli._should_show_limit_prompt(stmt, ['row'] * over_default)
    assert cli.row_limit is 0
    assert result is False

Example #7

File: test_rowlimit.py Project: ekkarat-w-gmail-com/mssql-cli

    def test_no_limit(self):
        cli_options = create_mssql_cli_options(row_limit=0)
        cli = MssqlCli(cli_options)
        assert cli.row_limit == 0

        result = cli._should_show_limit_prompt(['row'] * self.over_limit)
        assert not result

Example #8

def test_no_limit():
    cli_options = create_mssql_cli_options(row_limit=0)
    cli = MssqlCli(cli_options)
    assert cli.row_limit is 0
    stmt = "SELECT * FROM students"

    result = cli._should_show_limit_prompt(stmt, ['row'] * over_limit)
    assert result is False