Check DMAR after iommu is enabled

Change-Id: Ie1b4619f89ccc34ca227e3f3b87314c0d6ecc285
diff --git a/handlers/main.yml b/handlers/main.yml
index e699bf5..1e6905e 100644
--- a/handlers/main.yml
+++ b/handlers/main.yml
@@ -17,7 +17,18 @@
 - name: reboot
   reboot:
     test_command: "systemctl show -p LoadState sriov.service | grep loaded"
-  notify: ensure vfs
+  notify:
+    - ensure vtd
+    - ensure vfs
+
+- name: ensure vtd
+  shell: "set -o pipefail && dmesg | grep DMAR-IR"
+  args:
+    executable: /bin/bash
+  register: check_vt_d
+  changed_when: check_vt_d.rc != 0
+  failed_when: check_vt_d.rc != 0
+  when: molecule_handler_notest is not defined
 
 - name: ensure vfs
   command: "cat /sys/class/net/{{ sriov_pf_name }}/device/sriov_numvfs"
diff --git a/tasks/main.yml b/tasks/main.yml
index 702d868..f7c365e 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -10,15 +10,6 @@
 - name: include OS-specific tasks
   include_tasks: "{{ ansible_os_family }}.yml"
 
-- name: Ensure VT-d is enabled in BIOS
-  tags: molecule-notest
-  shell: "set -o pipefail && dmesg | grep DMAR-IR"
-  args:
-    executable: /bin/bash
-  register: check_vt_d
-  changed_when: check_vt_d.rc != 0
-  failed_when: check_vt_d.rc != 0
-
 - name: Load vfio_pci module to the kernel
   modprobe:
     name: vfio_pci