summaryrefslogtreecommitdiffstats
path: root/tests/units/client.py
blob: 21cf21fb12315bc00fe3280110220703a2c78a47 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#
# Copyright (C) 2020 Soleta Networks <info@soleta.eu>
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the
# Free Software Foundation, version 3.
#

import subprocess
import os

class Client():

    def __init__(self):
        self.null = open(os.devnull, 'wb')
        self.proc = subprocess.Popen(['python3', 'main.py'],
                                     cwd='../',
                                     stdout=self.null,
                                     stderr=self.null)

    def stop(self):
        self.proc.terminate()
        self.proc.kill()
        self.proc.wait()
        self.null.close()