Merge pull request #12 from RIT-SWEN-262/menu-reorganize

Menu reorganize

1 files changed, 11 insertions, 0 deletions

diff --git a/src/main/java/design/runtime/Session.java b/src/main/java/design/runtime/Session.java
index fbf4b9c..59d31d8 100644
--- a/src/main/java/design/runtime/Session.java
+++ b/src/main/java/design/runtime/Session.java
@@ -2,8 +2,11 @@ package design.runtime;
 
 import design.model.Golfer;
 
+import java.time.LocalDateTime;
+
 public final class Session {
     private static Golfer currentGolfer;
+    private static LocalDateTime timeOverride;
 
     public static Golfer getCurrentGolfer() {
         return currentGolfer;
@@ -12,4 +15,12 @@ public final class Session {
     public static void setCurrentGolfer(Golfer currentGolfer) {
         Session.currentGolfer = currentGolfer;
     }
+
+    public static LocalDateTime getDateTime() {
+        return timeOverride != null ? timeOverride : LocalDateTime.now();
+    }
+
+    public static void setTimeOverride(LocalDateTime timeOverride) {
+        Session.timeOverride = timeOverride;
+    }
 }