1 files changed, 18 insertions, 0 deletions
diff --git a/t/cobrand/staging.t b/t/cobrand/staging.t
new file mode 100644
index 000000000..5d79daa4b
--- /dev/null
+++ b/t/cobrand/staging.t
@@ -0,0 +1,18 @@
+use FixMyStreet::TestMech;
+my $mech = FixMyStreet::TestMech->new;
+
+subtest 'staging banner is visible by default on staging sites' => sub {
+    $mech->get_ok('/');
+    $mech->content_contains('<div class="dev-site-notice">');
+};
+
+FixMyStreet::override_config {
+    STAGING_FLAGS => { hide_staging_banner => 1 },
+}, sub {
+    subtest 'staging banner can be hidden through STAGING_FLAGS config' => sub {
+        $mech->get_ok('/');
+        $mech->content_lacks('<div class="dev-site-notice">');
+    };
+};
+
+done_testing();