/*
** Copyright (C) 2008-2020 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
**
** This program is free software; you can redistribute it and/or modify it
** under the terms of the GNU General Public License as published by the
** Free Software Foundation; either version 3, or (at your option) any
** later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software Foundation,
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**
*/

#if HAVE_CONFIG_H
#include "config.h"
#endif /*HAVE_CONFIG_H*/

#include <glib.h>
#include "mu-flags.hh"
#include "test-mu-common.hh"

using namespace Mu;

static void
test_mu_flag_char (void)
{
	g_assert_cmpuint (mu_flag_char (MU_FLAG_DRAFT),		==, 'D');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_FLAGGED),	==, 'F');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_PASSED),	==, 'P');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_REPLIED),	==, 'R');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_SEEN),		==, 'S');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_TRASHED),	==, 'T');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_NEW),		==, 'N');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_SIGNED),	==, 'z');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_ENCRYPTED),	==, 'x');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_HAS_ATTACH),	==, 'a');
	g_assert_cmpuint (mu_flag_char (MU_FLAG_UNREAD),	==, 'u');
	g_assert_cmpuint (mu_flag_char ((MuFlags)12345),        ==,  0);
}



static void
test_mu_flag_name (void)
{
	g_assert_cmpstr (mu_flag_name (MU_FLAG_DRAFT),		==, "draft");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_FLAGGED),	==, "flagged");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_PASSED),		==, "passed");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_REPLIED),	==, "replied");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_SEEN),		==, "seen");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_TRASHED),	==, "trashed");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_NEW),		==, "new");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_SIGNED),		==, "signed");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_ENCRYPTED),	==, "encrypted");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_HAS_ATTACH),	==, "attach");
	g_assert_cmpstr (mu_flag_name (MU_FLAG_UNREAD),		==, "unread");
	g_assert_cmpstr (mu_flag_name ((MuFlags)12345),	        ==,  NULL);
}

static void
test_mu_flags_to_str_s (void)
{
	g_assert_cmpstr (mu_flags_to_str_s((MuFlags)(MU_FLAG_PASSED|MU_FLAG_SIGNED),
					   MU_FLAG_TYPE_ANY),
			 ==, "Pz");
	g_assert_cmpstr (mu_flags_to_str_s(MU_FLAG_NEW, MU_FLAG_TYPE_ANY),
			 ==, "N");
	g_assert_cmpstr (mu_flags_to_str_s((MuFlags)(MU_FLAG_HAS_ATTACH|MU_FLAG_TRASHED),
					   MU_FLAG_TYPE_ANY),
			 ==, "Ta");
	g_assert_cmpstr (mu_flags_to_str_s(MU_FLAG_NONE, MU_FLAG_TYPE_ANY),
			 ==, "");

	g_assert_cmpstr (mu_flags_to_str_s((MuFlags)(MU_FLAG_PASSED|MU_FLAG_SIGNED),
					   MU_FLAG_TYPE_CONTENT),
			 ==, "z");

	g_assert_cmpstr (mu_flags_to_str_s(MU_FLAG_NEW, MU_FLAG_TYPE_MAILDIR),
			 ==, "N");
	g_assert_cmpstr (mu_flags_to_str_s((MuFlags)(MU_FLAG_HAS_ATTACH|MU_FLAG_TRASHED),
					   MU_FLAG_TYPE_MAILFILE),
			 ==, "T");

	g_assert_cmpstr (mu_flags_to_str_s(MU_FLAG_NONE, MU_FLAG_TYPE_PSEUDO),
			 ==, "");
}


