diff --git a/tests/unit/test_model.py b/tests/unit/test_model.py
index ae40416..a71dc28 100644
--- a/tests/unit/test_model.py
+++ b/tests/unit/test_model.py
@@ -94,6 +94,9 @@
         layout.addJob(python27diablo)
 
         project_config = configloader.ProjectParser.fromYaml(layout, {
+            '_source_project': project,
+            '_source_branch': 'master',
+            '_source_configrepo': True,
             'name': 'project',
             'gate': {
                 'jobs': [
@@ -247,6 +250,9 @@
         layout.addJob(python27diablo)
 
         project_config = configloader.ProjectParser.fromYaml(layout, {
+            '_source_project': project,
+            '_source_branch': 'master',
+            '_source_configrepo': True,
             'name': 'project',
             'gate': {
                 'jobs': [
@@ -309,6 +315,9 @@
         layout.addJob(python27)
 
         project_config = configloader.ProjectParser.fromYaml(layout, {
+            '_source_project': project,
+            '_source_branch': 'master',
+            '_source_configrepo': True,
             'name': 'project',
             'gate': {
                 'jobs': [
diff --git a/tests/unit/test_v3.py b/tests/unit/test_v3.py
index 31410fb..0ba5ff8 100644
--- a/tests/unit/test_v3.py
+++ b/tests/unit/test_v3.py
@@ -129,3 +129,9 @@
         self.assertEqual(build.result, 'SUCCESS')
         flag_path = os.path.join(self.test_root, build.uuid + '.flag')
         self.assertTrue(os.path.exists(flag_path))
+        pre_flag_path = os.path.join(self.test_root, build.uuid +
+                                     '.pre.flag')
+        self.assertTrue(os.path.exists(pre_flag_path))
+        post_flag_path = os.path.join(self.test_root, build.uuid +
+                                      '.post.flag')
+        self.assertTrue(os.path.exists(post_flag_path))