def test_update_with_db(self): expected_bw_usage1 = self._fake_bw_usage( time=self.expected_bw_usage['last_refreshed'], start_period=self.expected_bw_usage['start_period'], last_ctr_in=42, last_ctr_out=42) bw_usage = bandwidth_usage.BandwidthUsage(context=self.context) bw_usage.create(uuids.instance, 'fake_mac1', 100, 200, 42, 42, start_period=self.expected_bw_usage['start_period']) self._compare(self, expected_bw_usage1, bw_usage, ignored_fields=['last_refreshed', 'created_at']) bw_usage.create(uuids.instance, 'fake_mac1', 100, 200, 12345, 67890, start_period=self.expected_bw_usage['start_period']) self._compare( self, self.expected_bw_usage, bw_usage, ignored_fields=['last_refreshed', 'created_at', 'updated_at'])
def test_update(self, mock_update): expected_bw_usage1 = self._fake_bw_usage( time=self.expected_bw_usage['last_refreshed'], start_period=self.expected_bw_usage['start_period'], last_ctr_in=42, last_ctr_out=42) mock_update.side_effect = [expected_bw_usage1, self.expected_bw_usage] bw_usage = bandwidth_usage.BandwidthUsage() bw_usage.create(self.context, 'fake_uuid1', 'fake_mac1', 100, 200, 42, 42, start_period=self.expected_bw_usage['start_period']) self._compare(self, expected_bw_usage1, bw_usage) bw_usage.create(self.context, 'fake_uuid1', 'fake_mac1', 100, 200, 12345, 67890, start_period=self.expected_bw_usage['start_period']) self._compare(self, self.expected_bw_usage, bw_usage)
def test_create(self, mock_create): mock_create.return_value = self.expected_bw_usage bw_usage = bandwidth_usage.BandwidthUsage() bw_usage.create(self.context, 'fake_uuid', 'fake_mac', 100, 200, 12345, 67890, start_period=self.expected_bw_usage['start_period']) self._compare(self, self.expected_bw_usage, bw_usage)