def test_rejects_invalid_transfers(self):
self.station.set_real_id(True)
object_body = "foo bar baz butts lol"
req = MockRequest(self.station.id)
req.payload = "deadbeefdeadbeefdeadbeefdeadbeefdeadbeef"
self.station.station.registry.register(req)
git_pb = GitObject()
git_pb.data = object_body
git_pb.type = pygit2.GIT_OBJ_BLOB
self.station.payload = git_pb.SerializeToString()
self.assertRaises(AssertionError, handle_transfer, self.station)
def test_transfer_object(self):
self.station.set_real_id(True)
object_body = "foo bar baz butts lol"
oid = pygit2.hash(object_body)
req = MockRequest(self.station.id)
req.payload = oid2hex(oid)
self.station.station.registry.register(req)
git_pb = GitObject()
git_pb.data = object_body
git_pb.type = pygit2.GIT_OBJ_BLOB
self.station.payload = git_pb.SerializeToString()
handle_transfer(self.station)
self.assertTrue(self.station.station[oid2hex(oid)], object_body)