Paul Stewart | d408fdf | 2012-05-07 17:15:57 -0700 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "shill/dhcp_provider.h" |
Chris Masone | 2b10554 | 2011-06-22 10:58:09 -0700 | [diff] [blame] | 6 | |
Albert Chaulk | 0e1cdea | 2013-02-27 15:32:55 -0800 | [diff] [blame] | 7 | #include <base/file_util.h> |
| 8 | #include <base/files/scoped_temp_dir.h> |
| 9 | #include <base/stringprintf.h> |
| 10 | |
Chris Masone | 2b10554 | 2011-06-22 10:58:09 -0700 | [diff] [blame] | 11 | #include "shill/dhcp_config.h" |
Chris Masone | 19e3040 | 2011-07-19 15:48:47 -0700 | [diff] [blame] | 12 | #include "shill/mock_control.h" |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 13 | #include "shill/mock_glib.h" |
| 14 | |
Albert Chaulk | 0e1cdea | 2013-02-27 15:32:55 -0800 | [diff] [blame] | 15 | using base::FilePath; |
| 16 | using base::ScopedTempDir; |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 17 | using testing::Test; |
| 18 | |
| 19 | namespace shill { |
| 20 | |
| 21 | namespace { |
| 22 | const char kDeviceName[] = "testdevicename"; |
Paul Stewart | d32f484 | 2012-01-11 16:08:13 -0800 | [diff] [blame] | 23 | const char kHostName[] = "testhostname"; |
Paul Stewart | d408fdf | 2012-05-07 17:15:57 -0700 | [diff] [blame] | 24 | const char kStorageIdentifier[] = "teststorageidentifier"; |
| 25 | const bool kArpGateway = false; |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 26 | } // namespace {} |
| 27 | |
| 28 | class DHCPProviderTest : public Test { |
| 29 | public: |
Darin Petkov | f65e928 | 2011-06-21 14:29:56 -0700 | [diff] [blame] | 30 | DHCPProviderTest() : provider_(DHCPProvider::GetInstance()) { |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 31 | provider_->glib_ = &glib_; |
Chris Masone | 19e3040 | 2011-07-19 15:48:47 -0700 | [diff] [blame] | 32 | provider_->control_interface_ = &control_; |
Han Shen | 4afba20 | 2012-12-17 08:48:35 -0800 | [diff] [blame] | 33 | // DHCPProvider is a singleton, there is no guarentee that it is |
| 34 | // not setup/used elsewhere, so reset its state before running our |
| 35 | // tests. |
| 36 | provider_->configs_.clear(); |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 37 | } |
| 38 | |
| 39 | protected: |
Chris Masone | 19e3040 | 2011-07-19 15:48:47 -0700 | [diff] [blame] | 40 | MockControl control_; |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 41 | MockGLib glib_; |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 42 | DHCPProvider *provider_; |
| 43 | }; |
| 44 | |
| 45 | TEST_F(DHCPProviderTest, CreateConfig) { |
Paul Stewart | d408fdf | 2012-05-07 17:15:57 -0700 | [diff] [blame] | 46 | DHCPConfigRefPtr config = provider_->CreateConfig(kDeviceName, |
| 47 | kHostName, |
| 48 | kStorageIdentifier, |
| 49 | kArpGateway); |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 50 | EXPECT_TRUE(config.get()); |
| 51 | EXPECT_EQ(&glib_, config->glib_); |
Darin Petkov | f65e928 | 2011-06-21 14:29:56 -0700 | [diff] [blame] | 52 | EXPECT_EQ(kDeviceName, config->device_name()); |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 53 | EXPECT_TRUE(provider_->configs_.empty()); |
| 54 | } |
| 55 | |
Albert Chaulk | 0e1cdea | 2013-02-27 15:32:55 -0800 | [diff] [blame] | 56 | TEST_F(DHCPProviderTest, DestroyLease) { |
| 57 | ScopedTempDir temp_dir; |
| 58 | FilePath lease_file; |
| 59 | EXPECT_TRUE(temp_dir.CreateUniqueTempDir()); |
| 60 | provider_->root_ = temp_dir.path(); |
| 61 | lease_file = provider_->root_.Append(base::StringPrintf( |
| 62 | DHCPProvider::kDHCPCDPathFormatLease, |
| 63 | kDeviceName)); |
| 64 | EXPECT_TRUE(file_util::CreateDirectory(lease_file.DirName())); |
| 65 | EXPECT_EQ(0, file_util::WriteFile(lease_file, "", 0)); |
| 66 | EXPECT_TRUE(file_util::PathExists(lease_file)); |
| 67 | provider_->DestroyLease(kDeviceName); |
| 68 | EXPECT_FALSE(file_util::PathExists(lease_file)); |
| 69 | } |
| 70 | |
Darin Petkov | 98dd6a0 | 2011-06-10 15:12:57 -0700 | [diff] [blame] | 71 | } // namespace shill |