mirror of
https://github.com/calebstewart/pwncat.git
synced 2024-12-11 17:34:19 +01:00
25 lines
634 B
Python
25 lines
634 B
Python
#!/usr/bin/env python3
|
|
import io
|
|
|
|
import pytest
|
|
import paramiko
|
|
|
|
|
|
def test_digitalocean(digital_ocean):
|
|
|
|
key = paramiko.rsakey.RSAKey.from_private_key(
|
|
io.StringIO(digital_ocean.ssh_key.exportKey("PEM").decode("utf-8"))
|
|
)
|
|
|
|
ubuntu = digital_ocean.ubuntu
|
|
ubuntu.load()
|
|
|
|
client = paramiko.client.SSHClient()
|
|
client.load_system_host_keys()
|
|
client.set_missing_host_key_policy(paramiko.client.AutoAddPolicy)
|
|
client.connect(ubuntu.ip_address, username=digital_ocean.user, pkey=key)
|
|
|
|
stdin, stdout, stderr = client.exec_command("whoami")
|
|
|
|
assert stdout.read().strip().decode("utf-8") == "root"
|