move is_dropping_database tests to parseutils.

Author: tsroten

test/test_main.py

@@ -3,7 +3,7 @@
 import click
 from click.testing import CliRunner
 
-from mycli.main import MyCli, cli, thanks_picker, is_dropping_database, PACKAGE_ROOT
+from mycli.main import MyCli, cli, thanks_picker, PACKAGE_ROOT
 from mycli.packages.special.main import COMMANDS as SPECIAL_COMMANDS
 from .utils import USER, HOST, PORT, PASSWORD, dbtest, run
 
@@ -147,17 +147,6 @@ def test_thanks_picker_utf8():
     assert name and isinstance(name, str)
 
 
-def test_is_dropping_database():
-    is_dropping_text = "DROP DATABASE foo;"
-    assert is_dropping_database(is_dropping_text, 'foo')
-    is_not_dropping_text = "DROP DATABASE foo; CREATE DATABASE foo;"
-    assert not is_dropping_database(is_not_dropping_text, 'foo')
-    is_dropping_other_text = "DROP DATABASE bar;"
-    assert not is_dropping_database(is_dropping_other_text, 'foo')
-    is_not_dropping_other_text = "DROP DATABASE foo; CREATE DATABASE bar;"
-    assert is_dropping_database(is_not_dropping_other_text, 'foo')
-
-
 def test_help_strings_end_with_periods():
     """Make sure click options have help text that end with a period."""
     for param in cli.params:

test/test_parseutils.py

@@ -175,6 +175,8 @@ def test_query_has_where_clause(sql, has_where_clause):
         ('drop schema foo', 'bar', False),
         ('drop database bar', 'foo', False),
         ('drop database foo', None, False),
+        ('drop database foo; create database foo', 'foo', False),
+        ('drop database foo; create database bar', 'foo', True),
         ('select bar from foo; drop database bazz', 'foo', False),
         ('select bar from foo; drop database bazz', 'bazz', True),
         ('-- dropping database \n '