def test_handle_dbhash_returns_const_for_empty_db(self):
# hashlib.sha1().digest()
null_hash = '\xda9\xa3\xee^kK\r2U\xbf\xef\x95`\x18\x90\xaf\xd8\x07\t'
self.station.payload = ""
handle_listdbhash(self.station)
resp = self.station.stream.pop()
self.assertIsInstance(resp, response.Response)
db_hash = DBHash()
db_hash.ParseFromString(resp.payload)
self.assertEqual(db_hash.hash, null_hash)
self.assertEqual(db_hash.prefix, "")
def test_handle_dbhash_returns_const_for_empty_db(self):
# hashlib.sha1().digest()
null_hash = '\xda9\xa3\xee^kK\r2U\xbf\xef\x95`\x18\x90\xaf\xd8\x07\t'
self.station.payload = ""
handle_listdbhash(self.station)
resp = self.station.stream.pop()
self.assertIsInstance(resp, response.Response)
db_hash = DBHash()
db_hash.ParseFromString(resp.payload)
self.assertEqual(db_hash.hash, null_hash)
self.assertEqual(db_hash.prefix, "")
def test_prefixes_work_for_populated_db(self):
for i in xrange(100):
self.station.station.write("butts %d" % i)
hashes = {'a': "\xa9K}\xa3\x9d\x9a\xb4\xab\xb2\xb6i.\xb5#\x1d/\xfdKu\xf4",
'1c': "\x1bZ\x8c\x9dETL4H\"`\te\xca\xa9Vy\x9f\xfdS",
'eb9': "\xc4q\x9a\xe0\xadJ\x12\x19n\x06\x18\x9c*\xae\nT-\xe3\t>"
}
for prefix in hashes:
self.station.payload = prefix
handle_listdbhash(self.station)
resp = self.station.stream.pop()
self.assertIsInstance(resp, response.Response)
db_hash = DBHash()
db_hash.ParseFromString(resp.payload)
self.assertEqual(db_hash.hash, hashes[prefix])
self.assertEqual(db_hash.prefix, prefix)
def test_prefixes_work_for_populated_db(self):
for i in xrange(100):
self.station.station.write("butts %d" % i)
hashes = {
'a': "\xa9K}\xa3\x9d\x9a\xb4\xab\xb2\xb6i.\xb5#\x1d/\xfdKu\xf4",
'1c': "\x1bZ\x8c\x9dETL4H\"`\te\xca\xa9Vy\x9f\xfdS",
'eb9': "\xc4q\x9a\xe0\xadJ\x12\x19n\x06\x18\x9c*\xae\nT-\xe3\t>"
}
for prefix in hashes:
self.station.payload = prefix
handle_listdbhash(self.station)
resp = self.station.stream.pop()
self.assertIsInstance(resp, response.Response)
db_hash = DBHash()
db_hash.ParseFromString(resp.payload)
self.assertEqual(db_hash.hash, hashes[prefix])
self.assertEqual(db_hash.prefix, prefix)