class TestStorage(unittest.TestCase): def setUp(self): self.db = DB() self.db.insert("192.168.1.1/24", "Example Message") self.db.insert("192.168.1.92/28", "something else") def test_invalid(self): self.assertRaises(KeyError, self.db.get, "192.168.2.1") def test_value(self): self.assertEqual(self.db.get("192.168.1.1")[1], "Example Message") self.assertEqual(self.db.get("192.168.1.92")[1], "something else") def test_network(self): self.assertEqual(self.db.get("192.168.1.0")[1], "Example Message") self.assertEqual(self.db.get("192.168.1.80")[1], "something else") def test_delete(self): self.assertEqual(self.db.get("192.168.1.0")[1], "Example Message") del self.db["192.168.1.5/24"] self.assertRaises(KeyError, self.db.get, "192.168.1.0") self.assertEqual(self.db.get("192.168.1.80")[1], "something else")
def setUp(self): self.db = DB() self.db.insert("192.168.1.1/24", "Example Message") self.db.insert("192.168.1.92/28", "something else")