From 716b9c24c5c17b15c1ace22d11348c24dba5ed9a Mon Sep 17 00:00:00 2001
From: Abraham Williams <4braham@gmail.com>
Date: Sat, 19 Feb 2022 12:46:15 -0600
Subject: [PATCH] Add type guards

Fixes #227
---
 src/index.test.ts | 29 ++++++++++++++++++++++++++++-
 src/index.ts      |  9 +++++++++
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/index.test.ts b/src/index.test.ts
index 5bcc0a2..8387e64 100644
--- a/src/index.test.ts
+++ b/src/index.test.ts
@@ -1,4 +1,15 @@
-import { Failure, fold, Initialized, Kinds, Pending, Success } from './index';
+import {
+  Failure,
+  fold,
+  Initialized,
+  isFailure,
+  isInitialized,
+  isPending,
+  isSuccess,
+  Kinds,
+  Pending,
+  Success,
+} from './index';
 
 test('Kinds', () => {
   expect(Kinds).toEqual({
@@ -12,11 +23,15 @@ test('Kinds', () => {
 test('Initialized', () => {
   expect(new Initialized()).toBeInstanceOf(Initialized);
   expect(new Initialized().kind).toEqual('Initialized');
+  expect(isInitialized(new Initialized())).toBeTruthy();
+  expect(isInitialized(new Pending())).toBeFalsy();
 });
 
 test('Pending', () => {
   expect(new Pending()).toBeInstanceOf(Pending);
   expect(new Pending().kind).toEqual('Pending');
+  expect(isPending(new Pending())).toBeTruthy();
+  expect(isPending(new Initialized())).toBeFalsy();
 });
 
 test('Success', () => {
@@ -25,10 +40,16 @@ test('Success', () => {
   expect(state).toBeInstanceOf(Success);
   expect(state.kind).toEqual('Success');
   expect(state.data).toEqual(data);
+  expect(isSuccess(state)).toBeTruthy();
+  expect(isSuccess(new Initialized())).toBeFalsy();
 });
 
 test('Success without data', () => {
   expect(() => new Success()).toThrowError('Parameter "data" is required');
+  expect(() => new Success(null)).toThrowError('Parameter "data" is required');
+  expect(() => new Success(undefined)).toThrowError(
+    'Parameter "data" is required',
+  );
 });
 
 test('Failure', () => {
@@ -37,10 +58,16 @@ test('Failure', () => {
   expect(state).toBeInstanceOf(Failure);
   expect(state.kind).toEqual('Failure');
   expect(state.error).toEqual(error);
+  expect(isFailure(state)).toBeTruthy();
+  expect(isFailure(new Initialized())).toBeFalsy();
 });
 
 test('Failure without error', () => {
   expect(() => new Failure()).toThrowError('Parameter "error" is required');
+  expect(() => new Failure(null)).toThrowError('Parameter "error" is required');
+  expect(() => new Failure(undefined)).toThrowError(
+    'Parameter "error" is required',
+  );
 });
 
 test('fold initialized', () => {
diff --git a/src/index.ts b/src/index.ts
index 671f499..0e0a113 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -62,3 +62,12 @@ export function fold<T, E, D>(
     }
   };
 }
+
+export const isInitialized = (state: unknown): state is Initialized =>
+  state instanceof Initialized;
+export const isPending = (state: unknown): state is Pending =>
+  state instanceof Pending;
+export const isFailure = <E>(state: unknown): state is Failure<E> =>
+  state instanceof Failure;
+export const isSuccess = <D>(state: unknown): state is Success<D> =>
+  state instanceof Success;