From c8be8d4a184a0877184c457e6adbe6fa8ace0d5c Mon Sep 17 00:00:00 2001 From: Jacob Zimmerman Date: Mon, 27 Mar 2017 15:48:28 -0400 Subject: [PATCH] Add test for --create-if-not-exists flag --- examples/foo1.csv | 3 +++ examples/foo2.csv | 3 +++ tests/test_utilities/test_csvsql.py | 4 ++++ 3 files changed, 10 insertions(+) create mode 100644 examples/foo1.csv create mode 100644 examples/foo2.csv diff --git a/examples/foo1.csv b/examples/foo1.csv new file mode 100644 index 000000000..d79a39056 --- /dev/null +++ b/examples/foo1.csv @@ -0,0 +1,3 @@ +id,name,age +1,Jake,22 +2,Howard,21 diff --git a/examples/foo2.csv b/examples/foo2.csv new file mode 100644 index 000000000..14d8f98a4 --- /dev/null +++ b/examples/foo2.csv @@ -0,0 +1,3 @@ +id,name,age +3,Liz,20 +4,Tim,21 diff --git a/tests/test_utilities/test_csvsql.py b/tests/test_utilities/test_csvsql.py index aa7d7ad5c..efcbdb5ce 100644 --- a/tests/test_utilities/test_csvsql.py +++ b/tests/test_utilities/test_csvsql.py @@ -158,6 +158,10 @@ def test_query_with_prefix(self): self.get_output(['--prefix', 'OR IGNORE', '--no-create', '--insert', '--db', 'sqlite:///' + self.db_file, 'examples/dummy.csv']) + def test_create_if_not_exists(self): + self.get_output(['--insert', '--create-if-not-exists', '--tables', 'foo', '--db', 'sqlite:///' + self.db_file, 'examples/foo1.csv']) + self.get_output(['--insert', '--create-if-not-exists', '--tables', 'foo', '--db', 'sqlite:///' + self.db_file, 'examples/foo2.csv']) + def test_query(self): input_file = six.StringIO("a,b,c\n1,2,3\n")