static void
test_mu_flags_from_str (void)
{
	/* note, the 3rd arg to mu_flags_from_str determines whether
	 * invalid flags will be ignored (if TRUE) or MU_FLAG_INVALID (if FALSE)
	 */

	g_assert_cmpuint (mu_flags_from_str ("RP", MU_FLAG_TYPE_ANY, TRUE), ==,
			 (MuFlags)( MU_FLAG_REPLIED | MU_FLAG_PASSED));
	g_assert_cmpuint (mu_flags_from_str ("Nz", MU_FLAG_TYPE_ANY, TRUE), ==,
			  MU_FLAG_NEW | MU_FLAG_SIGNED);
	g_assert_cmpuint (mu_flags_from_str ("axD", MU_FLAG_TYPE_ANY, TRUE), ==,
			 (MuFlags)( MU_FLAG_HAS_ATTACH | MU_FLAG_ENCRYPTED | MU_FLAG_DRAFT));

	g_assert_cmpuint (mu_flags_from_str ("RP", MU_FLAG_TYPE_MAILFILE, TRUE), ==,
			  MU_FLAG_REPLIED | MU_FLAG_PASSED);
	g_assert_cmpuint (mu_flags_from_str ("Nz", MU_FLAG_TYPE_MAILFILE, TRUE), ==,
			  MU_FLAG_NONE);

	/* ignore errors or not */
	g_assert_cmpuint (mu_flags_from_str ("qwi", MU_FLAG_TYPE_MAILFILE, FALSE), ==,
			  MU_FLAG_INVALID);
	g_assert_cmpuint (mu_flags_from_str ("qwi", MU_FLAG_TYPE_MAILFILE, TRUE), ==,
			  0);


}

static void
test_mu_flags_from_str_delta (void)
{
	g_assert_cmpuint (mu_flags_from_str_delta ("+S-R",
						   (MuFlags)(MU_FLAG_REPLIED | MU_FLAG_DRAFT),
						   MU_FLAG_TYPE_ANY),==,
			  (MuFlags)(MU_FLAG_SEEN | MU_FLAG_DRAFT));

	g_assert_cmpuint (mu_flags_from_str_delta ("",
						   (MuFlags)(MU_FLAG_REPLIED | MU_FLAG_DRAFT),
						   MU_FLAG_TYPE_ANY),==,
			  (MuFlags)(MU_FLAG_REPLIED | MU_FLAG_DRAFT));

	g_assert_cmpuint (mu_flags_from_str_delta ("-N+P+S-D",
						   (MuFlags)(MU_FLAG_SIGNED | MU_FLAG_DRAFT),
						   MU_FLAG_TYPE_ANY),==,
			  (MuFlags)(MU_FLAG_PASSED | MU_FLAG_SEEN | MU_FLAG_SIGNED));
}


static void
test_mu_flags_custom_from_str (void)
{
	unsigned u;

	struct {
		const char *str;
		const char *expected;
	} cases[] = {
		{ "ABC", "ABC" },
		{ "PAF", "A" },
		{ "ShelloPwoFrDldR123", "helloworld123" },
		{ "SPD", NULL }
	};

	for (u = 0; u != G_N_ELEMENTS(cases); ++u) {
		char *cust;
		cust = mu_flags_custom_from_str (cases[u].str);
		if (g_test_verbose())
			g_print ("%s: str:%s; got:%s; expected:%s\n",
				 __func__, cases[u].str, cust, cases[u].expected);
		g_assert_cmpstr (cust, ==, cases[u].expected);
		g_free (cust);
	}
}



int
main (int argc, char *argv[])
{
	int rv;
	g_test_init (&argc, &argv, NULL);

	/* mu_msg_str_date */
	g_test_add_func ("/mu-flags/test-mu-flag-char", test_mu_flag_char);
	g_test_add_func ("/mu-flags/test-mu-flag-name",test_mu_flag_name);
	g_test_add_func ("/mu-flags/test-mu-flags-to-str-s",test_mu_flags_to_str_s);
	g_test_add_func ("/mu-flags/test-mu-flags-from-str",test_mu_flags_from_str);
	g_test_add_func ("/mu-flags/test-mu-flags-from-str-delta",test_mu_flags_from_str_delta );
	g_test_add_func ("/mu-flags/test-mu-flags-custom-from-str",
			 test_mu_flags_custom_from_str);

	g_log_set_handler (NULL,
			   (GLogLevelFlags)(G_LOG_LEVEL_MASK | G_LOG_FLAG_FATAL|
					    G_LOG_FLAG_RECURSION),
			   (GLogFunc)black_hole, NULL);

	rv = g_test_run ();

	return rv;
}
