def test_ssl_lookup_csv_send(self):
engine_config = SenderConfigSSL(address=(self.server, self.port),
key=self.key, cert=self.cert,
chain=self.chain)
con = Sender(engine_config)
lookup = Lookup(name=self.lookup_name, historic_tag=None, con=con)
with open(self.lookup_file) as f:
line = f.readline()
lookup.send_csv(self.lookup_file,
headers=line.rstrip().split(","),
key=self.lookup_key)
con.socket.shutdown(0)
def test_escape_quotes_in_send_csv(self):
engine_config = SenderConfigSSL(
address=(self.server, self.port),
key=self.key,
cert=self.cert,
)
con = Sender(engine_config)
lookup = Lookup(name=self.lookup_name,
historic_tag=None,
con=con,
escape_quotes=True)
with mock.patch.object(Lookup, 'clean_field',
wraps=Lookup.clean_field) as clean_field:
lookup.send_csv(path=self.lookup_file,
has_header=True,
key=self.lookup_key)
clean_field.assert_called_with('ffffff', True)
import os
from devo.sender import Sender, SenderConfigSSL, Lookup
server = "us.elb.relay.logtrust.net"
port = 443
key = os.getenv('DEVO_SENDER_KEY')
cert = os.getenv('DEVO_SENDER_CERT')
chain = os.getenv('DEVO_SENDER_CHAIN')
lookup_name = 'Test_Lookup_Csv'
lookup_file = "".join((os.path.dirname(os.path.abspath(__file__)), os.sep,
"example_data", os.sep, "example.csv"))
lookup_key = 'KEY'
engine_config = SenderConfigSSL(address=(server, port),
key=key,
cert=cert,
chain=chain)
con = Sender(engine_config)
lookup = Lookup(name=lookup_name, historic_tag=None, con=con)
with open(lookup_file) as f:
line = f.readline()
lookup.send_csv(lookup_file, headers=line.rstrip().split(","), key=lookup_key)
con.close()
