From 3e426cb80b032bf6a5b192d866e0e67c89a2b864 Mon Sep 17 00:00:00 2001
From: "Pfeil, Scott Robert" <scott.pfeil3@verizonwireless.com>
Date: Thu, 29 Aug 2019 10:19:28 -0400
Subject: [PATCH] Fix to two button view

---
 .../TwoButtonView.swift                        | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/MVMCoreUI/Molecules/HorizontalCombinationViews/TwoButtonView.swift b/MVMCoreUI/Molecules/HorizontalCombinationViews/TwoButtonView.swift
index 4b4f64d9..575c01ec 100644
--- a/MVMCoreUI/Molecules/HorizontalCombinationViews/TwoButtonView.swift
+++ b/MVMCoreUI/Molecules/HorizontalCombinationViews/TwoButtonView.swift
@@ -8,12 +8,13 @@
 
 import UIKit
 
-@objcMembers open class TwoButtonView: ButtonView {
-    open var secondaryButton: PrimaryButton?
+@objcMembers open class TwoButtonView: ViewConstrainingView {
+    open var primaryButton: PrimaryButton? = PrimaryButton.button()
+    open var secondaryButton: PrimaryButton? = PrimaryButton.button()
     open var viewForButtons: UIView?
     public var heightConstraint: NSLayoutConstraint?
     
-    public override init() {
+    public init() {
         super.init(frame: .zero)
     }
     
@@ -34,25 +35,32 @@ import UIKit
     open override func updateView(_ size: CGFloat) {
         super.updateView(size)
         MVMCoreDispatchUtility.performBlock(onMainThread: {
+            self.primaryButton?.updateView(size)
             self.secondaryButton?.updateView(size)
         })
     }
     
+    open override func setupView() {
+        super.setupView()
+        setupButton()
+    }
+    
     // MARK: - MVMCoreUIMoleculeViewProtocol
     open override func setWithJSON(_ json: [AnyHashable: Any]?, delegateObject: MVMCoreUIDelegateObject?, additionalData: [AnyHashable: Any]?) {
-        super.setWithJSON(json, delegateObject: delegateObject, additionalData: additionalData)
         let primaryButtonMap = json?.optionalDictionaryForKey("primaryButton")
         let secondaryButtonMap = json?.optionalDictionaryForKey("secondaryButton")
         set(primaryButtonJSON: primaryButtonMap, secondaryButtonJSON: secondaryButtonMap, delegateObject: delegateObject, additionalData: additionalData)
+        super.setWithJSON(json, delegateObject: delegateObject, additionalData: additionalData)
     }
     
     open override func reset() {
         super.reset()
+        primaryButton?.setAsStandardCustom()
         secondaryButton?.setAsSecondaryCustom()
     }
     
     // MARK: - Constraining
-    override func setupButton() {
+    func setupButton() {
         setupWithTwoButtons()
     }