From e6de3126c120da1e055423a4f7418c0ab1725c6f Mon Sep 17 00:00:00 2001
From: gpetrochenkov-usgs <gpetrochenkov@usgs.gov>
Date: Wed, 3 Jun 2020 17:26:33 -0400
Subject: [PATCH] Change floodData floodlevels state in tests

---
 .../components/hydrograph/index.spec.js       | 15 ++++--------
 .../components/hydrograph/legend.spec.js      | 15 ++++--------
 .../hydrograph/time-series-graph.spec.js      | 15 ++++--------
 .../selectors/flood-data-selector.spec.js     | 23 ++++++++-----------
 4 files changed, 24 insertions(+), 44 deletions(-)

diff --git a/assets/src/scripts/components/hydrograph/index.spec.js b/assets/src/scripts/components/hydrograph/index.spec.js
index b0615342c..d5205208f 100644
--- a/assets/src/scripts/components/hydrograph/index.spec.js
+++ b/assets/src/scripts/components/hydrograph/index.spec.js
@@ -151,16 +151,11 @@ const TEST_STATE = {
     },
     floodData: {
         floodLevels: {
-            sites:
-                [
-                    {
-                        site_no: '07144100',
-                        action_stage: '20',
-                        flood_stage: '22',
-                        moderate_flood_stage: '25',
-                        major_flood_stage: '26'
-                    }
-                ]
+            site_no: '07144100',
+            action_stage: '20',
+            flood_stage: '22',
+            moderate_flood_stage: '25',
+            major_flood_stage: '26'
         }
     }
 };
diff --git a/assets/src/scripts/components/hydrograph/legend.spec.js b/assets/src/scripts/components/hydrograph/legend.spec.js
index 4bcdfe086..c17b6349c 100644
--- a/assets/src/scripts/components/hydrograph/legend.spec.js
+++ b/assets/src/scripts/components/hydrograph/legend.spec.js
@@ -97,16 +97,11 @@ describe('UV: Legend module', () => {
         },
         floodData: {
             floodLevels: {
-                sites:
-                    [
-                        {
-                            site_no: '07144100',
-                            action_stage: '20',
-                            flood_stage: '22',
-                            moderate_flood_stage: '25',
-                            major_flood_stage: '26'
-                        }
-                    ]
+                site_no: '07144100',
+                action_stage: '20',
+                flood_stage: '22',
+                moderate_flood_stage: '25',
+                major_flood_stage: '26'
             }
         }
     };
diff --git a/assets/src/scripts/components/hydrograph/time-series-graph.spec.js b/assets/src/scripts/components/hydrograph/time-series-graph.spec.js
index 3b13a2032..af75275ea 100644
--- a/assets/src/scripts/components/hydrograph/time-series-graph.spec.js
+++ b/assets/src/scripts/components/hydrograph/time-series-graph.spec.js
@@ -180,16 +180,11 @@ const TEST_STATE = {
     },
     floodData: {
         floodLevels: {
-            sites:
-                [
-                    {
-                        site_no: '07144100',
-                        action_stage: '20',
-                        flood_stage: '22',
-                        moderate_flood_stage: '25',
-                        major_flood_stage: '26'
-                    }
-                ]
+            site_no: '07144100',
+            action_stage: '20',
+            flood_stage: '22',
+            moderate_flood_stage: '25',
+            major_flood_stage: '26'
         }
     }
 };
diff --git a/assets/src/scripts/selectors/flood-data-selector.spec.js b/assets/src/scripts/selectors/flood-data-selector.spec.js
index d7c0b9d87..433e0271c 100644
--- a/assets/src/scripts/selectors/flood-data-selector.spec.js
+++ b/assets/src/scripts/selectors/flood-data-selector.spec.js
@@ -164,20 +164,15 @@ describe('flood-data-selector', () => {
 
          it('Return true if waterwatch flood levels should be visible', () =>{
             expect(waterwatchVisible({
-                floodData: {
-                    floodLevels: {
-                        sites:
-                            [
-                                {
-                                    site_no: '07144100',
-                                    action_stage: '20',
-                                    flood_stage: '22',
-                                    moderate_flood_stage: '25',
-                                    major_flood_stage: '26'
-                                }
-                            ]
-                    }
-                },
+                 floodData: {
+                     floodLevels: {
+                         site_no: '07144100',
+                         action_stage: '20',
+                         flood_stage: '22',
+                         moderate_flood_stage: '25',
+                         major_flood_stage: '26'
+                     }
+                 },
                 ivTimeSeriesState: {
                     currentIVVariableID: '45807197'
                 },
-- 
GitLab