def test_should_remove_trigger_from_db(self): shutil.copyfile(self.test_db_file, self.temp_file.name) store = LightsSqliteStore(self.temp_file.name) store.remove_trigger(self.trigger.job_id) conn = sqlite3.connect(self.temp_file.name) cursor = conn.cursor() rows = cursor.execute("SELECT * FROM triggers").fetchall() assert_that(rows, is_(empty())) triggers = store.read_triggers() assert_that(triggers, is_(empty()))
def test_should_remove_only_single_trigger(self): shutil.copyfile("tests/two_triggers.db", self.temp_file.name) store = LightsSqliteStore(self.temp_file.name) conn = sqlite3.connect(self.temp_file.name) cursor = conn.cursor() rows = cursor.execute("SELECT * FROM triggers").fetchall() assert_that(rows, has_length(2)) triggers = store.read_triggers() assert_that(triggers, has_length(2)) store.remove_trigger(self.trigger.job_id) rows = cursor.execute("SELECT * FROM triggers").fetchall() assert_that(rows, has_length(1)) triggers = store.read_triggers() assert_that(triggers, has_length(1))