diff --git a/mockdriver/src/main/java/org/jaydebeapi/mockdriver/MockDriver.java b/mockdriver/src/main/java/org/jaydebeapi/mockdriver/MockDriver.java index d3c00ea..5db23bb 100644 --- a/mockdriver/src/main/java/org/jaydebeapi/mockdriver/MockDriver.java +++ b/mockdriver/src/main/java/org/jaydebeapi/mockdriver/MockDriver.java @@ -24,7 +24,7 @@ public class MockDriver implements Driver { @Override public Connection connect(String url, Properties info) throws SQLException { - if (!url.startsWith("jdbc:jaydebeapi://")) { + if (!acceptsURL(url)) { return null; } return Mockito.mock(Connection.class); @@ -32,7 +32,7 @@ public class MockDriver implements Driver { @Override public boolean acceptsURL(String url) throws SQLException { - return true; + return url.startsWith("jdbc:jaydebeapi://"); } @Override diff --git a/test/test_mock.py b/test/test_mock.py new file mode 100644 index 0000000..3b69637 --- /dev/null +++ b/test/test_mock.py @@ -0,0 +1,34 @@ +#-*- coding: utf-8 -*- + +# Copyright 2015 Bastian Bowe +# +# This file is part of JayDeBeApi. +# JayDeBeApi is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# JayDeBeApi is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with JayDeBeApi. If not, see +# . + +import jaydebeapi + +import unittest2 as unittest + +class MockTest(unittest.TestCase): + + def setUp(self): + self.conn = jaydebeapi.connect('org.jaydebeapi.mockdriver.MockDriver', 'jdbc:jaydebeapi://dummyurl') + + def tearDown(self): + self.conn.close() + + def test_execute(self): + cursor = self.conn.cursor() + cursor.execute("dummy stmt")