From 423a33c04ba593addfe168d42b19713209aa5259 Mon Sep 17 00:00:00 2001
From: Jordan Welch <jordanhwelch@gmail.com>
Date: Mon, 22 Jan 2024 20:50:24 -0600
Subject: [PATCH] Add test for template

---
 __tests__/templates/MMM-CTA.njk.spec.js | 108 +++++++++++++++++++++++-
 1 file changed, 105 insertions(+), 3 deletions(-)

diff --git a/__tests__/templates/MMM-CTA.njk.spec.js b/__tests__/templates/MMM-CTA.njk.spec.js
index dbd1b0b..d8b77be 100644
--- a/__tests__/templates/MMM-CTA.njk.spec.js
+++ b/__tests__/templates/MMM-CTA.njk.spec.js
@@ -2,11 +2,113 @@ nunjucks = require('../../__mocks__/nunjucks');
 
 translate = (str) => str;
 
+let data;
+let template;
+
 describe('loading', () => {
-  it('shows loading', () => {
-    const payload = { loading: true };
-    const template = nunjucks.render('MMM-CTA.njk', payload);
+  beforeEach(() => {
+    data = { loading: true };
+    template = nunjucks.render('MMM-CTA.njk', data);
+  });
 
+  it('shows loading', () => {
     expect(template).toContain('LOADING');
   });
 });
+
+describe('train stop', () => {
+  beforeEach(() => {
+    data = {
+      stops: [{
+        type: 'train',
+        name: 'Train Stop',
+        arrivals: [
+          {
+            direction: 'North',
+            arrival: 5,
+          },
+          {
+            direction: 'South',
+            arrival: 10,
+          },
+        ],
+      }],
+    };
+    template = nunjucks.render('MMM-CTA.njk', data);
+  });
+
+  it('shows train stop name', () => {
+    expect(template).toContain('Train Stop');
+  });
+
+  it('shows train stop directions', () => {
+    expect(template).toContain('North');
+    expect(template).toContain('South');
+  });
+
+  it('shows train stop arrivals', () => {
+    expect(template).toContain('5');
+    expect(template).toContain('10');
+  });
+});
+
+describe('bus stop', () => {
+  beforeEach(() => {
+    data = {
+      stops: [{
+        type: 'bus',
+        name: 'Bus Stop',
+        arrivals: [
+          {
+            direction: 'North',
+            arrival: 5,
+          },
+          {
+            direction: 'South',
+            arrival: 10,
+          },
+        ],
+      }],
+    };
+    template = nunjucks.render('MMM-CTA.njk', data);
+  });
+
+  it('shows bus stop name', () => {
+    expect(template).toContain('Bus Stop');
+  });
+
+  it('shows bus stop directions', () => {
+    expect(template).toContain('North');
+    expect(template).toContain('South');
+  });
+
+  it('shows bus stop arrivals', () => {
+    expect(template).toContain('5');
+    expect(template).toContain('10');
+  });
+});
+
+describe('multiple stops', () => {
+  beforeEach(() => {
+    data = {
+      stops: [
+        {
+          type: 'bus',
+          name: 'Bus Stop',
+          arrivals: [],
+        },
+        {
+          type: 'train',
+          name: 'Train Stop',
+          arrivals: [],
+        },
+      ],
+    };
+    template = nunjucks.render('MMM-CTA.njk', data);
+  });
+
+  it('shows all stop names', () => {
+    expect(template).toContain('Bus Stop');
+    expect(template).toContain('Train Stop');
+  });
+});