def test_insert_examination_if_storage_was_not_open(self): """Insert examination method must return False if storage was not open.""" m = Model() file_name = './copy-test.sme.sqlite' shutil.copy2('./test.sme.sqlite', file_name) try: m.insert_exam(create_test_exam()) result = True except AttributeError: result = False finally: self.assertFalse(result)
def test_insert_examination_to_new_storage(self): """Number of examination must be 1 after insertion examination to new empty storage.""" m = Model() file_name = './copy-test.sme.sqlite' m.create_storage(file_name) m.insert_exam(create_test_exam()) m.close_storage() conn = sqlite3.connect(file_name) c = conn.cursor() c.execute("select count(*) from examination") n = c.fetchone()[0] conn.close() remove_file(file_name) self.assertEqual(n, 1)
def test_insert_examination(self): """Number of examination must be increase after insert examination.""" m = Model() file_name = './copy-test.sme.sqlite' shutil.copy2('./test.sme.sqlite', file_name) def get_len(): conn = sqlite3.connect(file_name) c = conn.cursor() c.execute("select count(*) from examination") return c.fetchone()[0] conn.close() len_before = get_len() m.open_storage(file_name) m.insert_exam(create_test_exam()) m.close_storage() len_after = get_len() remove_file(file_name) self.assertTrue(len_after > len_before)