From 64ce0cad9e04aab19eb4c3f61333b203548281a7 Mon Sep 17 00:00:00 2001
From: Simon Glass <sjg@chromium.org>
Date: Mon, 6 Jul 2015 12:54:31 -0600
Subject: [PATCH] dm: test: Add a test for the ram uclass

Add a test to confirm that we can probe this device and get information on
the available RAM.

Signed-off-by: Simon Glass <sjg@chromium.org>
---
 arch/sandbox/dts/test.dts |  4 ++++
 configs/sandbox_defconfig |  1 +
 drivers/ram/Makefile      |  1 +
 drivers/ram/sandbox_ram.c | 38 ++++++++++++++++++++++++++++++++++++++
 test/dm/Makefile          |  1 +
 test/dm/ram.c             | 28 ++++++++++++++++++++++++++++
 6 files changed, 73 insertions(+)
 create mode 100644 drivers/ram/sandbox_ram.c
 create mode 100644 test/dm/ram.c

diff --git a/arch/sandbox/dts/test.dts b/arch/sandbox/dts/test.dts
index 79b017598cb..cdc2a7b2ef3 100644
--- a/arch/sandbox/dts/test.dts
+++ b/arch/sandbox/dts/test.dts
@@ -196,6 +196,10 @@
 		};
 	};
 
+	ram {
+		compatible = "sandbox,ram";
+	};
+
 	reset@0 {
 		compatible = "sandbox,warm-reset";
 	};
diff --git a/configs/sandbox_defconfig b/configs/sandbox_defconfig
index 907554f7386..99dc00f30ce 100644
--- a/configs/sandbox_defconfig
+++ b/configs/sandbox_defconfig
@@ -47,3 +47,4 @@ CONFIG_UT_DM=y
 CONFIG_UT_ENV=y
 CONFIG_CLK=y
 CONFIG_RESET=y
+CONFIG_RAM=y
diff --git a/drivers/ram/Makefile b/drivers/ram/Makefile
index 4494d8187b0..0e102491a4a 100644
--- a/drivers/ram/Makefile
+++ b/drivers/ram/Makefile
@@ -5,3 +5,4 @@
 # SPDX-License-Identifier:      GPL-2.0+
 #
 obj-$(CONFIG_RAM) += ram-uclass.o
+obj-$(CONFIG_SANDBOX) += sandbox_ram.o
diff --git a/drivers/ram/sandbox_ram.c b/drivers/ram/sandbox_ram.c
new file mode 100644
index 00000000000..06bf3ece2c9
--- /dev/null
+++ b/drivers/ram/sandbox_ram.c
@@ -0,0 +1,38 @@
+/*
+ * Copyright (c) 2015 Google, Inc
+ * Written by Simon Glass <sjg@chromium.org>
+ *
+ * SPDX-License-Identifier:	GPL-2.0+
+ */
+
+#include <common.h>
+#include <dm.h>
+#include <errno.h>
+#include <ram.h>
+#include <asm/test.h>
+
+DECLARE_GLOBAL_DATA_PTR;
+
+static int sandbox_get_info(struct udevice *dev, struct ram_info *info)
+{
+	info->base = 0;
+	info->size = gd->ram_size;
+
+	return 0;
+}
+
+static const struct ram_ops sandbox_ram_ops = {
+	.get_info	= sandbox_get_info,
+};
+
+static const struct udevice_id sandbox_ram_ids[] = {
+	{ .compatible = "sandbox,ram" },
+	{ }
+};
+
+U_BOOT_DRIVER(warm_ram_sandbox) = {
+	.name		= "ram_sandbox",
+	.id		= UCLASS_RAM,
+	.of_match	= sandbox_ram_ids,
+	.ops		= &sandbox_ram_ops,
+};
diff --git a/test/dm/Makefile b/test/dm/Makefile
index d28a22f0184..f6a955cfafc 100644
--- a/test/dm/Makefile
+++ b/test/dm/Makefile
@@ -20,6 +20,7 @@ obj-$(CONFIG_DM_ETH) += eth.o
 obj-$(CONFIG_DM_GPIO) += gpio.o
 obj-$(CONFIG_DM_I2C) += i2c.o
 obj-$(CONFIG_DM_PCI) += pci.o
+obj-$(CONFIG_RAM) += ram.o
 obj-$(CONFIG_RESET) += reset.o
 obj-$(CONFIG_DM_RTC) += rtc.o
 obj-$(CONFIG_DM_SPI_FLASH) += sf.o
diff --git a/test/dm/ram.c b/test/dm/ram.c
new file mode 100644
index 00000000000..3a7c5fffddd
--- /dev/null
+++ b/test/dm/ram.c
@@ -0,0 +1,28 @@
+/*
+ * Copyright (C) 2015 Google, Inc
+ *
+ * SPDX-License-Identifier:	GPL-2.0+
+ */
+
+#include <common.h>
+#include <dm.h>
+#include <ram.h>
+#include <dm/test.h>
+#include <test/ut.h>
+
+DECLARE_GLOBAL_DATA_PTR;
+
+/* Basic test of the ram uclass */
+static int dm_test_ram_base(struct unit_test_state *uts)
+{
+	struct udevice *dev;
+	struct ram_info info;
+
+	ut_assertok(uclass_get_device(UCLASS_RAM, 0, &dev));
+	ut_assertok(ram_get_info(dev, &info));
+	ut_asserteq(0, info.base);
+	ut_asserteq(gd->ram_size, info.size);
+
+	return 0;
+}
+DM_TEST(dm_test_ram_base, DM_TESTF_SCAN_PDATA | DM_TESTF_SCAN_FDT);
-- 
GitLab