From 7032be6049854e259921cfadf1ddb201b0b3b14b Mon Sep 17 00:00:00 2001
From: Dan Schaper <dan.schaper@pi-hole.net>
Date: Sun, 19 Feb 2017 21:38:02 -0800
Subject: [PATCH] Template for FTL engine type tests.

Signed-off-by: Dan Schaper <dan.schaper@pi-hole.net>
---
 test/test_automated_install.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/test/test_automated_install.py b/test/test_automated_install.py
index 89d9a0e0..5096e2b2 100644
--- a/test/test_automated_install.py
+++ b/test/test_automated_install.py
@@ -297,6 +297,35 @@ def test_update_package_cache_failure_no_errors(Pihole):
     assert 'ERROR' in updateCache.stdout
     assert 'done!' not in updateCache.stdout
 
+def test_FTL_download_aarch64_no_errors(Pihole):
+    ''' confirms only aarch64 package is downloaded for FTL engine '''
+    # mock uname to return aarch64 platform
+    mock_command('uname', {'-m':('aarch64', '0')}, Pihole)
+    detectPlatform = Pihole.run('''
+    source /opt/pihole/basic-install.sh
+    FTLdownload
+    touch /lib/ld-linux-aarch64.so.1
+    ''')
+    expected_stdout = 'Detected ARM-aarch64 architecture'
+    assert expected_stdout in detectPlatform.stdout
+    FTL_binary = Pihole.run('echo ${binary}').stdout
+    assert 'pihole-FTL-aarch64-linux-gnu' in FTL_binary
+
+def test_FTL_download_armv6l_Pi_Zero_no_errors(Pihole):
+    ''' confirms only armv6l package is downloaded for FTL engine (Pi-Zero) '''
+    # mock uname to return aarch64 platform
+    mock_command('uname', {'-m':('armv6l', '0')}, Pihole)
+    detectPlatform = Pihole.run('''
+    source /opt/pihole/basic-install.sh
+    FTLdownload
+    touch /lib/ld-linux-armhf.so.3
+    ls -lach /lib/
+    ''')
+    expected_stdout = 'Detected ARM-hf architecture'
+    assert expected_stdout in detectPlatform.stdout
+    FTL_binary = Pihole.run('echo ${binary}').stdout
+    assert 'pihole-FTL-arm-linux-gnueabihf' in FTL_binary
+
 # Helper functions
 def mock_command(script, args, container):
     ''' Allows for setup of commands we don't really want to have to run for real in unit tests '''