import testinfra.utils.ansible_runner
import re

def test_packages(host):
  package = host.package('squid')
  assert package.is_installed

def test_config_file(host):
  path = host.file('/etc/squid/squid.conf')
  assert path.exists
  assert path.is_file
  assert path.user == 'root'
  assert path.group == 'proxy'
  assert path.mode == 0o640
  assert path.contains('acl deny_domain dstdomain "/etc/squid/deny_domain.list"')
  assert path.contains('acl deny_domain dstdomain bing.com')
  assert path.contains('http_access deny deny_domain')

def test_list_file(host):
  path = host.file('/etc/squid/deny_domain.list')
  assert path.exists
  assert path.is_file
  assert path.user == 'root'
  assert path.group == 'proxy'
  assert path.mode == 0o640
  assert path.contains('free.fr\nsfr.fr')

def test_service(host):
  service = host.service('squid')
  assert service.is_running
  assert service.is_enabled

def test_sockets(host):
  socket = host.socket('tcp://0.0.0.0:3128')
  assert socket.is_listening

#def test_users(host):
#  cmd = host.run('influx -execute "SHOW USERS"')
#  assert cmd.succeeded
#  assert re.search(r'admin\s+true', cmd.stdout)
#  assert re.search(r'test\s+false', cmd.stdout)
#  assert not re.search('user_absent', cmd.stdout)
#
#def test_databases(host):
#  cmd = host.run('influx -execute "SHOW DATABASES"')
#  assert cmd.succeeded
#  assert 'test_db' in cmd.stdout
#
#def test_databases(host):
#  cmd = host.run('influx -execute "SHOW RETENTION POLICIES ON test_db"')
#  assert cmd.succeeded
#  assert re.search(r'default\s+24h0m0s\s+1h0m0s\s+1\s+true', cmd.stdout)
#
#def test_grants(host):
#  cmd = host.run('influx -execute "SHOW GRANTS FOR test"')
#  assert cmd.succeeded
#  assert re.search(r'test_db\s+WRITE', cmd.stdout)