From d25b596f4f23615ac72815aeb735c1ec2dad0023 Mon Sep 17 00:00:00 2001 From: Gleb Koveshnikov Date: Sat, 19 Jun 2021 14:22:41 +0300 Subject: [PATCH] fix coverage and linter --- pypika/queries.py | 2 +- pypika/tests/test_create.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/pypika/queries.py b/pypika/queries.py index 26443c97..e268992d 100644 --- a/pypika/queries.py +++ b/pypika/queries.py @@ -1857,7 +1857,7 @@ def foreign_key( reference_table: Union[str, Table], reference_columns: List[Union[str, Column]], on_delete: ReferenceOption = None, - on_update: ReferenceOption = None + on_update: ReferenceOption = None, ) -> "CreateQueryBuilder": """ Adds a foreign key constraint. diff --git a/pypika/tests/test_create.py b/pypika/tests/test_create.py index d79b3d5a..a4b5f5e3 100644 --- a/pypika/tests/test_create.py +++ b/pypika/tests/test_create.py @@ -156,6 +156,12 @@ def test_create_table_with_select_and_columns_fails(self): with self.assertRaises(AttributeError): Query.create_table(self.new_table).as_select(select).columns(self.foo, self.bar) + with self.subTest("repeated foreign key"): + with self.assertRaises(AttributeError): + Query.create_table(self.new_table) \ + .foreign_key([self.foo], self.existing_table, [self.bar]) \ + .foreign_key([self.foo], self.existing_table, [self.bar]) + def test_create_table_as_select_not_query_raises_error(self): with self.assertRaises(TypeError): Query.create_table(self.new_table).as_select("abc")