From bde4d1016b4cdebaf4237ea3f50dde306cbe4115 Mon Sep 17 00:00:00 2001
From: zenith391 <39484230+zenith391@users.noreply.github.com>
Date: Mon, 26 Feb 2024 17:41:12 +0100
Subject: [PATCH] fix tests

---
 src/components/Button.zig |  5 +++--
 src/components/Canvas.zig | 13 ++++++++-----
 src/data.zig              |  2 +-
 src/widget.zig            |  2 ++
 4 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/components/Button.zig b/src/components/Button.zig
index 6de9ec6..e7bdd44 100644
--- a/src/components/Button.zig
+++ b/src/components/Button.zig
@@ -64,9 +64,10 @@ pub fn button(config: Button.Config) *Button {
     return Button.alloc(config);
 }
 
-test "button" {
+test Button {
     var btn = button(.{ .label = "Test Label" });
-    defer btn.deinit();
+    btn.ref(); // because we're keeping a reference, we need to ref() it
+    defer btn.unref();
     try std.testing.expectEqualStrings("Test Label", btn.getLabel());
 
     btn.setLabel("New Label");
diff --git a/src/components/Canvas.zig b/src/components/Canvas.zig
index 9a9ae76..cb30ef5 100644
--- a/src/components/Canvas.zig
+++ b/src/components/Canvas.zig
@@ -107,17 +107,20 @@ pub fn rect(config: Rect.Config) *Rect {
 
 const fuzz = @import("../fuzz.zig");
 
-test "instantiate Canvas" {
+test Canvas {
     var cnv = canvas(.{});
-    defer cnv.deinit();
+    cnv.ref(); // because we keep a reference to canvas we must call ref()
+    defer cnv.unref();
 }
 
-test "instantiate Rect" {
+test Rect {
     var rect1 = rect(.{ .color = Color.blue });
-    defer rect1.deinit();
+    rect1.ref();
+    defer rect1.unref();
     try std.testing.expectEqual(Color.blue, rect1.color.get());
 
     var rect2 = rect(.{ .color = Color.yellow });
-    defer rect2.deinit();
+    rect2.ref();
+    defer rect2.unref();
     try std.testing.expectEqual(Color.yellow, rect2.color.get());
 }
diff --git a/src/data.zig b/src/data.zig
index 3234695..0bacfb5 100644
--- a/src/data.zig
+++ b/src/data.zig
@@ -860,7 +860,7 @@ test "animated atom" {
     var original = Atom(i32).of(0);
     defer original.deinit();
 
-    var animated = try Atom(i32).animated(&original, Easings.Linear, 1000);
+    var animated = try Atom(i32).withImplicitAnimation(&original, Easings.Linear, 1000);
     defer animated.deinit();
     defer _animatedAtoms.clearAndFree();
     defer _animatedAtomsLength.set(0);
diff --git a/src/widget.zig b/src/widget.zig
index 334960a..9c8743f 100644
--- a/src/widget.zig
+++ b/src/widget.zig
@@ -110,6 +110,8 @@ const TestType = struct {
         .getParentFn = undefined,
         .isDisplayedFn = undefined,
         // .cloneFn = undefined,
+        .ref_fn = undefined,
+        .unref_fn = undefined,
     };
 };