| // Copyright 2018 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <limits.h> |
| #include <stddef.h> |
| |
| #include <fbl/unique_ptr.h> |
| #include <unittest/unittest.h> |
| #include <zircon/types.h> |
| |
| #include "utils.h" |
| |
| namespace blkctl { |
| namespace testing { |
| namespace { |
| |
| bool TestBadCommand(void) { |
| BEGIN_TEST; |
| |
| // Missing everything! |
| EXPECT_TRUE(ParseAndRun(ZX_ERR_INVALID_ARGS, "")); |
| |
| // Missing command |
| EXPECT_TRUE(ParseAndRun(ZX_ERR_INVALID_ARGS, "blkctl")); |
| |
| // Gibberish |
| EXPECT_TRUE(ParseAndRun(ZX_ERR_INVALID_ARGS, "blkctl booplesnoot")); |
| |
| END_TEST; |
| } |
| |
| // blkctl ls |
| bool TestList(void) { |
| BEGIN_TEST; |
| ScopedRamdisk ramdisk; |
| ASSERT_TRUE(ramdisk.Init()); |
| |
| // Too many args |
| EXPECT_TRUE(ParseAndRun(ZX_ERR_INVALID_ARGS, "blkctl ls foo")); |
| |
| // Valid |
| EXPECT_TRUE(ParseAndRun(ZX_OK, "blkctl ls")); |
| |
| END_TEST; |
| } |
| |
| // blkctl -d <dev> dump |
| bool TestDump(void) { |
| BEGIN_TEST; |
| ScopedRamdisk ramdisk; |
| ASSERT_TRUE(ramdisk.Init()); |
| |
| // Missing device |
| EXPECT_TRUE(ParseAndRun(ZX_ERR_INVALID_ARGS, "blkctl dump")); |
| |
| // Too many args |
| EXPECT_TRUE(ParseAndRun(ZX_ERR_INVALID_ARGS, "blkctl dump %s foo", ramdisk.path())); |
| |
| // Valid |
| EXPECT_TRUE(ParseAndRun(ZX_OK, "blkctl dump %s", ramdisk.path())); |
| |
| END_TEST; |
| } |
| |
| BEGIN_TEST_CASE(CommandTest) |
| RUN_TEST(TestBadCommand) |
| RUN_TEST(TestList) |
| RUN_TEST(TestDump) |
| END_TEST_CASE(CommandTest) |
| |
| } // namespace |
| } // namespace testing |
| } // namespace blkctl |