示例#1
0
class TestDatabase(unittest.TestCase):
    def setUp(self):
        initialize_database()
        self.database = Database()
        self.database.delete_all()

    def test_add_user(self):
        self.database.add_user('testi', 'salis')
        self.database.add_user('testi123', 'salis')

        user = self.database.get_user('testi')
        self.assertTrue(user)

        user = self.database.get_user('testi123')
        self.assertTrue(user)

        user = self.database.get_user('esimerkki')
        self.assertFalse(user)

    def test_find_all_users_ingredients(self):
        self.database.insert_a_new_ingredient(1619778769, 'tomaatti',
                                              1619778769, 'testi')
        self.database.insert_a_new_ingredient(1619778769, 'omena', 1619778769,
                                              'testi')
        self.database.insert_a_new_ingredient(1619778769, 'omena', 1619778769,
                                              'testi123')

        ingredients = self.database.get_all_ingredients_by_a_user('testi')
        self.assertEqual(len(ingredients), 2)

        ingredients = self.database.get_all_ingredients_by_a_user('testi123')
        self.assertEqual(len(ingredients), 1)

    def test_mark_ingredient_as_eaten(self):
        self.database.insert_a_new_ingredient(1619778769, 'tomaatti',
                                              1619778769, 'testi')
        self.database.mark_ingredient_as_eaten('testi', 1)

        ingredients = self.database.get_all_ingredients_by_a_user('testi')
        self.assertEqual(len(ingredients), 1)
        ingredient = ingredients[0]

        self.assertEqual(ingredient[4], 1)

    def test_find_ingredient(self):
        self.database.insert_a_new_ingredient(1619778769, 'tomaatti',
                                              1619778769, 'testi')
        ingredient = self.database.find_ingredient('testi', 1)
        self.assertTrue(ingredient)

        ingredient = self.database.find_ingredient('testi', 2)
        self.assertFalse(ingredient)

    def tearDown(self):
        self.database.delete_all()
示例#2
0
class DatabaseTests:
  def __init__(self):
    db = "test.db"
    os.remove(db)
    self.__db = Database(db)
    self.__db.load()

    failures = 0
    try:
      self.__db.add_user("jrenggli", "*****@*****.**", 42)
      self.__db.add_user("yschaerli", "*****@*****.**", 8383)
      self.__db.add_user("jreNggli", "*****@*****.**", 28482)
    except sqlite3.IntegrityError:
      assert(False)
    except Exception:
      failures += 1
    finally:
      assert(failures == 1)

    src = None
    try:
      src = self.__db.add_source("RCSB", "text/pdb", "http://www.rcsb.org/pdb/download/downloadFile.do?fileFormat=pdb&compression=NO&structureId={0}", "RCSB Protein Data Bank")
    except sqlite3.IntegrityError, e:
      assert(False)
    except Exception, e:
      failures += 1
      assert(False)
class DatabaseTests:
    def __init__(self):
        db = "test.db"
        os.remove(db)
        self.__db = Database(db)
        self.__db.load()

        failures = 0
        try:
            self.__db.add_user("jrenggli", "*****@*****.**", 42)
            self.__db.add_user("yschaerli", "*****@*****.**", 8383)
            self.__db.add_user("jreNggli", "*****@*****.**", 28482)
        except sqlite3.IntegrityError:
            assert (False)
        except Exception:
            failures += 1
        finally:
            assert (failures == 1)

        src = None
        try:
            src = self.__db.add_source(
                "RCSB", "text/pdb",
                "http://www.rcsb.org/pdb/download/downloadFile.do?fileFormat=pdb&compression=NO&structureId={0}",
                "RCSB Protein Data Bank")
        except sqlite3.IntegrityError, e:
            assert (False)
        except Exception, e:
            failures += 1
            assert (False)